Python，Mockito，模拟另一个模块的进口

Question

假设我有module1.py

from my_library import useful_function

def do_something(x):
    return useful_function(x)

我正在测试module2.py

from module1 import do_something

def do_something_else(x):
    return do_something(x+1)

useful_function执行数据库调用，因此我想将其存根以返回常量值。

如何使用mockito来删除有用功能？我尝试导入my_library并使用

when( my_library ).useful_function('abc').thenReturn(...)

但这不起作用。如果我暂停Eclipse并将鼠标悬停在do_something（）内的useful_function行上，则看起来有用函数（）未被存根。

Answer 1

import module1
module1.useful_function = Mock(...)
return do_something(x+1)