python - 在 Python 中编写 Selenium 单元测试的正确方法？

Question

我有以下场景:

Load Page A
Check if Element_A exists in Page A
Click Button_A
On Clicking Button_A, Page B is loaded
Check if Element_B exists in Page B and Click Button_B
and so on...

我的代码是这样写的:

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

class MyWebsiteTest(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()

    def test_element_a_in_page_a(self):
        driver = self.driver
        driver.get(PAGE_A)
        element = driver.get_element_by_id("element_a")
        self.assertIsNotNone(element,"Oops.")

    def load_page_b_if_previous_function_is_true(self):
        #WHAT DO I DO HERE?
        #I Need to run this function given test_element_a_in_page_a succeeds.

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

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

我该怎么做？ 如何编写 load_page_b_if_element_a_in_page_a？

如果我将整个测试流程写在一个函数中，那么我可以很容易地实现它，但是我该如何分离关注点呢？如何在不多次打开 firefox 的情况下将任务分成不同的功能？

Answer 1

单元测试在设计上是相互独立的。顺序可以更改，但它们不应依赖于彼此的结果。

在您的情况下，您需要将 load_page_b_if_previous_function_is_true 放在 test_element_a_in_page_a 中。

通常，您需要为每个 Selenium 场景编写一个测试函数。您不能将相互依赖的操作拆分为多个测试用例。