如何在 Django 中模拟 function？

Question

我希望以下对which_user的调用返回self.user ，无论传入什么，但它的行为就好像它根本没有被嘲笑。

def test_user_can_retrieve_favs_using_impersonation(self):
        with mock.patch('impersonate.helpers.which_user', return_value=self.user):
            user = which_user(self.user2)

我在这里做错了什么？ 我像这样导入了which_user ：如果有帮助， from impersonate.helpers import which_user 。

Answer 1

您需要通过它的导入路径来模拟 function 。 例如，您在 myapp 的视图中使用 function。 因此，您需要通过以下方式进行操作：

with mock.patch('myapp.views.which_user', return_value=self.user):
        user = which_user(self.user2)

这是因为它只修补可测试模块中定义的变量，而不修补系统或 venv 中的原始库和包。