模擬功能，以及對這個功能的引用

Question

我需要模擬一個函數和所有指向這個函數的引用。

例子：

import mock 

def do(msg): 
    print(msg) 

def foo(): 
    do('foo') 

bar=foo 

with mock.patch(__name__ + '.foo', lambda *args: do('mocked')): 
    foo() 
    bar() 

輸出爲：

mocked 
foo 

但在我來說，我需要模擬真實的功能，而不只是它的名稱指向函數的實現。

背景：我想修補https://docs.djangoproject.com/en/dev/ref/models/fields/#django.db.models.PROTECT

我怎樣才能得到所需的輸出，而不使用「欄」。

mocked 
mocked 

Answer 1

而不是嘲笑foo，你可以簡單地嘲笑do代替。

with mock.patch(__name__ + '.do', lambda *args: print("mocked")): 
    foo() 
    bar()