如何避免Firefox与Chrome的Selenium测试代码重复？

嘿，这个问题绝对是Selenium自动化测试里的高频痛点——写了Chrome版本的测试，Firefox版本几乎全是复制粘贴，冗余得让人头疼！别担心，有好几种成熟的方法能帮你彻底消灭这种重复代码，下面给你拆解几个最实用的：

这是最直接的解决方案，借助测试框架的参数化功能，把浏览器类型作为参数传入，让同一个测试逻辑自动适配不同浏览器。

比如用pytest的话，代码可以写成这样：

import pytest
from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from webdriver_manager.firefox import GeckoDriverManager

@pytest.mark.parametrize("browser", ["chrome", "firefox"])
def test_python_homepage_title(browser):
    # 根据参数初始化对应浏览器
    if browser == "chrome":
        driver = webdriver.Chrome(ChromeDriverManager().install())
    elif browser == "firefox":
        driver = webdriver.Firefox(executable_path=GeckoDriverManager().install())
    
    # 下面是完全复用的测试逻辑，不用改一行
    driver.get("https://www.python.org")
    assert "Python" in driver.title, "首页标题不对！"
    
    driver.quit()

运行的时候，pytest会自动帮你跑两次测试——一次Chrome，一次Firefox，你只需要维护这一段测试逻辑就行。如果要加Edge浏览器，只需要给参数列表加个"edge"，再补个初始化分支就好。

如果用unittest框架，也可以用parameterized库来实现同样的效果，原理是一致的。

如果你习惯用类式的测试结构，写一个BaseTest基类是个好主意，把setUp、tearDown和浏览器初始化逻辑都放在基类里，测试类只需要继承它，专注写测试逻辑就行。

比如：

import unittest
from selenium import webdriver
from webdriver_manager.chrome import ChromeDriverManager
from webdriver_manager.firefox import GeckoDriverManager

class BaseTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # 这里可以通过环境变量、配置文件或者命令行参数指定浏览器
        cls.browser = "chrome"  # 改成"firefox"就切换浏览器
        if cls.browser == "chrome":
            cls.driver = webdriver.Chrome(ChromeDriverManager().install())
        elif cls.browser == "firefox":
            cls.driver = webdriver.Firefox(executable_path=GeckoDriverManager().install())
    
    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

class TestPythonSearch(BaseTest):
    def test_homepage_load(self):
        self.driver.get("https://www.python.org")
        self.assertIn("Python", self.driver.title)
    
    def test_search_function(self):
        # 这里写搜索相关的测试逻辑，同样不用管浏览器类型
        self.driver.get("https://www.python.org")
        search_box = self.driver.find_element("name", "q")
        search_box.send_keys("selenium")
        search_box.submit()
        self.assertIn("Selenium", self.driver.title)

这样所有测试类都继承BaseTest，不用再重复写浏览器初始化的代码，要切换浏览器只需要改基类里的cls.browser，甚至可以改成从命令行接收参数，运行时指定：python -m unittest tests.py --browser firefox（需要自己处理命令行参数的解析）。

你提到的Page Object Pattern本身就是为了减少重复而生的，再结合上面的方法，能让代码结构更清晰，复用性拉满。

先把页面的元素和操作封装成Page Object：

class PythonHomePage:
    def __init__(self, driver):
        self.driver = driver
        self.url = "https://www.python.org"
        self.search_box_locator = ("name", "q")
    
    def load(self):
        self.driver.get(self.url)
    
    def search_for(self, keyword):
        search_box = self.driver.find_element(*self.search_box_locator)
        search_box.send_keys(keyword)
        search_box.submit()
    
    def has_correct_title(self, keyword):
        return f"{keyword}" in self.driver.title

然后测试类继承BaseTest，调用Page Object的方法：

class TestPythonSearch(BaseTest):
    def test_search_selenium(self):
        home_page = PythonHomePage(self.driver)
        home_page.load()
        home_page.search_for("selenium")
        self.assertTrue(home_page.has_correct_title("Selenium"))

这时候不管是Chrome还是Firefox，测试逻辑完全一致，页面操作也只维护一份，哪怕页面元素改了，只需要改Page Object里的定位器就行，所有测试都会自动更新。

上面的代码里用到了webdriver-manager库，它能自动下载对应版本的浏览器驱动，不用你手动去官网下载、配置路径，省了很多麻烦。安装也很简单：pip install webdriver-manager。

总结一下，最推荐的组合是参数化测试 + Page Object Pattern，既减少了重复代码，又让测试结构清晰，后期维护成本极低。

内容的提问来源于stack exchange，提问作者serv-inc