需要在python测试用例中模拟一些基类行为

时间:2013-12-11 22:06:27

标签: python inheritance super base-class python-mock

我的头衔是相当具有描述性的,但是这里有。 假设我有这个设置。

class BaseClass(object):
    def __init__(self):
        pass
    def base_function(self, param="Hello World"):
        print param

#multiple inheritance probably irrelevant but my problem deals with it
class DerivedClass(BaseClass, AnotherBaseClass):
    def __init__(self):
        pass
    def advanced_function(self):
        #blah blah blah
        #code code code
        self.base_function()

现在,我遇到了一个测试派生类的情况,但是这样做,我需要确保调用它的基类方法。我尝试过做这样的事情

from mock import MagicMock

d = DerivedClass()
super(DerivedClass, d).base_function = MagicMock()
d.advanced_function()
super(DerivedClass, d).base_function.assert_called()

我100%确定此设置错误,因为

AttributeError: 'super' object has no attribute 'base_function'

我知道我在做超级错误,有人有想法吗?

1 个答案:

答案 0 :(得分:1)

通过BaseClass.base_function访问。由于您没有重载该方法,只需继承它,DerivedClass.base_function是同一个对象:

id(BaseClass.base_function) == id(DerivedClass.base_function)

创建实例时,它们会继承模拟。

相关问题