如何使用 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或其他設置上。