unittest框架

unittest框架

1、写用例：TestCase 类
2、加载用例：TestSuite 类    TestLoader 类
3、对比结果（执行结果/期望结果）：assert函数
4、执行并出结果：TextTestRunner 类
5、报告：HtmlTestReport 类

1）首先建一个math_method.py文件，用来存在被测试函数

# 被测类函数
class MathMethod():

    def add(self, a, b):
        return a + b

    def sub(self, a, b):
        return a - b

2）新建一个unittest_case1.py文件，用来存放测试用例代码

# from unittest.case import TestCase  #在文件unittest文件下面case文件中存在TestCase模块
# class TestAdd(TestCase):

import unittest
from test_unittest.math_method import MathMethod  # 导入被测试类


class TestAdd(unittest.TestCase):  # 测试类  专门测试加法类
    # 写用例 每一条用例都是一个类函数
    # 用例 def test_后缀(self) ：用例函数必须test开头

    def setUp(self):  # 测试环境的搭建
        print("我要开始测试咯")
        self.t = MathMethod()

    def tearDown(self):  # 测试环境的销毁
        print("我测试完成咯")

    def test_add_two_zero(self):  # 测试两个零相加的用例
        # t = MathMethod()  # 创建实例
        # res = t.add(0, 0)

        res = self.t.add(0, 0)
        # 断言函数的调用，注意:为啥用self，因为调用父类的方法
        self.assertEqual(1, res)
        print("测试的结果是：{0}".format(res))

    def test_add_two_positive(self):  # 测试两个整数相加的用例
        # t = MathMethod()  # 创建实例
        # res = t.add(4, 5)

        res = self.t.add(4, 5)
        print("测试的结果是：{0}".format(res))

    def test_add_two_negative(self):  # 测试两个负数相加的用例
        # t = MathMethod()  # 创建实例
        # res = t.add(-4, -5)

        res = self.t.add(-4, -5)
        print("测试的结果是：{0}".format(res))


class TestSub(unittest.TestCase):  # 测试类 专门测试减法
    # 写用例 每一条用例都是一个类函数
    # 用例 def test_后缀(self) ：用例函数必须test开头

    def test_sub_two_zero(self):  # 测试两个零相减的用例
        t = MathMethod()  # 创建实例
        res = t.sub(0, 0)
        print("测试的结果是：{0}".format(res))

    def test_sub_two_positive(self):  # 测试两个整数相减的用例
        t = MathMethod()  # 创建实例
        res = t.sub(4, 5)
        print("测试的结果是：{0}".format(res))

    def test_sub_two_negative(self):  # 测试两个负数相减的用例
        t = MathMethod()  # 创建实例
        res = t.sub(-4, -5)
        print("测试的结果是：{0}".format(res))


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

3）新建一个unittest_suite1.py文件，用来存放加载用例，执行用例，生成报告的代码

import unittest
from project.unittest_case1 import AddTest  # 导入测试加法类
from project.unittest_case1 import SubTest  # 导入测试减法类
from project import unittest_case1  #导入模块
from HTMLTestRunnerNew import HTMLTestRunner

#--------------------------加载测试用例并执行，最后生成测试报告（txt/html格式）-----------------------------

########### 创建实例#########
suite = unittest.TestSuite()

##########加载测试用例##########
# 方法一：TestSuite加载测试用例【以实例形式】
suite.addTest(AddTest("test1_1"))
suite.addTest(AddTest("test1_2"))
suite.addTest(AddTest("test1_3"))
suite.addTest(AddTest("test1_4"))
suite.addTest(AddTest("test1_5"))
suite.addTest(AddTest("test1_6"))

suite.addTest(SubTest("test2_1"))


# # 方法二：使用TestLoader加载测试用例【以函数、模块形式】
# loader = unittest.TestLoader()
# suite.addTest(loader.loadTestsFromModule(unittest_case1))
# suite.addTest(loader.loadTestsFromTestCase(SubTest))
# suite.addTest(loader.loadTestsFromTestCase(AddTest))


# -----------------------------------TextTestRunner执行测试用例，并生成报告（txt）-------------------------------------------
# with open("test_resul.txt", 'w+',encoding="utf-8") as file:
#     runner = unittest.TextTestRunner(stream=file,
#                                      descriptions='接口测试报告',
#                                      verbosity=2)
#     runner.run(suite)

#  -----------------------------------TextTestRunner执行测试用例，并生成报告（txt）-------------------------------------------
with open("test_resul.html", 'wb+') as file:
    runner = HTMLTestRunner(stream=file,
                            verbosity=2,
                            title='接口测试报告',
                            description=None,
                            tester='稳稳的幸福')
    runner.run(suite)

备注：

执行结果，如果鼠标放在某一个测试用例函数后面则只执行这个用例；若放在最后一行测试用例的下面则执行所有测试用例
执行结果顺序按照函数名ASCII码排序0-9，A-Z, a-z，test_add_two_negative test_add_two_positive test_add_two_zero
使用main函数则也是执行整个所有用例