Python unittest 学习
Python unittest 学习
[[toc]]
前言
由于目前项目需要做接口测试，故特意去学习了Python中的unittest测试框架，以下是个人的一些学习经验，如有错误之处，请不吝指出。
一、什么是unittest
unittest翻译过来就是“单元测试”，单元测试负责对最小的软件设计单元（模块）进行验证。而引用python官方文档：
unittest是python自带的单元测试框架，有时候又被称为”PyUnit”，是 python版本的JUint实现。该框架的作者是 Kent Beck和Erich Gamma。
二、基本概念
在开始运用unittest框架进行测试前，我们需要先学习unittest框架中4个重要的概念：test fixture、test case、test suite、test runner。
官方文档：
1. test fixture ：表示执行一个或多个测试所需的准备, 以及任何关联的清理操作。例如, 这可能涉及创建临时或代理数据库、目录或启动服务器进程。
2. test case ：测试用例是最小的测试单元。它检查特定的输入集的响应。单元测试提供了一个基类测试用例, 可用于创建新的测试用例。
3. test suite ：测试套件是测试用例、测试套件或两者的集合。它用于聚合应一起执行的测试。
4. test runner ：测试运行程序是协调测试执行并向用户提供结果的组件。运行程序可以使用图形界面、文本界面或返回特殊值来指示执行测试的结果。
而我们可以这样简单的去理解这4个概念：
- test fixture:是初始化和清理测试数据及环境，通过覆盖TestCase的setUp（）和tearDown（）方法来实现
- test case:是测试用例
- test suite:是用例集合，即测试套件，通过addTest加载TestCase到TestSuite中，从而返回一个TestSuite实例。
- test runner:的作用是运行用例并返回结果，通过TextTestRunner类提供的run（）方法来执行test suite/test case。
三、基本用法
在理解了4个基本概念（test fixture、test case、test suite、test runner）之后，我们通过以下的“加法”例子学习：
首先建立一个加法类： calculator.py
# 计算器类
class Count:
def __init__(self, a, b):
self.a = int(a)
self.b = int(b)
# 计算加法
def add(self):
return self.a + self.b
接着写test.py去测试“加法类：calculator.py”:
from testpro.calculator import Count
import unittest
class TestAdd(unittest.TestCase):
def setUp(self):
print('test start')
def test_add(self):
j = Count(2, 3)
self.assertEqual(j.add(), 5, '计算错误！')
def tearDown(self):
print('test end')
if __name__ == '__main__':
# 构造测试集
suite = unittest.TestSuite()
suite.addTest(TestAdd('test_add'))
# 执行测试集合
runner = unittest.TextTestRunner()
runner.run(suite)
在 test.py 中，首先引入unittest模块，创建TestAdd类继承unittest的TestCase类。
setUp（）方法用于测试用例执行前的初始化工作，而这里用来打印“test start”信息。
tearDow（）方法与setUp（）相对应，用于测试用例之后的工作，这里打印“test end”信息。
asserEqual（）方法对add（）的返回值进行断言，判断两者是否相等，assertEqual（）方法由TestCase类继承而来。
TestSuite（）类来创建测试套件，通过它提供的addTest（）方法来添加测试用例test_add（）。
TextTestRunner（）类的run（）方法来运行suite所组装的测试用例。
四、执行用例的方法
目前学习中，发现有3种执行项目中测试用例的方法（ps：应该还有很多方法）：
- main（） ：unittest提供的全局方法，可以方便地将一个单元测试模块变成可以直接运行的测试脚本。main（）方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法，并自动执行它们。
- run（） ：是unittest的TextTestRunner（）类的方法，用来运行suite套件中的测试用例集。
- discover（） ：TestLoader类中提供的方法，用来自动识别项目中的测试用例
例子：
from testpro.calculator import Count
import unittest
class TestAdd(unittest.TestCase):
def setUp(self):
print('test start')
def test_add(self):
j = Count(2, 3)
self.assertEqual(j.add(), 5, '计算错误！')
def tearDown(self):
print('test end')
if __name__ == '__main__':
# main（）方法
unittest.main()
# run()方法
# 构造测试集
#suite = unittest.TestSuite()
# suite.addTest(TestAdd('test_add'))
# 执行测试集合
#runner = unittest.TextTestRunner()
#runner.run(suite)
# discover方法
#test_dir = './test_case'
#discover = unittest.defaultTestLoader.discover(test_dir, pattern='test*.py')
#if __name__ == '__main__':
# runner = unittest.TextTestRunner()
# runner.run(discover)
五、断言的方法
在执行用例的过程中，最终用例是否执行通过，是通过判断测试得到的实际结果与预期结果是否相等决定的。unittest框架的TestCase类提供下面这些方法用于测试结果的判断。
常用的：
- assertEqual(first, second, msg=None):断言第一个参数和第二个参数是否相等，如果不相等则测试失败。msg为可选参数，用于定义测试时打印想信息。
- assertTure(expr, msg=None):判断测试表达式是true或false
- assertIN（first, second, msg=None）:断言第一个参数是否在第二个参数中，反过来讲，第二个参数是否包含在第一个参数。
| 方法 | 检查点 |
|---|---|
| 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 |
六、用例失败如何重跑
目前在unittest中并没有可以用来失败用例重跑的方法或插件，只能自己去封装一个方法。但通过资料查找，发现python下另一个测试框架pytest中就有2个方法可以进行失败用例重跑的，分别是flaky和rerunfailures。目前没有去深入学习pytest框架，就不详写了。
七、知识点补充
1. 测试用例分类
通过前面对unittest单元测试框架的学习，我们来用它运行web自动化测试脚本。在 开始之前，需要规划好测试目录，因为一旦测试用例多起来之后，后期维护就很麻烦。所以用例需要按照所测试的功能进行拆分，分散到不同的测试文件中。但通过addTest（）添加/删除测试用例变得很麻烦，所以我们要用TestLoder类中提供的discover（）方法解决:
discover(start_dir, pattern='test*.py, top_level_dir=None)
- start_dir:要测试的模块名称或测试用例目录。
- pattern='test.py':表示用例文件名的匹配原则。以“test”开头的 ‘,py’类型的文件，星号“”表示任意多个字符。
- top_level_dir=None:测试模块的顶层目录，如果没有顶层目录，默认为None。
让unittest框架查找到子目录中的测试文件，需要在每个子目录下放一个“\__init__.py”文件，文件可以为空。
2.if _name_ == '_main_'语句说明
if代表此句子为条件语句。_name_作为模块的内置属性，即是.py文件的调用方式。如果_name_ 等于 '_main_'就表示可以直接调用。
3.用例执行的顺序
unittest框架默认根据ASCII码的顺序加载测试用例的，数字与字母的顺序为：“0-9”，“A-Z”，“a-z”。而想要用例按顺序执行，需要通过TestSuite类的addTest（）方法按照一定的顺序来加载。
4.跳过测试和预期失败
在运行测试时，有时需要直接跳过某些测试用例，或者当用例符合某个条件跳过测试，又或者直接将测试用例设置为失败。unittest提供了实现这些需求的装饰器
- unittest.skip(reason):无条件跳过装饰的测试，说明跳过的原因
- unittest.skipIf(condition, reason)：跳过装饰的测试，如果条件为真
- unittest.skipUnless(condition, reason):跳过装饰的测试，除非条件为真
- unittest.expectedFailure()：测试标记为失败。不管执行结果是否失败，同意标记失败。
