如何让每个TestCase在setUp之后调用一个钩子？

Question

在我的项目中，我有一个 TestCase 子类（称为 BaseTest），它会做一些事情来准备然后重置测试环境，以及大量实际的测试用例子类（大约 80 个），每个子类都有自己独特的设置方法。

我希望每个单独的测试用例子类在它们的 setUp 方法结束时调用一个特定的钩子。我想在不改变每一种方法的情况下做到这一点。

基本上，情况大致类似于这个示例文件：

import unittest


class BaseTest(unittest.TestCase):
    def setUp(self):
        super().setUp()
        print('prepping env')

    def tearDown(self):
        super().tearDown()
        print('resetting env')

    def post_setup_hook(self):
        print('in post_setup_hook')


class TestFeatureA(BaseTest):
    def setUp(self):
        super().setUp()
        print('prepping a')

    def tearDown(self):
        super().tearDown()

    def test_0(self):
        print('testing a0')

    def test_1(self):
        print('testing a1')


class TestFeatureB(BaseTest):
    def setUp(self):
        super().setUp()
        print('prepping b')

    def tearDown(self):
        super().tearDown()

    def test_0(self):
        print('testing b0')

    def test_1(self):
        print('testing b1')


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

我希望运行python -m unittest example 的结果在每次打印“准备 a”或“准备 b”后打印“在设置后挂钩中”，但不修改 TestFeatureA 或 TestFeatureB。这个可以吗？

请注意，我使用的是 python 3.6。我认为这不会在 python 2.x 中运行。

Answer 1

您可以覆盖BaseTest(TestCase) 的run 方法来调用post_setup_hook after the setUp。

只需将run 方法复制粘贴到BaseTest 中，然后在setUp 运行后添加post_setup_hook 调用。

Answer 2

在这里使用装饰器怎么样？

def post_setup(f):
    def wrapper(*args):
        print('in post_setup_hook')
        f(*args)  # execute BaseTest setUp
        print(args[0].FOO)  # Access BasteTest class attribute
        print(args[0].foo)  # Access BasteTest object foo attribute
        print('out post_setup_hook')
    return wrapper


class BaseTest(unittest.TestCase):
    FOO = 'BAR'

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.foo = 'bar'

    @post_setup
    def setUp(self):
        super().setUp()
        print('BaseTest - prepping env')

    def tearDown(self):
        super().tearDown()
        print('BaseTest - resetting env')


class TestFeatureA(BaseTest):
    def test_0(self):
        print('TestFeatureA - testing a0')


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

输出：

在 post_setup_hook BaseTest - 准备环境 酒吧 酒吧 出 post_setup_hook TestFeatureA - 测试 a0 BaseTest - 重置环境