Python单元测试框架：unittest（二）

一、直接使用TestCase

注意所有测试方法都需要以test开头。代码如下：

import unittest

class Test1(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        print("execute setUpClass")

    @classmethod
    def tearDownClass(self):
        print("execute tearDownClass")

    def setUp(self):
        print("execute setUp")

    def tearDown(self):
        print("execute tearDown")

    def test_one(self):
        print('execute test_one')
        self.assertTrue('FOO'.isupper())

    def test_two(self):
        print('execute test_two')


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

执行如下：

二、使用TestSuite

直接写TestCase执行时是按字母排序的顺序执行的，如果要设定测试用例的执行先后顺序则需要将TestCase封装到TestSuite。代码如下：

import unittest

class Test2(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        print("execute setUpClass")

    @classmethod
    def tearDownClass(self):
        print("execute tearDownClass")

    def setUp(self):
        print("execute setUp")

    def tearDown(self):
        print("execute tearDown")

    def test_one(self):
        print('execute test_one')
        self.assertTrue('FOO'.isupper())

    def test_two(self):
        print('execute test_two')


if __name__ == '__main__':
        suite = unittest.TestSuite()
        # Test2是要测试的类名，test_two是要执行的测试方法
        suite.addTest(Test2("test_two"))
        suite.addTest(Test2("test_one"))
        runner = unittest.TextTestRunner()
        runner.run(suite)

执行如下，可以看到先添加的test_two先执行：

三、使用HTMLTestRunner

我们可能还会希望生成一个简单的HTML报告，可使用HTMLTestRunner实现。但pypi和官网上最新的都是只支持python2.x的0.8.2版本。可将以下代码自行保存成HTMLTestRunner.py放到自己项目目录下

View Code

代码如下：

import unittest
import HTMLTestRunner

class Test3(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        print("execute setUpClass")

    @classmethod
    def tearDownClass(self):
        print("execute tearDownClass")

    def setUp(self):
        print("execute setUp")

    def tearDown(self):
        print("execute tearDown")

    def test_one(self):
        print('execute test_one')
        self.assertTrue('FOO'.isupper())

    def test_two(self):
        print('execute test_two')


if __name__ == '__main__':
        suite = unittest.TestSuite()
        # Test3是要测试的类名，test_one是要执行的测试方法
        suite.addTest(Test3("test_one"))
        suite.addTest(Test3("test_two"))
        # 实践中发现执行时的当前路径，不一定是此文件所在的文件夹，所以使用绝对路径
        # print(f"{os.getcwd()}")
        filename = 'F:\\PycharmProjects\\test3\\apptestresult.html'
        fb = open(filename, 'wb')
        runner = HTMLTestRunner.HTMLTestRunner(stream=fb, title="测试HTMLTestRunner", description="测试HTMLTestRunner")
        runner.run(suite)
        fb.close()

执行如下：