本文节选自霍格沃兹《测试开发实战进阶》课程教学内容。
在 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):
passSearch 模块可以搜索一支股票，还可以获取股票的价格，比如下图：
封装代码如下：
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（文章来源于霍格沃兹测试学院）