unittest框架的使用

先简单介绍一下unittest的核心组成部分：

• 测试夹具：Test Fixture
  • 一般用于执行测试用例的准备或者清理工作，比如测试开始前的数据准备或者测试结束的数据清理等。通过setUp()、tearDown()、setUpClass()、tearDownClass()这四个钩子函数实现了测试用例的准备和清理工作。
  • setUp()：在每一个测试用例执行前都会执行一次
  • tearDown()：在每一个测试用例执行后都会执行一次
  • setUpClass()：仅在整个测试类开始执行前执行，且必须使用@classmethod装饰
  • tearDownClass()：仅在整个测试类结束执行后执行，且必须使用@classmethod装饰
• 测试用例：Test Case
  • 是unittest的最小单元，测试用例会继承TestCase这个基类
• 测试用例集：Test Suite
  • 又称为测试套件，可以包含一个或者多个测试用例
  • 测试用例集包括一个或多个TestSuite类，每一个TestSuite类包括一个或多个TestCase类，也可以包括其他TestSuite类。TestSuite通过addTest()方法或者addTests()方法把一个个测试用例、测试用例集组装成一个新的测试用例集。
• 测试加载器：Test Loader
  • 用于从提供的类（class）和模块（module）中生成测试用例集，unittest默认提供一个测试加载器（Default Test Loader）
  • 测试加载器加载本地或从外部文件定义好的TestCase类或者TestSuite类
  • 默认加载以test开头的.py文件
• 测试运行器：Test Runner
  • 用于执行测试用例和输出测试结果

unittest还支持忽略执行某些测试用例，在要忽略的测试用例上添加如下装饰器即可：

• @unittest.skip():执行时直接忽略被装饰的测试用例
• @unittest.skipIf():如果skipIf里的条件成立，执行时直接忽略被装饰的测试用例
• @unittest.skipUnless():永久在执行时忽略被装饰的测试用例，除非skipUnless里的条件成立
• @unittest.expectedFailure:期望被装饰的测试用例是失败的，如果失败的，则此条测试用例被标记为测试通过

接下来我们对上一篇文章中的pageobject三层架构中，业务层（测试用例层）的代码使用unittest框架进行改造。

test_login.py文件的代码如下：

import time

from Module.login import Login
from selenium import webdriver
import unittest
import ddt


test_data = [
    {"username":"test123", "password":"123456"},
    {"username":"test124", "password":"123456"}
]

# 使用unittest框架
@ddt.ddt
class Login_test(unittest.TestCase):

    def setUp(self) -> None:
        self.driver = webdriver.Chrome()
        self.driver.get('https://mail.163.com')
        self.driver.maximize_window()

    def tearDown(self) -> None:
        time.sleep(5)
        self.driver.quit()

    @ddt.data(*test_data)
    def test_login(self, test_data):
        print(test_data)
        username = test_data['username']
        password = test_data['password']
        login = Login()
        # login.login(self.driver, 'test123', '123456')
        login.login(self.driver, username, password)


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

在项目目录下创建一个Reports文件夹，用来存放测试报告的结果。

执行入口run.py文件的修改如下：

import time
import unittest
from Setting.Config import Test_Data_Path, Report_Path, Test_Case_Path
from TestRunner import HTMLTestRunner
import TestRunner



if __name__ == '__main__':
    now = time.strftime('%Y-%m-%d %H:%M:%S')
    # pytest.main(['-s', '-v', Test_Data_Path])
    suit = unittest.TestSuite()
    loader = unittest.TestLoader()
    suit.addTests(loader.discover(start_dir=Test_Case_Path))
    
    # 生成测试报告，测试报告存放地址及测试报告的名称
    with open(Report_Path + '/' + now + 'report.html', 'wb') as fp:
        runner = HTMLTestRunner(
            stream=fp,
            title='test_unittest'
        )
        runner.run(suit)