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

Question

如果你为Selenium编写测试来检查，例如无论是Firefox还是Chrome，测试看起来非常相似，除了setUp，参见例如https://gist.github.com/devinmancuso/54904c005f8d237f6fec，具有相同的功能test_search_in_python_chrome和test_search_in_python_firefox。存在避免硒中的代码重复的模式，例如the Page Object Pattern。

有没有办法避免这种巨大的代码重复？

Answer 1

您提供的链接中的setUp的内容是driver初始化。您可以在另一个类中处理此问题，并从属性文件中获取浏览器类型

def get_driver(self):
    browser = get_value_from_configurations_file()
    url = get_value_from_configurations_file()

    driver = None
    if browser == 'chrome':
         driver = webdriver.Chrome()
    elif browser == 'firefox':
         driver = webdriver.Firefox()

    driver.maximize_window()
    driver.get(url)

    return driver

我通常在驱动程序工厂和测试之间使用“中间人”来处理driver操作并获得PO起点

class WebApplication:

    def __init__(self):
        self.__driver = WebDriverFactory().get_driver()

    def get_driver(self):
        return self.__driver

    def get_home_page(self):
        return HomePage(self.__driver)        

并从基础测试类中使用它

@pytest.mark.usefixtures("run_for_test")
class AbstractTest(ABC):

    web_application = None

    @pytest.fixture()
    def run_for_test(self, request):
        AbstractTest.web_application = WebApplication()
        # set up for all tests
        yield
        # tear down for all tests

示例测试：

class TestExample(AbstractTest):

    def test_example(self):
        home_page = \
            (self.web_application
             .get_home_page())

这样您只需更改配置文件就可以编写一次测试并在不同的浏览器上运行它。

Answer 2

@盖伊的答案显示了方向。更简单的A solution是使用MixIns：

class DuckMixin(object):
    def testDuckLoads(self):
        self.browser.get("https://duckduckgo.com")
        self.assertIn("duckduckgo", self.browser.title.lower())

    def testDuckSafe(self):
        self.browser.get("https://duckduckgo.com")
        (self.browser
         .find_element_by_id("search_form_input_homepage")
         .send_keys("porn" + webdriver.common.keys.Keys.RETURN))
        # click to disable temporarily, then wait, then see if available
        self.browser.implicitly_wait(10)
        self.browser.find_element_by_class_name("js-safe-search-temp").click()
        # assert that browser.find_element_by_class_name("js-safe-search-temp") still exists, else throws exception
        self.browser.find_element_by_class_name("js-safe-search-temp")


class DuckDuckGoTestCaseFirefox(unittest.TestCase, DuckMixin):
    def setUp(self):
        profile = webdriver.FirefoxProfile()
        profile.add_extension(extension=os.path.join(DIR, "..", "addon"))
        self.browser = webdriver.Firefox(firefox_profile=profile)

    def tearDown(self):
        self.browser.close()

class DuckDuckGoTestCaseChromium(unittest.TestCase, DuckMixin):
    def setUp(self):
        profile = webdriver.chrome.options.Options()
        profile.add_extension(extension=os.path.join(DIR, "..", "safe.zip"))
        self.browser = webdriver.Chrome(chrome_options=profile)

    def tearDown(self):
        self.browser.close()