如何使用 pytest_mock 模拟从另一个类调用的类方法

Question

在下面的文件中，我有InternalDogWebhookResource调用VisitOrchestrator.fetch_visit 。 我正在尝试为InternalDogWebhookResource编写一个测试，但模拟VisitOrchestrator.fetch_visit因为它是一个网络调用。

我试过模拟路径：

• api.dog.handlers.internal.VisitOrchestrator.fetch_visit
• api.dog.handlers.internal.InternalDogWebhookResource.VisitOrchestrator.fetch_visit
• api.dog.handlers.internal.InternalDogWebhookResource.fetch_visit

和许多其他人，但我总是收到AssertionError: assert None

我可以确认测试中的client.post有效，因为当我删除模拟断言时，我从 api 得到响应，这意味着fetch_visit 。

如何找到mocker.patch路径？

api/dog/handlers/internal.py

from api.dog.helpers.visits import VisitOrchestrator

@api.route("/internal/dog/webhook")
class InternalDogWebhookResource():

    def post(self) -> JsonResponse:
        if event_type == EventType.CHANGE:
            VisitOrchestrator.fetch_visit(event['visitId'])

        return JsonResponse(status=204)

api/dog/helpers/visits.py

class VisitOrchestrator:
    @classmethod
    def fetch_visit(cls, visit_id: str) -> VisitModel:
        # do stuff

        return visit

tests/v0/dog/handlers/test_webhook.py

import pytest
from pytest_mock import MockerFixture

from api.dog.handlers.internal import InternalDogWebhookResource, EventType
from tests.v0.utils import url_for


def test_webhook_valid(client, config, mocker: MockerFixture):
    visit_id = '1231231'

    mock_object = mocker.patch(
        'api.dog.handlers.internal.VisitOrchestrator.fetch_visit',
        return_value=visit_id,
    )

    res = client.post(
        url_for(InternalDogWebhookResource),
        json={'blag': 'blargh'}
    )
    assert mock_object.assert_called_once()

Answer 1

您正在做正确的事情 - 您的第二种方法通常是使用api.dog.handlers.internal.InternalDogWebhookResource.VisitOrchestrator.fetch_visit的方法（ api.dog.handlers.internal.InternalDogWebhookResource.VisitOrchestrator.fetch_visit ）

我会尝试做最小的测试代码功能：

def test_webhook_valid(mocker):
    mock_fetch_visit = mocker.MagicMock(name='fetch_visit')
    mocker.patch('api.dog.handlers.internal.VisitOrchestrator.fetch_visit',
                 new=mock_fetch_visit)

    InternalDogWebhookResource().post()

    assert 1 == mock_fetch_visit.call_count

如果这对您有用 - 可能问题出在测试方法中的client或其他设置上。