python - 修补 Python 子类中的所有方法

Question

我有很好的基础测试类，它从 django 测试用例和另一个类扩展而来:

 class MyTestCase(TestCase, TestFreshProvisionedEachTest):

现在一切正常，只是为了让它正常工作，我们必须(使用 Foord 的模拟库)修补中间件中的几个函数。

我的想法是这会工作得很好:

@patch('spotlight.middleware.extract_host_name', new=lambda host_name:
TEST_DOMAIN)
@patch('spotlight.middleware.extract_host_key', new=lambda host_key:
company_name_to_db(TEST_DOMAIN))
class MyTestCase(TestCase, TestFreshProvisionedEachTest):

不过好像不行，子类的方法没有打补丁。我当时想可能有一种方法可以做类似的事情

MyTestCase = patch(MyTestCase, 'spotlight.middleware.extract_host_name', new=lambda host_name: TEST_DOMAIN)

但这也是不可能的。

有没有办法避免这种重复并在父类(super class)上打补丁哪个也修补所有子类方法？

Answer 1

最好为此使用元类，因为它比手动应用装饰器更容易处理测试用例的继承。类似这些(我还没有测试过，但你应该明白了):

class TestMeta(type(TestCase)):
    def patch_method(cls, meth):
        raise NotImplementedError

    def __new__(mcls, name, bases, dct):
        to_patch = []

        for methname, meth in dct.items():
            if methname.startswith('test') and callable(meth):
                to_patch.append(methname)

        cls = super().__new__(mcls, name, bases, dct)

        for methname in to_patch:
            meth = getattr(cls, methname)
            meth = cls.patch_method(meth)
            setattr(cls, methname, meth)

        return cls

class PatchedTestCase(TestCase, metaclass=TestMeta):
    @classmethod
    def patch_method(cls, meth):
        meth = patch('spotlight.middleware.extract_host_name', 
                     new=lambda host_name: TEST_DOMAIN)(meth)

        meth = patch('spotlight.middleware.extract_host_key', 
                     new=lambda host_key: company_name_to_db(TEST_DOMAIN))(meth)

        return meth

class MyTestCase(PatchedTestCase, TestFreshProvisionedEachTest):
    ...

通过这种方法，PatchedTestCase 的所有子类的所有方法都将被修补。您还可以使用不同的 patch_method 实现定义其他基类。