Python的模拟补丁的功能缺失的参数

Question

我在模块下面的代码：

class MyManager(models.Manager): 

    def dowork(self, value1, value2): 
     print value1, value2 

在模块B：

class MyModel(models.Model): 
    objects = MyManager() 
    value1 = ...   
    value2 = ... 

    def call_manager(self): 
     self.objects.dowork(self.value1, self.value2) 

在单元测试我修补dowork与不同的机构，例如如：

def new_dowork(self, value1, value2): 
    print 123 

with patch('a.MyManager.dowork', new_callable=new_dowork): 
    record = MyModel(value=111) 
    record.call_manager() 
    ... 

但我正在逐渐TypeError: new_dowork() takes exactly 3 arguments (0 given)

Answer 1

我对Mock了解不多，但我认为主要问题是您应该在new参数中传递模拟函数，而不是在new_callable中，并使用patch.object实际覆盖对象中的方法。

这似乎工作：

from mock import patch 

class MyManager(object): 

    def dowork(self, value1, value2): 
     print value1, value2 

class MyModel(object): 
    objects = MyManager() 
    value1 = 'foo'   
    value2 = 'bar' 

    def __init__(self, value=0): 
     # You don't need this __init__... Is just so I don't have to 
     # use models.MyModel thingy 
     pass 

    def call_manager(self): 
     self.objects.dowork(self.value1, self.value2) 

def new_dowork(self, value1, value2): 
    print "Mock called" 
    print 123 

with patch.object(MyManager, 'dowork', new=new_dowork): 
    record = MyModel(value=111) 
    record.call_manager() 

此输出：

Mock called 
123