UI 自动化测试框架设计与 PageObject 改造

在 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

以上，供大家参考，欢迎一起留言探讨。

(文章来源于霍格沃兹测试学院）