单元测试实例

百度的单元测试

from selenium import webdriver
import unittest
from HTMLTestRunner import HTMLTestRunner


class MyTest(unittest.TestCase):
    def setUp(self) -> None:
    	# 创建操作句柄
        self.driver = webdriver.Chrome(executable_path=r'E:\chromdriver\chromedriver.exe')
        self.base_url = 'https://www.baidu.com'
        # 浏览器窗口最大化
        self.driver.maximize_window()
        # 隐式等待
        self.driver.implicitly_wait(10)

    def test_baidu(self):
        driver = self.driver
        driver.get(self.base_url)
        driver.find_element_by_id('kw').send_keys('图片')
        driver.find_element_by_id('su').click()
        print(driver.title)
        self.assertEqual(driver.title,'百度一下，你就知道')

    def tearDown(self) -> None:
    	# 关闭浏览器
        self.driver.quit()

if __name__ == '__main__':
	# 创建测试集
    suite = unittest.TestSuite()
    suite.addTest(MyTest('test_baidu'))
    # 自定义报告存放路径
    file = open('./report.html','wb')
    # 生成报告
    runner = HTMLTestRunner(file,title='百度搜索报告',description='测试用例测试情况')
    runner.run(suite)
    # 关闭文件
    file.close()

结果
同时还会生成一个report.html文件。

百度一下，你就知道


Ran 1 test in 10.106s

OK