Python+Selenium+PO设计模式实战指南

1. PO设计模式概述

PO设计模式将网页的页面元素和操作封装到类中，每个页面对应一个类，每个类中的方法代表与页面交互的操作。这种方式有助于将页面元素的定位和操作与测试脚本的逻辑解耦，从而提高代码的可维护性。

2. PO设计模式的核心思想

• 页面与测试代码分离：页面类只关心页面元素的定位和操作，而测试类只关注测试逻辑。
• 封装页面元素和操作：页面对象类封装了页面中的所有元素以及对这些元素的操作。
• 提高复用性：相同页面的不同测试用例可以重用相同的页面对象类。

3. Page Object模型的优缺点

优点：

• 代码复用：多个测试用例可以共享一个页面对象，避免重复代码。
• 易于维护：当页面发生变化时，只需要修改页面对象类，不需要改动所有的测试用例。
• 清晰的层次结构：测试脚本关注测试逻辑，页面对象关注页面交互，职责分离。

缺点：

• 学习曲线：对于初学者来说，理解和实现PO模式需要一定的学习时间。
• 类的数量增加：每个页面都需要一个页面对象类，可能导致类文件过多。

4. PO设计模式的结构

PO设计模式通常包括三个部分：

• Page Object：页面对象类，封装了页面元素和操作。
• Test Case：测试用例类，调用页面对象进行测试。
• Test Driver：测试驱动脚本，运行测试用例。

5. 如何使用Selenium + unittest结合PO设计模式

使用Selenium和unittest框架编写自动化测试时，可以按照以下步骤实现PO设计模式。

封装基类

from selenium.common import TimeoutException
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait


class Base:
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.get(r'http://localhost:8080/')

    def base_find_element(self, loc, timeout=10, poll=0.5):
        """
        定位元素
        :param loc: 元组 (By.ID,属性值) (By.CLASS_NAME,属性值) (By.XPATH,属性值)
        :param timeout: 超时时间
        :param poll: 轮询间隔
        :return: 元素定位对象
        """
        return WebDriverWait(self.driver, timeout, poll).until(lambda x: x.find_element(*loc))

    def base_click(self, loc):
        self.base_find_element(loc).click()

    def base_input(self, loc, value):
        el = self.base_find_element(loc)
        el.clear()
        el.send_keys(value)

    def base_get_info(self, loc, path):
        try:
            self.base_find_element(loc)
            return True
        except TimeoutException:
            self.base_get_image(path)
            return False

    def base_get_image(self, path):
        return self.driver.get_screenshot_as_file(path)

步骤1：定义页面对象类

页面对象类封装了与页面元素的交互，常见的操作包括点击按钮、输入文本、获取页面内容等。

from selenium.webdriver.common.by import By

"""登录页数据"""
login_login_link = By.XPATH, "/html/body/header/div/div/nav/div/div/a[1]"
login_username = By.NAME, 'username'
login_password = By.NAME, 'password'
login_login_button = By.XPATH, "/html/body/div[1]/div/div[2]/form/button"
login_login_info = By.XPATH, "/html/body/header/div/div/nav/div/div/div/button"

from v4.base import Base
from v4 import page


class PageLogin(Base):

    def page_click_login(self):
        """点击登录链接"""
        self.base_click(page.login_login_link)

    def page_input_username(self, username):
        """输入用户名"""
        self.base_input(page.login_username, username)

    def page_input_password(self, password):
        """输入密码"""
        self.base_input(page.login_password, password)

    def page_click_login_button(self):
        """点击登录按钮"""
        self.base_click(page.login_login_button)

    def page_get_info(self, path, username, password):
        """获取登录信息"""
        if self.base_get_info(page.login_login_info, path):
            print(f"账号：{username} 密码：{password} 登录成功")
        else:
            print(f"账号：{username} 密码：{password} 登录失败")

    def page_login(self, username, password, path):
        """登录操作"""
        self.page_click_login()
        self.page_input_username(username)
        self.page_input_password(password)
        self.page_click_login_button()
        self.page_get_info(path, username, password)

步骤2：编写测试用例

测试用例类继承unittest.TestCase，在其中编写具体的测试逻辑。页面对象类的实例化和方法调用将在测试用例中进行。

import unittest
from parameterized import parameterized
from v4.page.page_login import PageLogin


class TestLogin(unittest.TestCase):

    def setUp(self):
        self.login = PageLogin()

    def tearDown(self):
        self.login.driver.quit()

    @parameterized.expand([
        ("demo", "demo123", "demo_demo123.png"),
        ("falamo1", "falamo1", "falamo1_falamo1.png")
    ])
    def test_login(self, username, password, path):
        self.login.page_login(username, password, path)


if __name__ == '__main__':
    unittest.main()

6. 总结

PO设计模式可以让自动化测试更加模块化、易于维护和扩展。通过将页面操作封装到页面对象类中，我们可以让测试用例更加简洁，便于理解和修改。此外，PO模式能够有效地提高自动化测试的复用性和可扩展性。