### 使用 Python 和 Selenium 封装 POM 设计模式 #### 1. 页面对象模型（POM）简介 页面对象模型是一种用于 Web UI 测试的设计模式，它通过将页面上的交互逻辑抽象成独立的对象来提高代码的可维护性和重用性。这种设计模式的核心思想是将每个网页视为一个单独的类，并将该网页上的所有元素及其操作封装到这个类中。 --- #### 2. 创建项目结构 为了更好地管理代码，建议按照以下方式组织项目的文件夹结构： ``` project/ │ ├── page_objects/ # 存放所有的页面对象类 │ ├── base_page.py # 基础页面类 │ ├── login_page.py # 登录页面类 │ └── news_page.py # 新闻页面类 │ ├── tests/ # 存放测试脚本 │ ├── test_login.py # 登录功能测试 │ └── test_news.py # 新闻功能测试 │ ├── conftest.py # Pytest 配置文件 └── pytest.ini # Pytest 的全局配置 ``` --- #### 3. 编写基础页面类 `base_page.py` 基础页面类提供通用的功能，比如初始化 WebDriver、等待时间设置等。 ```python from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC class BasePage: def __init__(self, driver): self.driver = driver def find_element(self, locator, timeout=10): """查找单个元素""" return WebDriverWait(self.driver, timeout).until( EC.presence_of_element_located(locator) ) def find_elements(self, locator, timeout=10): """查找多个元素""" return WebDriverWait(self.driver, timeout).until( EC.presence_of_all_elements_located(locator) ) ``` --- #### 4. 编写具体的页面对象类 以下是两个示例页面对象类：`login_page.py` 和 `news_page.py`。 ##### （1）登录页面类 `login_page.py` 此页面对象负责处理与登录相关的所有操作。 ```python from selenium.webdriver.common.by import By from .base_page import BasePage class LoginPage(BasePage): username_locator = (By.ID, 'username') password_locator = (By.ID, 'password') submit_button_locator = (By.ID, 'submit') def enter_username(self, username): element = self.find_element(self.username_locator) element.clear() element.send_keys(username) def enter_password(self, password): element = self.find_element(self.password_locator) element.clear() element.send_keys(password) def click_submit(self): button = self.find_element(self.submit_button_locator) button.click() def is_login_successful(self): try: self.find_element((By.CLASS_NAME, 'welcome-message'), timeout=5) return True except Exception: return False ``` ##### （2）新闻页面类 `news_page.py` 此页面对象负责处理与新闻浏览相关的所有操作。 ```python from selenium.webdriver.common.by import By from .base_page import BasePage class NewsPage(BasePage): search_input_locator = (By.ID, 'search-input') search_button_locator = (By.ID, 'search-button') def enter_search_query(self, query): input_field = self.find_element(self.search_input_locator) input_field.clear() input_field.send_keys(query) def perform_search(self): button = self.find_element(self.search_button_locator) button.click() ``` --- #### 5. 编写测试脚本 测试脚本调用页面对象的方法完成特定的操作并验证结果。 ##### （1）登录测试脚本 `tests/test_login.py` ```python import pytest from selenium import webdriver from ..page_objects.login_page import LoginPage @pytest.fixture(scope="module") def setup(): driver = webdriver.Chrome() yield driver driver.quit() def test_valid_login(setup): driver = setup driver.get("http://example.com/login") login_page = LoginPage(driver) login_page.enter_username("admin") login_page.enter_password("password123") login_page.click_submit() assert login_page.is_login_successful(), "Login failed" ``` ##### （2）新闻测试脚本 `tests/test_news.py` ```python import pytest from selenium import webdriver from ..page_objects.news_page import NewsPage @pytest.fixture(scope="module") def setup(): driver = webdriver.Chrome() yield driver driver.quit() def test_search_news(setup): driver = setup driver.get("http://example.com/news") news_page = NewsPage(driver) news_page.enter_search_query("technology") news_page.perform_search() result_title = news_page.find_element((By.TAG_NAME, 'h1')).text assert "technology" in result_title.lower(), "Search results do not match the query" ``` --- #### 6. 配置 Pytest 在项目根目录创建 `pytest.ini` 文件，定义 HTML 报告和其他选项[^2]。 ```ini [pytest] addopts = --html=report.html --self-contained-html ``` --- #### 7. 运行测试 使用以下命令运行测试并生成报告： ```bash pytest ``` --- ### 总结 以上展示了如何基于 Python 和 Selenium 构建 POM 模式的自动化测试框架。核心在于分离页面对象和测试逻辑，从而提升代码的清晰度和可维护性[^1]^。