无法在函数内修补对象

Question

我是 Python 新手。 我想为以下函数编写单元测试：

from common.ds_factory import DSFactory
class MyClass:

def load(self, parsed_file_key):

    ds = DSFactory.getDS()

    ...

现在我无法使用@patch(my_class.DSFactory)模拟DSFactory因为我在函数内部使用DSFactory 。

Answer 1

您需要为使用它的模块修补DSFactory 。 如评论中所述，假设MyClass在文件my_module.py定义，您可以使用模块名称对其进行修补： @patch("my_module.DSFactory") 。

有一个问题 - 您的模块必须是可导入的，因此它需要您创建模块。