时间：2020-09-27 python教程 查看: 837
UnitTest 框架是 Python 自带的一个作为单元测试的测试框，相当于 Java中的 JUnit，随着自动化技术的成熟，UnitTest 成为了测试框架第一选择，可以完整的结合 Selenium、Requests 来实现 Ul 和接口的自动化，由 UnitTest 再衍生出 PyTest，PyTest 可以完美结合 UnitTest 来实现自动化。
基本应用：
1、环境搭建，Python 中已经直接加载了 UnitTest 框架，无须额外安装
2、四大组件：
a. test fixture：setUp(前置条件）、tearDown（后置条件），用于初始化测试用例及清理和释放资源
b. test case：测试用例，通过集成 unttest.TestCase，来实现用例的继承，在 Unitfest 中，测试用例都是通过 test 来识别的，测试用例命名 test_XXX
c. test suite：测试套件，也称之为测试用例集
d. test runner：运行器，一般通过 runner 来调用 suite 去执行测试
UnitTest 运行机制：通过在 main 函数中，调用 unitest.main() 运行所有内容
本节知识：1. 对 UnitTest 有直观感受
讲了这么多，也完全不明白，没关系，通过实例先有个直观的了解，UnitTest 是个什么东西
import unittest
# 通过继承 unittest。TestCase 来实现用例
class forTest(unittest.TestCase):
# 类的初始化
@classmethod
def setUpClass(cls) -> None:
print('class')
# 类的释放
@classmethod
def tearDownClass(cls) -> None:
print('tclass')
# 测试用例初始化
def setUp(self) -> None:
print("setUp")
# 测试用例释放
def tearDown(self) -> None:
print("teadDown")
# 测试用例
def test_a(self):
print("a")
# 测试用例
def test_b(self):
print("b")
# 函数
def add(self, a, b):
return a + b
# 测试用例
def test_c(self):
c = self.add(1, 3)
print('c =', c)
if __name__ == "__main__":
unittest.main(verbosity=2)# 参数 verbosity=2 的目的是为了让打印的信息更加完整，也可以不要对上面的程序进行讲解：
1、类的初始化与释放
def setUpClass(cls) -> None 表示类的初始化，在执行测试用例之前执行，只执行一次，函数参数为 cls 表示这是一个类方法
def tearDownClass(cls) -> None 表示类的释放，在执行测试用例之后执行，只执行一次
2、测试用例的初始化与释放
def setUp(self) -> None 用于测试用例的初始化，在每个测试用例之前都会执行，有多少个测试用例，就会执行多少次
def tearDown(self) -> None 用于测试用例释放，在每个测试用例执行之后执行，有多少个测试用例，就会执行多少次
注意：方法 setUpClass，tearDownClass，setUp，def tearDown 的方法名是固定的，不能改动，不然框架无法识别
3、测试用例的定义
测试用例的命名规则为 test_xxx，这样测试用例就会自动执行
注意：只有测试用例才会被执行，不以test_xxx 命名的函数是方法，方法是不能被执行的
4、执行测试用例
通过在 main 函数中，调用 unitest.main() 运行所有内容
运行结果如下：
class
setUp
a
teadDown
setUp
b
teadDown
setUp
c = 4
teadDown
tclass相信有了上面的例子，已经对UnitTest 有了一个初步的印象。
下面我们进行一个实战操作
本节知识：1. 自动化测试减少冗余，便于维护，2. ddt数据驱动
1、自动化测试减少冗余，便于维护
有了类的初始化与释放，测试用例的初始化与释放，我们可以将多个测试用例中相同的代码提取出来，减少自动化测试冗余，这样便于维护
下面看这样一个例子，我们打开谷歌浏览器，输入百度网址并进行搜索，搜索后关闭浏览器
#coding=utf-8
import unittest
from selenium import webdriver
import time
class forTest(unittest.TestCase):
# 测试用例初始化
# 打开谷歌浏览器，并进入百度
def setUp(self) -> None:
self.driver = webdriver.Chrome()
self.driver.get('http://www.baidu.com')
# 测试用例释放
# 等待 3s，关闭浏览器
def tearDown(self) -> None:
time.sleep(3)
self.driver.quit()
# 输入‘战狼'，并点击搜索
def test_1(self):
pass
self.driver.find_element_by_id('kw').send_keys('战狼')
self.driver.find_element_by_id('su').click()
# 输入‘红海行动'，并点击搜索
def test_2(self):
pass
self.driver.find_element_by_id('kw').send_keys('红海行动')
self.driver.find_element_by_id('su').click()
if __name__ == '__main__':
unittest.main()上面的案例中，我们将打开谷歌浏览器，进入百度，放在 setUp 中，完成每个测试用例之前的初始化，浏览器的关闭放在tearDown 中，完成测试用例的释放
在实际测试中，单个测试是需要用多种不同的条件（测试数据）对其进行测试的。
ddt 中最基本的应用；在 class 前定义 @ddt，用于表示要使用 ddt 了，再基于实际的应用。选择对应的装饰器来使用即可，说白了，就是一个装饰器
直接看例子比较直观
#coding=utf-8
import unittest
from ddt import ddt
from ddt import data# 导入data
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
# data用于设定参数
@data('战狼', '哪吒', '流浪地球', '复仇者联盟')
def test_1(self, txt):
print(txt)
if __name__ == '__main__':
unittest.main(verbosity=2)运行结果：
========测试开始========
战狼
========测试结束========
========测试开始========
哪吒
========测试结束========
========测试开始========
流浪地球
========测试结束========
========测试开始========
复仇者联盟
========测试结束========
可以看到测试用例 def test_1(self, txt) 被执行了四次，data 用于设定参数，将参数依次放入测试用例中进行测试。
我们改变一下设定的参数，将 data 设定的参数改为 ((‘战狼', ‘哪吒'), (‘流浪地球', ‘复仇者联盟'))，再进行测试，如下所示
#coding=utf-8
import unittest
from ddt import ddt
from ddt import data
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
# data 用于设定参数，将包 ('战狼', '哪吒') 作为一个整体赋值给 txt
@data(('战狼', '哪吒'), ('流浪地球', '复仇者联盟'))
def test_1(self, txt):
print(txt)
if __name__ == '__main__':
unittest.main(verbosity=2)运行结果如下：
========测试开始========
('战狼', '哪吒')
========测试结束========
========测试开始========
('流浪地球', '复仇者联盟')
========测试结束========
可以看到，传入参数 ((‘战狼', ‘哪吒'), (‘流浪地球', ‘复仇者联盟')) 时，将包 (‘战狼', ‘哪吒') 和 (‘流浪地球', ‘复仇者联盟') 作为一个整体，传递给测试用例了，如果我们希望将包里面的数据解开，传递给测试用例不同的参数，就需要用到 unpack 进行解包。
加入解包后的代码如下所示：
#coding=utf-8
import unittest
from ddt import ddt
from ddt import data
from ddt import unpack # 导入unpack
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
@data(('战狼', '哪吒'), ('流浪地球', '复仇者联盟'))
# 解包，将 ('战狼', '哪吒') 解包，'战狼' 赋值给 txt1，'哪吒'赋值给 txt2
@unpack
def test_3(self, txt1, txt2):
print(txt1)
print(txt2)
if __name__ == '__main__':
unittest.main(verbosity=2)执行结果如下：
========测试开始========
战狼
哪吒
========测试结束========
========测试开始========
流浪地球
复仇者联盟
========测试结束========
可以看到，unpack 对每次传入的包进行解包，例如将 (‘战狼', ‘哪吒') 解包，‘战狼' 赋值给 txt1，'哪吒'赋值给 txt2
上面的例子中，我们将输入的参数直接固定了，其实也可以通过文件读取，读取结果决定
#coding=utf-8
import unittest
from ddt import ddt
from ddt import data
from ddt import unpack
def readFile():
params = []
file = open('ddt.txt', 'r', encoding = 'gbk')
for line in file.readlines():
params.append(line.strip('\n').split(','))
return params
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
# 从文件中读取
@data(*readFile())
@unpack
def test_1(self, txt1, txt2):
print(txt1)
print(txt2)
if __name__ == '__main__':
unittest.main(verbosity=2)ddt.txt 文件中的内如下：
战狼,哪吒
流浪地球,复仇者联盟
运行结果：
函数 readFile 从文件中读取数据，unpack 进行解包
========测试开始========
战狼
哪吒
========测试结束========
========测试开始========
流浪地球
复仇者联盟
========测试结束========
上面从文件中读取数据时先读取文件，再处理读取的结果，下面介绍一个直接操作文件的方法
从 ddt 中导入 file_data，导入 yaml，读取数据的文件类型必须为 .yml 类型的文件。
#coding=utf-8
import unittest
from ddt import ddt
from ddt import data
from ddt import unpack
from ddt import file_data
import yaml
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
# 直接的文件读取，直接操作一个文件
@file_data('ddt2.yml')
def test_5(self, txt):
print(txt)
if __name__ == '__main__':
unittest.main(verbosity=2)ddt2.yml 文件内容如下：
name: 'skx'
info: 'hust'运行结果：
========测试开始========
skx
========测试结束========
========测试开始========
hust
========测试结束========
打开浏览器进入百度查询的例子中我们发现除了输入的参数不同，test_1(self) 和 test_2(self) 完全相同，这里我们就要通过 data 设定参数实现在一个测试用例中输入不同的参数
#coding=utf-8
import unittest
from selenium import webdriver
import time
from ddt import ddt
from ddt import data
# 在 class 前定义 @ddt，用于表示要使用 ddt 了
@ddt
class forTestTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.get('http://www.baidu.com')
def tearDown(self):
time.sleep(3)
self.driver.quit()
# data 用于设定参数
@data('战狼', '哪吒', '流浪地球')
def test_1(self, txt):
self.driver.find_element_by_id('kw').send_keys(txt)
self.driver.find_element_by_id('su').click()
if __name__ == "__main__":
unittest.main()运行结果，谷歌浏览器被打开三次，进入百度，分别输入 ‘战狼', ‘哪吒', ‘流浪地球'，每次浏览器关闭之后，才打开下一次，再进行搜索
上面的例子中，我们将输入的参数直接固定了，其实也可以通过文件读取，决定进入哪一个 url 和输入的参数
#coding=utf-8
import unittest
from selenium import webdriver
import time
from ddt import ddt
from ddt import data
from ddt import unpack
def readFile():
params = []
file = open('forTest3.txt', 'r', encoding = 'gbk')
for line in file.readlines():
params.append(line.strip('\n').split(','))
return params
@ddt
class forTestTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
def tearDown(self):
time.sleep(3)
self.driver.quit()
# data 用于设定参数，表示参数由 readFile() 函数的返回值决定
# unpack 用于解析参数，例如将['http://www.baidu.com', '战狼'] 分别 赋值给 url 和 txt
@data(*readFile())
@unpack
def test_1(self, url, txt):
self.driver.get(url)
self.driver.find_element_by_id('kw').send_keys(txt)
self.driver.find_element_by_id('su').click()
if __name__ == "__main__":
unittest.main()forTest3.txt 文件中的内容如下：
http://www.baidu.com,战狼
http://www.baidu.com,哪吒分析：
运行结果，谷歌浏览器被打开两次，进入百度，分别输入 ‘战狼', ‘哪吒'，每次浏览器关闭之后，才打开下一次，再进行搜索
file_data 是 ddt 中用于读取 yml 文件的装饰器
这个插入一个小插曲，上面提到了 yml 文件，这里就简单讲解一下 yml 文件怎么使用。
从yml 文件中直接读取数据可以生成字典，列表等，yml 文件由一定的格式，我们通过实例来说明，yml_test.py 从 a.yml 文件中读取文件并打印出来。
yml_test.py
import yaml
file = open('a.yml', encoding = 'utf-8')
res = yaml.load(file, Loader = yaml.FullLoader)
print(res)a.yml 文件中的内容如下所示，冒号代表字典，字典结构可以嵌套，也可以生成列表，具体格式参考下面的 a.yml 文件。
a.yml
name: 'skx'
age: 18
data:
a: 1
b: 2
c: 3
d: 4
list:
- a
- b
- c
- d打印的结果如下所示，生成四个字典元素，第三个字典元素为嵌套字典结构，第四个字典对应的 value 为列表。
{'name': 'skx', 'age': 18, 'data': {'a': 1, 'b': 2, 'c': 3, 'd': 4}, 'list': ['a', 'b', 'c', 'd']}
如果将 a.yml 文件中的数据改为如下结构，则生成一个纯列表，打印的结果如下所示。
a.yml
- a
- b
- c
- d['a', 'b', 'c', 'd']有了 yml 文件，我们就可以将测试数据放到 yml 文件中，从文件中获取参数，传入测试函数，完成测试。还是通过例子来讲解，yml_test2.yml 中是一个列表，每个列表元素是一个字典，字典中有两个元素，name 和 age，使用 file_data 直接可以将 yml_test2.yml 传入测试用例中。
read_yml2.py
#coding=utf-8
import unittest
from ddt import ddt
from ddt import file_data
import yaml
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
@file_data('read_yml2_data.yml')
def test_yam(self, **kwargs):
# 获取参数中key 为 name 的 value
print(kwargs["name"])
# 获取为 text 的 value
print(kwargs["age"])
if __name__ == '__main__':
unittest.main(verbosity=2)read_yml2_data.yml
-
name: 'Tom'
age: 13
-
name: 'Carl'
age: 20
-
name: 'Edward'
age: 28运行结果：
========测试开始========
Tom
13
========测试结束========
========测试开始========
Carl
20
========测试结束========
========测试开始========
Edward
28
========测试结束========
在 UnitTest中，TestCase 已经提供有封装好的断言方法进行断言校验。
断言：用于校验实际结果与预期结果是否匹型，在断言的内容选择上，是有要求的。
断言强调的是对于整个测试流程的结果进行判断，所以断言的内容是极为核心的。
上面的代码
#coding=utf-8
import unittest
from ddt import ddt
from ddt import file_data
import yaml
# 类之前定义装饰器，表示在类中要使用ddt了
@ddt
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
print('{:=^20}'.format("测试开始"))
def tearDown(self) -> None:
print("{:=^20}".format('测试结束'))
@file_data('read_yml2_data.yml')
def test_yam(self, **kwargs):
# 获取参数中key 为 name 的 value
name = kwargs['name']
print(name)
# 这里做断言，当断言不相等的时候返回 msg
self.assertEqual(name, 'Tom', msg = 'NotEqual')
# 获取为 text 的 value
print(kwargs["age"])
if __name__ == '__main__':
unittest.main()-
name: 'Tom'
age: 13
-
name: 'Carl'
age: 20
-
name: 'Edward'
age: 28可以看到第一个例子执行正确，后面的例子，执行结果和预期不一致，返回 NotEqual，左侧的日志可以看到第一个用例执行成功，后面两个例子执行失败。
unittest 框架的 TestCase 类提供以下方法用于测试结果的判断
| 方法 | 检查 |
|---|---|
| assertEqual(a, b) | a ==b |
| assertNotEqual(a, b) | a !=b |
| assertTrue(x) | bool(x) is True |
| assertFalse(x) | Bool(x) is False |
| assertIs(a, b) | a is b |
| assertIsNot(a, b) | a is not b |
| assertIsNone(x) | x is None |
| assertIsNotNone(x) | x is not None |
| assertIn(a, b) | a in b |
| assertNotIn(a, b) | a not in b |
| assertIsInstance(a, b) | isinstance(a,b) |
| assertNotIsInstance(a, b) | not isinstance(a,b) |
假设我们有很多测试用例，有些我们需要执行，有些我们不想执行，不想执行的测试用例如何才能不执行呢，这就需要用到 skip。
Skip用法：
还是通过案例进行讲解，下面有 6 个测试用例，2-5测试用例被屏蔽了，使用的方法不同，
直接看例子更加直观
skip_t.py
#coding=utf-8
import unittest
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
pass
def tearDown(self) -> None:
pass
def test_1(self):
print('1')
# 无条件跳过该条用例
@unittest.skip("不想运行")
def test_2(self):
print('2')
# 有条件跳过操作，条件为True跳过
@unittest.skipIf(1 < 2, '1 < 2 为True，条件成立，跳过执行')
def test_3(self):
print('3')
# 有条件执行跳过，条件为False跳过
@unittest.skipUnless(1 > 2, 'Unless的理由，条件为False，跳过执行')
def test_4(self):
print('4')
# 如果用例执行失败，则不计入失败的case数中
@unittest.expectedFailure
def test_5(self):
print('5')
self.assertEqual(4, 3, msg = 'NotEqual')
def test_6(self):
print('6')
if __name__ == "__main__":
unittest.main()执行结果如下，可以看到，test_2，test_3，test_4 跳过，test_5执行失败，但是不计入case数中
测试套件 Suite 作用：
通过例子讲解最容易理解，看一个最简单的例子，下面的代码中有五个测试用例，程序运行的结果和测试用例在代码中位置是没有关系的，结果永远打印 1 2 3 4 5，这是因为测试用例的执行顺序默认是按照字典顺序执行的，如何才能控制测试用例的执行顺序呢，这就需要使用测试套件了。
suite_case.py
import unittest
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
pass
def tearDown(self) -> None:
pass
def test_2(self):
print("2")
def test_1(self):
print("1")
def test_4(self):
print("4")
def test_3(self):
print("3")
def test_5(self):
print("5")
if __name__ == "__main__":
unittest.main()运行结果：
1
2
3
4
5
再建一个py 文件
#coding=utf-8
import unittest
from suite_case import *
# 创建一个测试套件 list
suite = unittest.TestSuite()
# 方法一，添加测试用例（子元素）到测试套件（集合）
suite.addTest(MyTestCase('test_3'))
suite.addTest(MyTestCase("test_1"))
suite.addTest(MyTestCase("test_5"))
# 套件通过TextTestRunner对象运行，功能相当于unittest.main()
runner = unittest.TextTestRunner()
runner.run(suite)我们首先创建一个测试套件，然后向测试套件中添加测试用例，最后创建 TextTestRunner 对象，调用 run 函数运行测试用例。这样我们不仅可以控制测试用例的执行顺序，还可以控制运行哪个测试用例。
结果如下：
3
1
5
上面的方法每次添加测试用例都需要调用 addTest 函数，能不能一次添加多个测试用例呢，可以的，将测试用例写成一个列表，通过addTests 函数可以一次添加多个测试用例
#coding=utf-8
import unittest
from suite_case import *
# 创建一个测试套件 list
suite = unittest.TestSuite()
# 方法二，批量添加测试用例
cases = [MyTestCase('test_3'), MyTestCase('test_1'), MyTestCase('test_5')]
suite.addTests(cases)
# 套件通过TextTestRunner对象运行，功能相当于unittest.main()
runner = unittest.TextTestRunner()
runner.run(suite)如果测试用例非常多，或者有多个文件中的测试用例都需要测试，这样添加也不是很方便，我们好可以按照文件路径，将该路径下需要测试的文件添加进测试套件中
#coding=utf-8
import unittest
from suite_case import *
# 创建一个测试套件 list
suite = unittest.TestSuite()
# 方法三，批量运行多个unittest类
test_dir = './'
# start_dir 参数指定文件路径，pattern 执行规则，'s*.py' 表示以 "s" 开头，".py" 的都加入测试套件中
discover = unittest.defaultTestLoader.discover(start_dir = test_dir, pattern = 's*.py')
runner = unittest.TextTestRunner()
runner.run(discover)# 通过 run 函数运行测试用例还可以执行类的名字，执行该类下面所有的测试用例，使用 loadTestsFromName 函数或者 loadTestsFromTestCase 都可以，案例如下：
#coding=utf-8
import unittest
from suite_case import *
# 创建一个测试套件 list
suite = unittest.TestSuite()
# 方法四，给出文件名和类名，就能测试所有的测试用例
suite.addTests(unittest.TestLoader().loadTestsFromName('suite.MyTestCase'))
# 套件通过TextTestRunner对象运行，功能相当于unittest.main()
runner = unittest.TextTestRunner()
runner.run(suite)#coding=utf-8
import unittest
from suite_case import *
# 创建一个测试套件 list
suite = unittest.TestSuite()
# 方法五，给出类名，就能测试所有的测试用例
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))
# 套件通过TextTestRunner对象运行，功能相当于unittest.main()
runner = unittest.TextTestRunner()
runner.run(suite)通过 HTMLTestRunner 我们可以将测试结果生成 html 文件，通过网页端进行查看。步骤如下：
下载 HTMLTestRunner.py 文件，下载地址
点进入HTMLTestRunner.py，右键另存为就可以下载到本地。
下载后，把HTMLTestRunner.py 文件复制到 Python 安装路径下的 lib 文件夹中（我的安装路径是：C:\Users\Administrator\AppData\Local\Programs\Python\Python38\Lib）。在python3中用HTMLTestRunner.py 报 importError“：No module named 'StringIO'解决办法，原因是官网的是python2语法写的，看官手动把官网的 HTMLTestRunner.py 改成 python3 的语法。
修改内容：
from HTMLTestRunner import HTMLTestRunner
下面就通过案例进行演示
suite_case.py 文件
import unittest
class MyTestCase(unittest.TestCase):
def setUp(self) -> None:
pass
def tearDown(self) -> None:
pass
def test_2(self):
print("2")
def test_1(self):
print("1")
def test_4(self):
print("4")
def test_3(self):
print("3")
def test_5(self):
print("5")
if __name__ == "__main__":
unittest.main()#coding=utf-8
import unittest
from suite_case import MyTestCase
from HTMLTestRunner import HTMLTestRunner
import os
suite = unittest.TestSuite()
report_path = './report/'
report_file = report_path + 'report.html'
# 路径不存在就创建一个文件夹
if not os.path.exists(report_path):
os.mkdir(report_path)
else:
pass
report_name = '测试报告名称'
report_title = '测试报告标题'
report_desc = '测试报告描述'
with open(report_file, 'wb') as report:
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(MyTestCase))
# 套件结合 TextTestRunner 对象进行运行，相当于 unittest.mian()
# 如果结合 HTMLTestRunner 使用，则需要调用 HTMLTestRunner 中的运行器
runner = HTMLTestRunner(stream = report, title = report_title, description = report_desc)
runner.run(suite)运行 就会成成 report.html 文件，浏览器打开该文件，如下所示：
这样就生成一个比较直观的测试报告
总结
到此这篇关于Python自动化之UnitTest框架实战的文章就介绍到这了,更多相关Python自动化UnitTest框架实战内容请搜索python博客以前的文章或继续浏览下面的相关文章希望大家以后多多支持python博客！