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