Unittest常用用法

1. 基本概念

在开始运用unittest框架进行测试前，我们需要了解unittest框架中4个重要的概念：test fixture、test case、test suite、test runner。
test fixture:是初始化和清理测试数据及环境，通过覆盖TestCase的setUp（）和tearDown（）方法来实现
test case:是测试用例
test suite:是用例集合，即测试套件，通过addTest加载TestCase到TestSuite中，从而返回一个TestSuite实例。
test runner:的作用是运行用例并返回结果，通过TextTestRunner类提供的run（）方法来执行test suite/test case。

2. 基本用法

我们通过以下的“加法”例子学习unittest的基本用法：
首先建立一个加法类： 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”信息。
tearDown（）方法与setUp（）相对应，用于测试用例之后的工作，这里打印“test end”信息。
asserEqual（）方法对add（）的返回值进行断言，判断两者是否相等，assertEqual（）方法由TestCase类继承而来。
TestSuite（）类来创建测试套件，通过它提供的addTest（）方法来添加测试用例test_add（）。
TextTestRunner（）类的run（）方法来运行suite所组装的测试用例

3. 执行用例的方法

有3种执行项目中测试用例的方法：
1、main（） ：unittest提供的全局方法，可以方便地将一个单元测试模块变成可以直接运行的测试脚本。 main（）方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法，并自动执行它们。
2、run（） ：是unittest的TextTestRunner（）类的方法，用来运行suite套件中的测试用例集。
3、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)

4. 用例执行的顺序

1、unittest框架默认根据ASCII码的顺序加载测试用例的，数字与字母的顺序为：“0-9”，“A-Z”，“a-z”。而想要用例按顺序执行，需要通过TestSuite类的addTest（）方法按照一定的顺序来加载。
2、通过前面对unittest单元测试框架的学习，我们来用它运行web自动化测试脚本。在开始之前，需要规划好测试目录，因为一旦测试用例多起来之后，后期维护就很麻烦。所以用例需要按照所测试的功能进行拆分，分散到不同的测试文件中。但通过addTest（）添加/删除测试用例变得很麻烦，所以我们要用TestLoder类中提供的discover（）方法解决:
discover(start_dir, pattern='test*.py, top_level_dir=None)
3、start_dir:要测试的模块名称或测试用例目录。
4、pattern=‘test.py’:表示用例文件名的匹配原则。以“test”开头的 ‘,py’类型的文件，星号“”表示任意多个字符。
5、top_level_dir=None:测试模块的顶层目录，如果没有顶层目录，默认为None
让unittest框架查找到子目录中的测试文件，需要在每个子目录下放一个“_init_.py”文件，文件可以为空。

5. 断言的方法

在执行用例的过程中，最终用例是否执行通过，是通过判断测试得到的实际结果与预期结果是否相等决定的。unittest框架的TestCase类提供下面这些方法用于测试结果的判断。
常用的：
1、assertEqual(first, second, msg=None):断言第一个参数和第二个参数是否相等，如果不相等则测试失败。msg为可选参数，用于定义测试失败时想打印的信息。
2、assertTure(expr, msg=None):判断测试表达式是true或false
3、assertIN（first, second, msg=None）:断言第一个参数是否在第二个参数中，反过来讲，第二个参数是否包含在第一个参数

6. 跳过测试和预期失败

在运行测试时，有时需要直接跳过某些测试用例，或者当用例符合某个条件跳过测试，又或者直接将测试用例设置为失败。unittest提供了实现这些需求的装饰器
1、unittest.skip(reason):无条件跳过测试的装饰器，说明跳过的原因
2、unittest.skipIf(condition, reason)：如果条件为真，跳过测试的装饰器
3、unittest.skipUnless(condition, reason):跳过测试的装饰器，除非条件为真
4、unittest.expectedFailure()：测试标记为失败。不管执行结果是否失败，统一标记失败。