Python接口自动化之unittest单元测试

一、单元测试三连问

（一）什么是单元测试？

• 将访问接口的过程封装在函数里面；
• 接口测试就变成了单元测试；
• 单元测试就是通过传参，对某个模块、某个类、某个函数进行结果输出后验证的测试。

（二）为什么要做单元测试？

1. 单元测试之后，才是集成测试，单个的功能模块测试通过之后，才能把单个功能模块集成起来做集成测试，为了从底层发现bug，减少合成后出现的问题。
2. 越早发现bug越好，否则问题累计到后期，如果做错了就要推倒重来，对于时间和人力成本来说非常耗费精力。

对于我们测试来说：单元测试是为了执行测试用例。

（三）怎么做单元测试？

1. Unittest（Python自带）；
2. Pytest（下载安装）；

二、unittest模块说明

1.unittest简介

unittest框架的特性：

• 提供用例组织与执行：当测试用例只有几条的时候可以不考虑用例的组织，但是当测试用例数量较多时，此时就需要考虑用例的规范与组织问题。unittest单元测试框架就是用来解决这个问题的。
• 提供丰富的断言方法：既然是测试，就有一个预期结果和实际结果的比较问题。比较就是通过断言来实现，unittest单元测试框架提供了丰富的断言方法，通过捕获返回值，并且与预期值进行比较，从而得出测试通过与否。
• 提供丰富的日志：每一个失败用例我们都希望知道失败的原因，所有用例执行结束我们有希望知道整体执行情况，比如总体执行时间，失败用例数，成功用例数。unittest单元测试框架为我们提供了这些数据。

2.unittest组成

• TestCase（测试用例）：一个TestCase的实例就是一个测试用例。什么是测试用例呢？就是一个完整的测试流程。包括测试前准备环境的搭建(setUp)、实现测试过程的代码(run)，以及测试后环境的还原(tearDown)。单元测试(Unittest)的本质也就在这里，一个测试用例就是一个完整的测试单元，通过运行这个测试单元，可以对某一个功能进行验证。
• TestSuite（测试套件）：一个功能的验证往往需要多个测试用例，可以把多个测试用例集合在一起执行，这就产生了测试套件TestSuite的概念。TestSuite用来组装单个测试用例。可以通过addTest加载TestCase到TestSuite中，从而返回一个TestSuite实例。而且TestSuite也可以嵌套TestSuite。
• TestLoader（测试用例加载器）：用来加载TestCase到TestSuite中的，其中loadTestsFrom__()方法用于寻找TestCase，并创建它们的实例，然后添加到TestSuite中，返回TestSuite实例；
• TextTestRunner（执行测试用例）：用来执行测试用例，其中run(test)会执行TestSuite/TestCase中的run(result)方法，并将测试结果保存到TextTestResult实例中，包括运行了多少测试用例，成功多少，失败多少等信息；
• Test Fixture（测试环境数据准备和清理）：一个测试用例的初始化准备及环境还原，主要是setUp() 和 tearDown()方法；比如说在测试用例中需要访问数据库，那么可以在setUp()中建立数据库连接以及进行一些初始化，在tearDown()中清除在数据库中产生的数据，然后关闭连接。注意tearDown的过程很重要，要为以后的TestCase留下一个干净的环境。

3.unittest核心工作原理

三、unittest单元测试

1.实现思路

2.使用介绍

1. 要想使用unittest单元测试框架，必须得先导入：import unittest
2. 查看unittest源码；

import unittest
print(help(unittest))

import unittest
class IntegerArithmeticTestCase(unittest.TestCase):
    def testAdd(self):  # test method names begin with 'test'
        self.assertEqual((1 + 2), 3)
        self.assertEqual(0 + 1, 1)
    def testMultiply(self):
        self.assertEqual((0 * 10), 0)
        self.assertEqual((5 * 8), 40)
if __name__ == '__main__':
    unittest.main()

四、unittest实例

1.TestCase(测试用例)

import unittest
class TestDemo(unittest.TestCase):
    # test_sub用例
    def test_sub(self):
        self.assertEqual(2-1,1)
    # test_add用例
    def test_add(self):
        self.assertEqual(2+1,3)
if __name__ == "__main__":
    # unittest.main()是运行主函数
    unittest.main(verbosity=2)

test_add (__main__.TestDemo) ... ok
test_sub (__main__.TestDemo) ... ok
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK

