用Python编写Selenium单元测试的正确方法?

时间:2013-12-26 12:18:14

标签: python unit-testing selenium python-unittest

我有以下情况:

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的情况下将任务划分为不同的函数?

1 个答案:

答案 0 :(得分:6)

单元测试在设计上彼此独立。排序可以改变,但它们不应该取决于彼此的结果。

在您的情况下,您需要将load_page_b_if_previous_function_is_true放在test_element_a_in_page_a内。

通常,您需要为每个Selenium方案编写一个测试函数。您无法将相互依赖的操作拆分为多个测试用例。