tests/services/test_case_loader_client.py - master - tsps/core

tests/services/test_case_loader_client.py
54 строки2 KB
Василий Петров
Вызов case_loader
31 янв 2026, 19:49
31 янв 2026, 19:4977442a1
from unittest.mock import patch, MagicMock
import httpx
from app.services.case_loader_client import notify_case_loader
from app.core.config import config


@patch("app.services.case_loader_client._create_http_client")
@patch("app.services.case_loader_client.logger")
def test_notify_case_loader_success(mock_logger, mock_create_client):
    mock_response = httpx.Response(202, json={})
    mock_client = MagicMock()
    mock_client.post.return_value = mock_response
    mock_create_client.return_value.__enter__.return_value = mock_client

    notify_case_loader(42)

    mock_client.post.assert_called_once_with(
        f"{config.CASE_LOADER_URL}/load-documents",
        json={"document_import_job_id": 42}
    )
    mock_logger.info.assert_called_with("Успешно отправлена задача 42 в case_loader")


@patch("app.services.case_loader_client._create_http_client")
@patch("app.services.case_loader_client.logger")
def test_notify_case_loader_http_error(mock_logger, mock_create_client):
    mock_response = httpx.Response(500, text="Internal Server Error")
    mock_client = MagicMock()
    mock_client.post.return_value = mock_response
    mock_create_client.return_value.__enter__.return_value = mock_client

    notify_case_loader(42)

    mock_client.post.assert_called_once_with(
        f"{config.CASE_LOADER_URL}/load-documents",
        json={"document_import_job_id": 42}
    )
    mock_logger.error.assert_called()


@patch("app.services.case_loader_client._create_http_client")
@patch("app.services.case_loader_client.logger")
def test_notify_case_loader_exception(mock_logger, mock_create_client):
    mock_client = MagicMock()
    mock_client.post.side_effect = httpx.ConnectTimeout("Connection timed out")
    mock_create_client.return_value.__enter__.return_value = mock_client

    notify_case_loader(42)

    mock_client.post.assert_called_once_with(
        f"{config.CASE_LOADER_URL}/load-documents",
        json={"document_import_job_id": 42}
    )
    mock_logger.exception.assert_called()