• 使用unittest前，需导入unittest框架。
• TestDemo这个类必须继承unittest.TestCase，TestCase类，所有测试用例类继承的基类。
• 类内的方法必须以test开头，比如test_add。
• 断言：assertEqual用来断言预期结果和实际结果是否一致。当然unittest还包含很多其他断言方法，后面统一介绍。
• 用例执行顺序。在代码中test_sub方法写在test_add前，但实际，test_add比test_sub先运行。为什么呢？unittest执行测试用例，默认是根据ASCII码的顺序加载测试用例，数字与字母的顺序为：0-9，A-Z，a-z。
• verbosity是一个选项,表示测试结果的信息复杂度，有0、1、2 三个值。verbosity=0 : 你只能获得测试用例数总的结果；verbosity=1 (默认模式): 在每个成功的用例前面有个“.” 每个失败的用例前面有个 “F”；verbosity=2 (详细模式):测试结果会显示每个测试用例的所有相关的信息。

import unittest
#测试MathMethod类
class TestMathMethod(unittest.TestCase):
    #编写测试用例
    def test_add_two_positive(self): #测试两个正数相加
        res=MathMethod(1,1).add()
        print("1+1的结果是：",res)
        self.assertEqual(2,res,"两个正数相加出错！") #断言
    def test_add_two_zero(self): #测试两个0相加
        res = MathMethod(0, 0).add()
        print("0+0的结果是：", res)
        self.assertEqual(0, res, "两个0相加出错！") #断言
    def test_add_two_negative(self): #测试两个负数相加
        res = MathMethod(-1, -1).add()
        print("-1+(-1)的结果是：", res)
        self.assertEqual(-2, res, "两个负数相加出错！") #断言
if __name__ == '__main__':
    unittest.main()

class TestMathMethod(unittest.TestCase):
def setUp(self):
    print("开始准备执行测试用例！")
    
def tearDown(self):
    print("结束！")
    
#编写测试用例
def test_add_two_positive(self): #测试两个正数相加
    res=MathMethod(1,1).add()
    print("1+1的结果是：",res)

• setUp函数：初始化环境（执行每条用例之前，都要执行setUp函数下面的代码，每次都要执行）;
• tearDown函数：清洗环境（执行每条用例之后，都要执行tearDown函数下面的代码，每次都要执行）;
• setUp()、tearDown()是TestCase里的方法，写在测试类中，就是方法的重写。
• 执行顺序是：setUp->testA->tearDown->setUp->testB>tearDown

2.TestSuit(测试集)

import unittest
from xxx import xxx      #测试用例的类
suite=unittest.TestSuit()
suite.addTest(测试用例的类("用例名称1"))    #用例名称用字符串的形式传入
suite.addTest(测试用例的类("用例名称2"))
suite.addTest(测试用例的类("用例名称3"))
.....

3.TestLoader(加载测试用例)

• 方式一：通过测试类来加载用例（loadTestsFromTestCase）

import unittest
    from xxx import xxx      #测试用例的类
    suite=unittest.TestSuit()
    loader=unittest.TestLoader()
    suite.addTest(loader.loadTestsFromTestCase(测试用例类名1))  
    #测试用例类名直接传入

• 方式二：通过测试类所在的模块加载用例（loadTestsFromModule）

import unittest
from xxx import xxx      #测试用例模块
suite=unittest.TestSuit()
loader=unittest.TestLoader()
suite.addTest(loader.loadTestsFromTestCase(测试用例模块名))   #测试用例模块名直接传入

4.生成测试报告

方式一：使用unittest自带的TextTestRunner生成测试报告（文本格式，不推荐使用）。

• stream 输出报告的路径，默认输出控制台；
• verbosity 控制输出报告的详细程度，从0-2，越来越详细；

• file：文件
• verbosity：详细程度
• title：标题
• description：描述
• tester：作者

with open("接口测试报告.html","wb") as file:
    runner = HTMLTestRunnerNew.HTMLTestRunner(stream=file,  
                                              verbosity=2,
                                              title="接口自动化测试报告",
                                              description="接口测试V1",
                                              tester="ITester软件测试小栈")
    runner.run(suite)

import unittest
import HTMLTestRunnerNew

all_testcases=unittest.defaultTestLoader.discover(contants.testcases_dir, pattern='test_*.py',top_level_dir=None) 

#利用上下文管理器自动关闭资源
with open(contants.reports_html,"wb+") as file:    #选择绝对路径，把文件打开，写进内容 (报告的文件名直接写在路径里面)
    runner=HTMLTestRunnerNew.HTMLTestRunner(stream=file,
                                            title="接口自动化测试报告",
                                            description="接口测试V1",
                                            tester="ITester软件测试小栈")
    runner.run(all_testcases)