74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""Тесты для POST /admin/scrape/ekburg-permits.
|
||
|
||
Проверяет:
|
||
- валидный запрос без year → scope all_years_2022_2026, task_id в ответе
|
||
- валидный запрос с year=2026 → scope year_2026
|
||
- year < 2022 или > 2030 → 422
|
||
|
||
Авторизация здесь НЕ проверяется и никогда не проверялась: строка «отсутствие
|
||
X-Admin-Token → 401/503» стояла в этом докстринге, но соответствующего теста в
|
||
файле нет. Гейт `/api/v1/admin/*` — middleware `rbac_guard` (app/main.py),
|
||
покрыт tests/test_rbac.py.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from unittest.mock import MagicMock, patch
|
||
|
||
import pytest
|
||
from fastapi.testclient import TestClient
|
||
|
||
from app.main import app
|
||
|
||
ENDPOINT = "/api/v1/admin/scrape/ekburg-permits"
|
||
|
||
|
||
def _mock_task(task_id: str = "fake-task-id-123") -> MagicMock:
|
||
result = MagicMock()
|
||
result.id = task_id
|
||
return result
|
||
|
||
|
||
def test_trigger_refresh_all_returns_task_id() -> None:
|
||
"""POST без year → refresh_all queued, scope=all_years_2022_2026."""
|
||
mock_result = _mock_task("task-all-001")
|
||
|
||
with (
|
||
patch("app.workers.tasks.ekburg_permits_sync.refresh_all") as mock_refresh_all,
|
||
patch("app.workers.tasks.ekburg_permits_sync.refresh_year"),
|
||
):
|
||
mock_refresh_all.apply_async.return_value = mock_result
|
||
client = TestClient(app)
|
||
response = client.post(ENDPOINT, json={})
|
||
|
||
assert response.status_code == 200, response.text
|
||
body = response.json()
|
||
assert body["task_id"] == "task-all-001"
|
||
assert body["scope"] == "all_years_2022_2026"
|
||
assert "queued_at" in body
|
||
|
||
|
||
def test_trigger_refresh_year_returns_task_id() -> None:
|
||
"""POST year=2026 → refresh_year queued, scope=year_2026."""
|
||
mock_result = _mock_task("task-year-002")
|
||
|
||
with (
|
||
patch("app.workers.tasks.ekburg_permits_sync.refresh_year") as mock_refresh_year,
|
||
patch("app.workers.tasks.ekburg_permits_sync.refresh_all"),
|
||
):
|
||
mock_refresh_year.apply_async.return_value = mock_result
|
||
client = TestClient(app)
|
||
response = client.post(ENDPOINT, json={"year": 2026})
|
||
|
||
assert response.status_code == 200, response.text
|
||
body = response.json()
|
||
assert body["task_id"] == "task-year-002"
|
||
assert body["scope"] == "year_2026"
|
||
|
||
|
||
@pytest.mark.parametrize("bad_year", [2021, 2031, 1999, 9999])
|
||
def test_trigger_invalid_year_returns_422(bad_year: int) -> None:
|
||
"""year вне диапазона [2022, 2030] → 422 Unprocessable Entity."""
|
||
client = TestClient(app)
|
||
response = client.post(ENDPOINT, json={"year": bad_year})
|
||
assert response.status_code == 422, f"year={bad_year} должен возвращать 422"
|