bot-backend
31e316e04e
Some checks failed
CI / changes (push) Successful in 7s
CI / changes (pull_request) Successful in 6s
CI / frontend-tests (push) Has been cancelled
CI / openapi-codegen-check (push) Has been cancelled
CI / backend-tests (push) Has been cancelled
CI / backend-tests (pull_request) Has been cancelled
CI / frontend-tests (pull_request) Has been cancelled
CI / openapi-codegen-check (pull_request) Has been cancelled
113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
"""Тесты РИАСУРТ Свердл client (#108) — get_riasurt_sverdl_in_bbox + topic-классификация.
|
||
|
||
Сеть не дёргается: мокаем get_features_in_bbox_grid на инстансе NSPDClient. Проверяем
|
||
mapping layerId → features, классификацию topic, graceful-skip упавшего слоя.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
import pytest
|
||
|
||
from app.services.scrapers.nspd_client import (
|
||
RIASURT_SVERDL_LAYERS,
|
||
NSPDClient,
|
||
NSPDFeature,
|
||
NspdLiteWafError,
|
||
riasurt_layer_topic,
|
||
)
|
||
|
||
|
||
def test_riasurt_catalog_has_14_key_layers() -> None:
|
||
"""Каталог содержит ровно 14 ключевых layerId из issue #108."""
|
||
expected = {
|
||
845274,
|
||
846381,
|
||
844759,
|
||
844774,
|
||
845298,
|
||
846365,
|
||
846369,
|
||
845392,
|
||
846373,
|
||
845425,
|
||
846378,
|
||
844795,
|
||
844478,
|
||
846379,
|
||
}
|
||
assert set(RIASURT_SVERDL_LAYERS) == expected
|
||
assert len(RIASURT_SVERDL_LAYERS) == 14
|
||
|
||
|
||
def test_riasurt_layer_topic_known() -> None:
|
||
"""topic корректно резолвится по layerId."""
|
||
assert riasurt_layer_topic(845274) == "territorial_zone"
|
||
assert riasurt_layer_topic(846381) == "functional_zone"
|
||
assert riasurt_layer_topic(844759) == "red_lines"
|
||
assert riasurt_layer_topic(846365) == "szz"
|
||
assert riasurt_layer_topic(845392) == "szo"
|
||
assert riasurt_layer_topic(845425) == "flood_zone"
|
||
assert riasurt_layer_topic(844478) == "krt"
|
||
assert riasurt_layer_topic(846379) == "oez"
|
||
|
||
|
||
def test_riasurt_layer_topic_unknown_is_other() -> None:
|
||
"""Неизвестный layerId → 'other' (graceful)."""
|
||
assert riasurt_layer_topic(999999) == "other"
|
||
|
||
|
||
def _feat(layer_id: int) -> NSPDFeature:
|
||
return NSPDFeature.from_raw(
|
||
{
|
||
"id": str(layer_id),
|
||
"geometry": {"type": "Polygon", "coordinates": [[[1, 1], [2, 1], [2, 2], [1, 1]]]},
|
||
"properties": {"label": f"zone-{layer_id}"},
|
||
}
|
||
)
|
||
|
||
|
||
def test_get_riasurt_sverdl_in_bbox_maps_layers(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""Каждый запрошенный layerId → его features; форма dict[int, list]."""
|
||
client = NSPDClient()
|
||
calls: list[int] = []
|
||
|
||
def fake_grid(layer_id: int, bbox: Any, **kw: Any) -> list[NSPDFeature]:
|
||
calls.append(layer_id)
|
||
return [_feat(layer_id)]
|
||
|
||
monkeypatch.setattr(client, "get_features_in_bbox_grid", fake_grid)
|
||
|
||
bbox = (6770000.0, 7710000.0, 6782000.0, 7722000.0)
|
||
result = client.get_riasurt_sverdl_in_bbox(bbox, [845274, 844478])
|
||
|
||
assert set(result) == {845274, 844478}
|
||
assert len(result[845274]) == 1
|
||
assert result[845274][0].feature_id == "845274"
|
||
assert calls == [845274, 844478]
|
||
|
||
|
||
def test_get_riasurt_sverdl_in_bbox_default_all_14(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""layers=None → все 14 ключевых слоёв."""
|
||
client = NSPDClient()
|
||
monkeypatch.setattr(client, "get_features_in_bbox_grid", lambda lid, bbox, **kw: [_feat(lid)])
|
||
result = client.get_riasurt_sverdl_in_bbox((0.0, 0.0, 100.0, 100.0))
|
||
assert set(result) == set(RIASURT_SVERDL_LAYERS)
|
||
|
||
|
||
def test_get_riasurt_sverdl_in_bbox_skips_failed_layer(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""Упавший слой (WAF) → его ключ = [], остальные продолжают."""
|
||
client = NSPDClient()
|
||
|
||
def fake_grid(layer_id: int, bbox: Any, **kw: Any) -> list[NSPDFeature]:
|
||
if layer_id == 846365:
|
||
raise NspdLiteWafError("HTTP 429")
|
||
return [_feat(layer_id)]
|
||
|
||
monkeypatch.setattr(client, "get_features_in_bbox_grid", fake_grid)
|
||
result = client.get_riasurt_sverdl_in_bbox((0.0, 0.0, 100.0, 100.0), [845274, 846365, 844478])
|
||
|
||
assert result[846365] == []
|
||
assert len(result[845274]) == 1
|
||
assert len(result[844478]) == 1
|