在 UI 自动化测试过程中，面对复杂的业务场景，经常会遇到这样的挑战：
简单的录制/回放速度快，但无法适应复杂场景；
编写自动化测试脚本比较灵活，但工作量大且可维护性差；
以往的封装技术（PageObject）可以适应各种 UI 场景，但结构松散，无法在多项目中迁移；
因此，测试团队通常还需要一种定制测试框架，用以弥补现有框架的缺点。
测试框架封装思想
由于 UI 自动化测试框架围绕 UI 界面使用，因此，依旧选用 PageObject 设计模式对 UI 及测试进行封装，同时配合 Pytest 单元测试将脚本能够有效的组织、连贯应用起来，从而提高框架的可维护性和可读性。
由于测试框架基于 PageObject 设计模式，主要方向为 PO 改进，数据驱动，异常处理等，比如：
测试数据的数据驱动：将数据存储到外部 yaml 文件中，利用 yaml 工具进行数据读取；
数据步骤的数据驱动：将操作步骤放到外部 yaml 文件中，利用 yaml 工具对操作步骤进行读取，用专门函数解析并实现操作步骤；
自动化异常处理机制：对元素查找模块进行封装和改进，包括如何处理弹窗；
Page_Object 改造
作为通用的 UI 测试框架， PageObjet 不仅适用于 Web 自动化测试，也可适用 Appium 移动自动化测试，其优点如下：
减少代码重复
提高测试用例可读性
提高测试用例可维护性
PO 改造实例（基于雪球 App）
本案例将对雪球 App 进行 Page Objetct 封装与改进。
当启动雪球 App 时，会进入首页。点击搜索框进入搜索页，搜索某支股票然后判断股价是否大于 200：
PageObjetct 的模块关系如下，所有的模块要继承 BasePage ， App 实现启动，重启，停止等操作， Main 实现进入搜索页，进入股票页等操作：
base_page 模块是所有 page 类的父类，其中定义了公共方法，比如封装下面的 find 方法后，可以让子类调用 find ：
from appium.webdriver.webdriver import WebDriver
class BasePage:
_driver: WebDriver
def __init__(self, driver: WebDriver = None):
self._driver = driver
def find(self, locator, value: str = None):
#如果传进来的是tuple，只需使用一个参数：locator
if isinstance(locator, tuple):
return self._driver.find_element(*locator)
else:
return self._driver.find_element(locator, value)
App 模块封装 app 的启动，重启，停止等方法，当 app 启动时会进入 main 页面，因此在下面的 main 方法要 return Main ，Main 类的定义在后面会讲解:
from appium import webdriver
from test_appium.page.base_page import BasePage
from test_appium.page.main import Main
class App(BasePage):
#指定app的包名和activity名
_package = "com.xueqiu.android"
_activity = ".view.WelcomeActivityAlias"
def start(self):
#如果driver为空则初始化
if self._driver is None:
caps = {}
caps["platformName"] = "android"
caps["deviceName"] = "hogwarts"
caps["appPackage"] = self._package
caps["appActivity"] = self._activity
caps["noReset"] = True
#初始化driver
self._driver = webdriver.Remote("http://localhost:4723/wd/hub", caps)
self._driver.implicitly_wait(30)
#如果driver不为空，则直接启动activity
else:
print(self._driver)
self._driver.start_activity(self._package, self._activity)
return self
def restart(self):
pass
def stop(self):
pass
def main(self) -> Main:
#当app启动时，跳转到（实例化）Main
return Main(self._driver)
Main 模块是首页的 PageObject ，其中的方法封装了首页的重要功能，比如下面代码中的 goto_search_page 封装了点击搜索并跳转到 Search 页：
from appium.webdriver.common.mobileby import MobileBy
from selenium.webdriver.common.by import By
from test_appium.page.base_page import BasePage
from test_appium.page.profile import Profile
from test_appium.page.search import Search
class Main(BasePage):
#点击搜索按钮后，进入搜索页
def goto_search_page(self):
self.find(MobileBy.ID, "tv_search").click()
#进入搜索页
return Search(self._driver)
def goto_stocks(self):
pass
def goto_trade(self):
pass
def goto_messages(self):
pass
Search 模块可以搜索一支股票，还可以获取股票的价格，比如下图：
封装代码如下：
from appium.webdriver.common.mobileby import MobileBy
from selenium.webdriver.remote.webdriver import WebDriver
class Search:
_driver: WebDriver
def __init__(self, driver):
self._driver = driver
#输入要搜索的内容
def search(self, key: str):
self._driver.find_element(MobileBy.ID, "search_input_text").send_keys(key)
self._driver.find_element(MobileBy.ID, "name").click()
return self
#获取股票价格，用于判断
def get_price(self, key: str) -> float:
return float(self._driver.find_element(MobileBy.ID, "current_price").text)
最后对上述代码建立测试，新建测试模块 test_search ：
import pytest
from test_appium.page.app import App
class TestSearch:
def setup(self):
self.main = App().start().main()
def test_search(self):
assert self.main.goto_search_page().search("alibaba").get_price("BABA") > 200
1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
以上，供大家参考，欢迎一起留言探讨。
-----------------------------------
©著作权归作者所有：来自51CTO博客作者ceshiren2022的原创作品，请联系作者获取转载授权，否则将追究法律责任
实战 | UI 自动化测试框架设计与 PageObject 改造
