单元测试框架

1、简介

python内置的Unittest单元测试框架，不需要额外进行安装，主要有以下四个核心概念：

2、简单示例

通过一个简单的示例演示unittest框架的使用，首先引入依赖库，接着定义测试用例类，然后将每条测试用例内公用的前置代码与后置代码分别添加至setUpClass(cls)与tearDownClass(cls)函数中，最后定义测试用例。

import unittest
import time
from selenium import webdriver
from selenium.webdriver.common.by import By
#测试模块的前置操作
def setUpModule():
    print("setUpModule正在执行")
        
#测试模块的后置操作
def tearDownModule():
    print("tearDownModule正在执行")
class TestBaidu(unittest.TestCase):
    #测试类内需要进行的前置操作
    @classmethod 
    def setUpClass(cls) -> None:
        print("setUpClass正在执行")
        cls.driver = webdriver.Edge()
        cls.driver.maximize_window() #窗口最大化
        cls.driver.get("https://www.baidu.com")
        time.sleep(1)
    @classmethod
    #测试类内需要进行的前置操作
    def tearDownClass(cls) -> None:
        print("tearDownClass正在执行")
        time.sleep(3)
        #退出浏览器
        cls.driver.quit()

    #每条测试用例的前置操作
    def setUp(self) -> None:
        print("setup正在执行")
        
    #每条测试用例的后置操作
    def tearDown(self) -> None:
        print("tearDown正在执行")

    #点击新闻
    def test_01_news(self):
        self.driver.find_element(By.XPATH,"//a[text()='新闻']").click()
        all_handles = self.driver.window_handles # 获取所有标签句柄
        # 将driver操作对象切换至新的标签页
        self.driver.switch_to.window(all_handles[-1])
        time.sleep(1)
        #关闭新闻标签页
        self.driver.close()
        self.driver.switch_to.window(all_handles[0])
        time.sleep(1)
    #点击图片
    def test_02_picture(self):
        self.driver.find_element(By.XPATH,"//a[text()='图片']").click()
        all_handles = self.driver.window_handles # 获取所有标签句柄
        # 将driver操作对象切换至新的标签页
        self.driver.switch_to.window(all_handles[-1])
        time.sleep(1)
        #关闭图片标签页
        self.driver.close()
        self.driver.switch_to.window(all_handles[0])
        time.sleep(1)
if __name__ == '__main__':
    unittest.main()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

值得注意的是，在上述示例中一共用到了三种夹具，分别为模块级别的夹具（setUpModule()与tearDownModule()）、类级别的夹具（setUpClass(cls)与tearDownClass(cls)）以及实例级别的夹具（setUp(self)与tearDown(self)），代码输出如下所示：

从上图中可以看出，模块级别的夹具会贯穿整个测试模块的始末，一个测试模块无论有多少测试类，都只会运行一次模块级别的夹具，类级别的夹具贯穿测试类运行的始末，一个测试类无论有多少测试用例都只会运行一次类级别的夹具，对于实例级别的夹具则会在每次实例运行时运行。

3、增加断言

增加断言的目的是确保接下来的操作符合我们的预期，断言成功程序会继续执行，否则会抛出AssertionError，程序中断。所有的unittest断言都有msg参数，用于输出断言失败后的提示信息。

    def test_01_news(self):
        self.driver.find_element(By.XPATH,"//a[text()='新闻']").click()
        all_handles = self.driver.window_handles # 获取所有标签句柄
        # 将driver操作对象切换至新的标签页
        self.driver.switch_to.window(all_handles[-1])
        self.assertIn('新闻',self.driver.title)
        time.sleep(1)
        #关闭新闻标签页
        self.driver.close()
1
2
3
4
5
6
7
8
9