对百度搜索的一次web自动化测试

代码：

import unittest
from time import sleep
from selenium import webdriver

class TestBaiDu(unittest.TestCase):
    @classmethod
    #用setUp，每执行一条测试用例就会启动和关闭一次浏览器，所以此处用setUpClass
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.base_url = "https://www.baidu.com/"

    # 模块化，将搜索封装成一个方法
    def baidu_search(self,search_key):
        self.driver.get(self.base_url)
        self.driver.find_element_by_id("kw").send_keys(search_key)
        self.driver.find_element_by_id("su").click()
        sleep(2)

    def test_search_key_selenium(self):
        search_key = "selenium"
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    def test_search_key_unittest(self):
        search_key = "unittest"
        self.baidu_search(search_key)
        self.assertEqual(self.driver.title,search_key+"_百度搜索")

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

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

结果：

Ran 2 tests in 17.891s

OK