本系列教程使用的python版本是3.6.3。

背景

这一节我们对unittest模块做一丁点的扩展，我们将实现动态运行用例的功能。

测试用例如下，基本上是从python官方文档上摘取的。

import unittest

class TestStringMethods(unittest.TestCase):

    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_split_important(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello', 'world'])
        # check that s.split fails when the separator is not a string
        with self.assertRaises(TypeError):
            s.split(2)

运行TestCase下全部的用例

这个需求是最常见的。根据官网的文档，代码实现如下

# load all test methods from a TestCase class
common_suite = unittest.TestLoader().loadTestsFromTestCase(TestStringMethods)
unittest.TextTestRunner(verbosity=2).run(common_suite)

使用TestLoader类中自带的loadTestsFromTestCase方法创建TestSuite运行既可。

运行TestCase下的某些用例

我们可以手动添加TestCase下的部分用例，然后一次执行。核心方法是使用addTest()方法

# add only 1 test method
one_case_suite = unittest.TestSuite()
one_case_suite.addTest(TestStringMethods('test_upper'))
unittest.TextTestRunner(verbosity=2).run(one_case_suite)

给用例打标签，只运行某些具有标签的用例

很多的测试框架都有类似的功能，我们可以用最简单的方式在unittest中实现类似功能。这里有一个小小的trick，就是我们通过命名测试用例的方式来实现打标签的功能。比如我们可以在测试方法名称后加上_important表示这是重要的测试方法，需要在每天进行回归。

# only run test methods the name of which ends with important
only_run_important_suite = unittest.TestSuite()
for case in [method for method in dir(TestStringMethods) if method.endswith('important')]:
    only_run_important_suite.addTest(TestStringMethods(case))
unittest.TextTestRunner(verbosity=2).run(only_run_important_suite)

这里的关键点是

• 在测试方法名后加上标签的名字
• 使用dir(ClassName)的方式获取该类下所有定义的方法名
• 使用addTest()动态添加用例

自定义用例的运行顺序

下面的代码演示了如何按照用例名称的长度，由长到短运行用例方法。也就是名称长的先执行，名称短的后执行。

# run test methods in custom order
custom_order_suite = unittest.TestSuite()
for case in sorted([method for method in dir(TestStringMethods) if method.startswith('test_')], key=len, reverse=True):
    custom_order_suite.addTest(TestStringMethods(case))
unittest.TextTestRunner(verbosity=2).run(custom_order_suite)

源码地址

github地址