在为Firefox和Chrome编写selenium测试时避免代码重复

在为Firefox和Chrome编写Selenium测试时，避免代码重复的关键在于使用设计模式和工具来提高代码的可重用性和可维护性。以下是一些基础概念和相关策略：

基础概念

Selenium WebDriver: 一个用于自动化浏览器操作的工具。
Page Object Model (POM): 一种设计模式，将页面元素和操作分离，使得测试代码更简洁、易于维护。
Driver Factory: 一个工厂类，用于根据不同的浏览器类型创建相应的WebDriver实例。

类型与应用场景

Page Object Model (POM): 适用于复杂的Web应用，特别是当页面元素经常变化时。
Driver Factory: 适用于需要在不同浏览器上运行相同测试的场景。

示例代码

1. Page Object Model (POM)

创建一个基础页面类和具体的页面类：

from selenium.webdriver.common.by import By

class BasePage:
    def __init__(self, driver):
        self.driver = driver

class HomePage(BasePage):
    def __init__(self, driver):
        super().__init__(driver)
        self.url = "http://example.com"
        self.search_box = (By.ID, "search-box")
        self.search_button = (By.ID, "search-button")

    def open(self):
        self.driver.get(self.url)

    def search(self, query):
        self.driver.find_element(*self.search_box).send_keys(query)
        self.driver.find_element(*self.search_button).click()

2. Driver Factory

创建一个工厂类来管理不同浏览器的WebDriver实例：

from selenium import webdriver

class WebDriverFactory:
    def get_driver(self, browser_name):
        if browser_name.lower() == "chrome":
            return webdriver.Chrome()
        elif browser_name.lower() == "firefox":
            return webdriver.Firefox()
        else:
            raise ValueError(f"Unsupported browser: {browser_name}")

3. 测试用例

使用上述组件编写测试用例：

import unittest
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

class TestHomePage(unittest.TestCase):
    def setUp(self):
        self.driver_factory = WebDriverFactory()
        self.driver = self.driver_factory.get_driver("chrome")  # 或 "firefox"
        self.home_page = HomePage(self.driver)

    def test_search(self):
        self.home_page.open()
        self.home_page.search("Selenium")
        # 添加断言来验证搜索结果

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

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