《Python Web开发测试驱动方法》第2章“unittest模块扩展功能测试”

打开《functional_tests.py》文件

from selenium import webdriver
import unittest

class NewVisitorTest(unittest.TestCase):
    def setUp(self):
        # 给出webdriver的路径
        drive_path = 'C:\Program Files (x86)\Google\Chrome\Application\chromedriver\chromedriver.exe'
        # 这是为了忽略浏览器提示“正受到自动化软件的控制”
        option = webdriver.ChromeOptions()
        option.add_argument('disable-infobars')
        self.browser = webdriver.Chrome(executable_path=drive_path, chrome_options=option)
        #隐式等待，在页面中查找内容时让selenium等待，让内容出现
        self.browser.implicitly_wait(3)
    def tearDown(self):
        self.browser.quit()
    def test_can_start_a_list_and_retrieve_it_later(self):
        self.browser.get('http://localhost:8000')
        self.assertIn('To-Do', self.browser.title)
        self.fail('finish the test!')
if __name__ == '__main__':
    unittest.main(warnings='ignore')

上述代码：setUp和tearDown是特殊方法，分别在测试开始前和结束后运行，用来打开浏览器和关闭浏览器。测试出错tearDown任会运行，setUp出错tearDown不会运行。

编写了一个测试方法“test_can_start_a_list_and_retrieve_it_later”，判断“To-Do”是否在浏览器标题中。不管怎样self.fail都会失败，用来提示测试结束。