我正在使用 python 模拟框架进行测试 (http://www.voidspace.org.uk/python/mock/)，我想模拟一个父类(super class)并专注于测试子类的添加行为。

(对于那些感兴趣的人，我已经扩展了 pymongo.collection.Collection，我只想测试我添加的行为。我不想为了测试目的而将 mongodb 作为另一个进程运行。)

对于本次讨论，A 是父类(super class)，B 是子类。此外，我定义了直接和间接父类(super class)调用，如下所示:

class A(object):
    def method(self):
        ...

    def another_method(self):
        ...

class B(A):
    def direct_superclass_call(self):
        ...
        A.method(self)

    def indirect_superclass_call(self):
        ...
        super(A, self).another_method()

方法#1

为 A 定义一个名为 MockA 的模拟类，并使用 mock.patch 将其替换为运行时的测试。这处理直接父类(super class)调用。然后操纵 B.__bases__ 来处理间接父类(super class)调用。 (见下文)

出现的问题是我必须编写 MockA 并且在某些情况下(如 pymongo.collection.Collection 的情况)这可能涉及很多努力解开所有模拟出来的内部调用。

方法#2

所需的方法是以某种方式使用 mock.Mock() 类来及时处理对 mock 的调用，以及在测试中定义的 return_value 或 side_effect。这样一来，我就可以通过避免 MockA 的定义来做更少的工作。

我遇到的问题是我无法弄清楚如何更改 B.__bases__ 以便可以将 mock.Mock() 的实例作为父类(super class)放置(我必须需要以某种方式在这里进行一些直接绑定(bind))。到目前为止，我已经确定，super() 检查 MRO，然后调用定义相关方法的第一个类。我无法弄清楚如何让父类(super class)处理对它的检查并在遇到模拟类时成功。在这种情况下，似乎没有使用 __getattr__ 。我希望 super 认为该方法已在此时定义，然后像往常一样使用 mock.Mock() 功能。

super() 如何发现在 MRO 序列中的类中定义了哪些属性？有没有办法让我在这里插话并以某种方式让它动态利用 mock.Mock()？

import mock

class A(object):
    def __init__(self, value):
        self.value = value      

    def get_value_direct(self):
        return self.value

    def get_value_indirect(self):
        return self.value   

class B(A):
    def __init__(self, value):
        A.__init__(self, value)

    def get_value_direct(self):
        return A.get_value_direct(self)

    def get_value_indirect(self):
        return super(B, self).get_value_indirect()


# approach 1 - use a defined MockA
class MockA(object):
    def __init__(self, value):
        pass

    def get_value_direct(self):
        return 0

    def get_value_indirect(self):
        return 0

B.__bases__ = (MockA, )  # - mock superclass 
with mock.patch('__main__.A', MockA):  
    b2 = B(7)
    print '\nApproach 1'
    print 'expected result = 0'
    print 'direct =', b2.get_value_direct()
    print 'indirect =', b2.get_value_indirect()
B.__bases__ = (A, )  # - original superclass 


# approach 2 - use mock module to mock out superclass

# what does XXX need to be below to use mock.Mock()?
#B.__bases__ = (XXX, )
with mock.patch('__main__.A') as mymock:  
    b3 = B(7)
    mymock.get_value_direct.return_value = 0
    mymock.get_value_indirect.return_value = 0
    print '\nApproach 2'
    print 'expected result = 0'
    print 'direct =', b3.get_value_direct()
    print 'indirect =', b3.get_value_indirect() # FAILS HERE as the old superclass is called
#B.__bases__ = (A, )  # - original superclass

最佳答案

is there a way for me to interject here and to somehow get it to utilize a mock.Mock() on the fly?

可能有更好的方法，但您始终可以编写自己的 super() 并将其注入(inject)到包含您要模拟的类的模块中。让它根据调用它的内容返回它应该返回的任何内容。

您可以在当前命名空间中只定义 super() (在这种情况下，重新定义仅适用于定义后的当前模块)，或者您可以 import __builtin__ 并将重新定义应用到 __builtin__.super，在这种情况下，它将在 Python session 中全局应用。

您可以使用默认参数捕获原始 super 函数(如果您需要从您的实现中调用它):

def super(type, obj=None, super=super):  
    # inside the function, super refers to the built-in

关于python - 如何使用 python mock 直接模拟父类(super class)？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/11194847/