如何使用 pytest 对本地导入的模块进行猴子补丁以进行 Python 测试？

Question

我正在处理一个非常大的项目，并且已经有很多使用pytest的monkeypatch夹具的测试。 我希望从属于导入模块的类中修补特定方法，例如：

from project.common import services

在服务包中有一个类，其中包含我希望修补的方法，例如：

services.utils.Calculations.get_area()

我尝试mock和monkeypatch ：

mocked_get_area_method= Mock(return_value=500)
monkeypatch.setattr(
   'services.utils.Calculations.get_area',
    mocked_get_user_ip_method,
)

然后我在我的测试中创建一个对象：

class TestCommon:
    def test_calculations(self,monkeypatch):
        mocked_get_area_method= Mock(return_value=500)
        monkeypatch.setattr(
           'services.utils.Calculations.get_area',
            mocked_get_user_ip_method,
        )
        calculations = services.utils.Calculations()
        calculations.get_area()
        mocked_get_user_ip_method.assert_called_once()

但我收到一条错误消息： ModuleNotFoundError: No module named 'services'.

我相信错误来自这样一个事实，即可能是monkeypatch从高级主项目文件夹开始查找对象。 如果我尝试使用这条路径进行monkeypath路径：

        monkeypatch.setattr(
           'project.common.services.utils.Calculations.get_area',
            mocked_get_user_ip_method,
        )

monkeypatching 工作但我没有在我的assert中得到True因为我相信 monkeypatching 已经改变了主要项目中的对象但是因为我已经导入它并从本地导入的模块实例化services.utils.Calculations()补丁不起作用。

我怎样才能使这项工作？

注意： pytest 从主项目目录运行。

Answer 1

您需要将补丁应用于具有from project.common import services语句的模块。 例如，假设我有一个文件project/frontend.py ，如下所示：

from project.common import services


def a_method():
    calc = services.utils.Calculations()
    return calc.get_area()

我想测试a_method是否正确返回它从get_area()调用接收到的任何值，所以在tests/test_frontend.py我可能会写：

import project.frontend

from unittest import mock


def test_a_method(monkeypatch):
    fake_get_area = mock.Mock(return_value=2.0)

    monkeypatch.setattr(
        project.frontend.services.utils.Calculations, "get_area", fake_get_area
    )

    res = project.frontend.a_method()

    assert res == 2.0

在这里，我们正在修补project.frontend.services.utils.Calculations因为我们已经导入了project.frontend并且frontend.py导入了project.common.services作为名称services 。