無法在函數內修補對象

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") 。

有一個問題 - 您的模塊必須是可導入的，因此它需要您創建模塊。