Python：嘲笑我正在測試的模塊正在使用的模塊

Question

我想模擬某個模塊以測試使用該模塊的一段代碼。

也就是說，我有一個模塊my_module，我想測試。 my_module引入外部模塊real_thing，並呼籲real_thing.compute_something()：

#my_module 
import real_thing 
def my_function(): 
    return real_thing.compute_something() 

我需要模擬real_thing，這樣在測試它會像fake_thing，我已經創建了一個模塊：

#fake_thing 
def compute_something(): 
    return fake_value 

測試調用my_module.my_function()這就要求real_thing.compute_something()：

#test_my_module 
import my_module 
def test_my_function(): 
    assert_something(my_module.my_function()) 

我要補充什麼的測試代碼以便my_function()將在測試內部調用fake_thing.compute_something()而不是real_thing.compute_something()？

我想弄清楚如何與Mock做到這一點，但我沒有。

Answer 1

只要沒有？破解這個sys.modules

#fake_thing.py 
def compute_something(): 
    return 'fake_value' 

#real_thing.py 
def compute_something(): 
    return 'real_value' 

#my_module.py 
import real_thing 
def my_function(): 
    return real_thing.compute_something() 

#test_my_module.py 
import sys 

def test_my_function(): 
    import fake_thing 
    sys.modules['real_thing'] = fake_thing 
    import my_module 
    print my_module.my_function() 

test_my_function() 

輸出： 'fake_value'

Answer 2

http://code.google.com/p/mockito-python/

>>> from mockito import * 
>>> dog = mock() 
>>> when(dog).bark().thenReturn("wuff") 
>>> dog.bark() 
'wuff' 

http://technogeek.org/python-module.html - 如何更換，加載模塊動態