feat: add project-scoped operations
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import AsyncMock
|
||||
from uuid import uuid4
|
||||
|
||||
@@ -7,7 +8,7 @@ from fastapi.testclient import TestClient
|
||||
|
||||
from app.api.v1.endpoints import place_search
|
||||
from app.api.v1.rest_router import api_router
|
||||
from app.auth.project_dependencies import ProjectContext, get_project_context
|
||||
from app.auth.project_dependencies import ProjectContext, get_project_context, get_metadata_repository
|
||||
from app.domain.schemas.place_search import PlaceSearchItem, PlaceSearchResponse
|
||||
from app.services.place_search import PlaceSearchError
|
||||
|
||||
@@ -17,13 +18,18 @@ def app():
|
||||
application = FastAPI()
|
||||
application.include_router(api_router, prefix="/api/v1")
|
||||
application.dependency_overrides[get_project_context] = lambda: ProjectContext(uuid4(), "demo", uuid4(), "viewer")
|
||||
application.dependency_overrides[get_metadata_repository] = lambda: SimpleNamespace(
|
||||
get_project_by_id=AsyncMock(return_value=SimpleNamespace(map_config={
|
||||
"place_search": {"enabled": True, "provider": "tianditu", "admin_code": "310120", "region": "奉贤区"}
|
||||
}))
|
||||
)
|
||||
return application
|
||||
|
||||
|
||||
def test_viewer_can_search_with_existing_project_read_permission(app, monkeypatch):
|
||||
service = AsyncMock(return_value=PlaceSearchResponse(items=[PlaceSearchItem(
|
||||
id="place-1", name="南桥路", address="上海市奉贤区南桥路", longitude=121.44, latitude=31.22,
|
||||
)], total=1))
|
||||
id="place-1", name="南桥路", address="上海市奉贤区南桥路", longitude=121.44, latitude=31.22, district="奉贤区",
|
||||
)], total=1, region="奉贤区"))
|
||||
monkeypatch.setattr(place_search, "search_tianditu_places", service)
|
||||
response = TestClient(app).post("/api/v1/place-search-requests", json={"keyword": "南桥路"})
|
||||
assert response.status_code == 200
|
||||
@@ -63,3 +69,73 @@ def test_search_contract_has_typed_results_and_required_project_header(app):
|
||||
assert operation["responses"]["200"]["content"]["application/json"]["schema"]["$ref"].endswith("PlaceSearchResponse")
|
||||
assert set(contract["components"]["schemas"]["PlaceSearchRequest"]["properties"]) == {"keyword"}
|
||||
assert contract["components"]["schemas"]["PlaceSearchResponse"]["properties"]["items"]["maxItems"] == 10
|
||||
reverse = contract["paths"]["/api/v1/reverse-geocoding-requests"]["post"]
|
||||
assert any(param["name"] == "X-Project-Id" and param["required"] for param in reverse["parameters"])
|
||||
assert reverse["responses"]["200"]["content"]["application/json"]["schema"]["$ref"].endswith("PlaceSearchItem")
|
||||
coordinates = contract["components"]["schemas"]["ReverseGeocodingRequest"]
|
||||
assert set(coordinates["properties"]) == {"longitude", "latitude"}
|
||||
assert coordinates["additionalProperties"] is False
|
||||
|
||||
|
||||
@pytest.mark.parametrize('map_config', [{}, {'place_search': None},
|
||||
{'place_search': {'enabled': False, 'admin_code': '341126', 'region': '凤阳县'}},
|
||||
{'place_search': {'enabled': True, 'admin_code': 'invalid', 'region': '凤阳县'}}])
|
||||
def test_unconfigured_or_invalid_project_does_not_search(app, monkeypatch, map_config):
|
||||
service = AsyncMock()
|
||||
monkeypatch.setattr(place_search, 'search_tianditu_places', service)
|
||||
app.dependency_overrides[get_metadata_repository] = lambda: SimpleNamespace(
|
||||
get_project_by_id=AsyncMock(return_value=SimpleNamespace(map_config=map_config)))
|
||||
response = TestClient(app).post('/api/v1/place-search-requests', json={'keyword': '地名'})
|
||||
assert response.status_code == 503
|
||||
service.assert_not_awaited()
|
||||
|
||||
|
||||
def test_search_uses_configuration_of_authenticated_project(app, monkeypatch):
|
||||
project_id = uuid4()
|
||||
app.dependency_overrides[get_project_context] = lambda: ProjectContext(project_id, 'fengyang_v2', uuid4(), 'viewer')
|
||||
repository = SimpleNamespace(get_project_by_id=AsyncMock(return_value=SimpleNamespace(
|
||||
map_config={'place_search': {'enabled': True, 'admin_code': '341126', 'region': '凤阳县'}})))
|
||||
app.dependency_overrides[get_metadata_repository] = lambda: repository
|
||||
service = AsyncMock(return_value=PlaceSearchResponse(items=[], total=0, region='凤阳县'))
|
||||
monkeypatch.setattr(place_search, 'search_tianditu_places', service)
|
||||
response = TestClient(app).post('/api/v1/place-search-requests', json={'keyword': '地名'})
|
||||
assert response.status_code == 200 and response.json()['region'] == '凤阳县'
|
||||
repository.get_project_by_id.assert_awaited_once_with(project_id)
|
||||
assert service.await_args.kwargs['config'].admin_code == '341126'
|
||||
|
||||
|
||||
def test_viewer_reverse_lookup_uses_project_configuration_and_typed_point(app, monkeypatch):
|
||||
service = AsyncMock(return_value=PlaceSearchItem(id='map-point', name='南桥路', address='奉贤区南桥路',
|
||||
longitude=121.45, latitude=30.92, district='奉贤区'))
|
||||
monkeypatch.setattr(place_search, 'reverse_geocode_tianditu', service)
|
||||
response = TestClient(app).post('/api/v1/reverse-geocoding-requests', json={'longitude': 121.45, 'latitude': 30.92})
|
||||
assert response.status_code == 200 and response.json()['name'] == '南桥路'
|
||||
assert service.await_args.args[0].longitude == 121.45
|
||||
assert service.await_args.kwargs['config'].admin_code == '310120'
|
||||
|
||||
|
||||
@pytest.mark.parametrize('body', [{'longitude': 181, 'latitude': 0},
|
||||
{'longitude': 121, 'latitude': 91}, {'longitude': 121, 'latitude': 31, 'region': '全国'}])
|
||||
def test_reverse_lookup_rejects_invalid_requests_before_provider_call(app, monkeypatch, body):
|
||||
service = AsyncMock()
|
||||
monkeypatch.setattr(place_search, 'reverse_geocode_tianditu', service)
|
||||
assert TestClient(app).post('/api/v1/reverse-geocoding-requests', json=body).status_code == 422
|
||||
service.assert_not_awaited()
|
||||
|
||||
|
||||
@pytest.mark.parametrize('role,status', [('none', 403), ('viewer', 503)])
|
||||
def test_reverse_lookup_requires_project_permission_and_configuration(app, monkeypatch, role, status):
|
||||
app.dependency_overrides[get_project_context] = lambda: ProjectContext(uuid4(), 'demo', uuid4(), role)
|
||||
app.dependency_overrides[get_metadata_repository] = lambda: SimpleNamespace(
|
||||
get_project_by_id=AsyncMock(return_value=SimpleNamespace(map_config={})))
|
||||
service = AsyncMock()
|
||||
monkeypatch.setattr(place_search, 'reverse_geocode_tianditu', service)
|
||||
assert TestClient(app).post('/api/v1/reverse-geocoding-requests', json={'longitude': 121, 'latitude': 31}).status_code == status
|
||||
service.assert_not_awaited()
|
||||
|
||||
|
||||
@pytest.mark.parametrize('status', [404, 422, 502, 503, 504])
|
||||
def test_reverse_lookup_preserves_actionable_errors(app, monkeypatch, status):
|
||||
monkeypatch.setattr(place_search, 'reverse_geocode_tianditu', AsyncMock(side_effect=PlaceSearchError(status, '请重新取点')))
|
||||
response = TestClient(app).post('/api/v1/reverse-geocoding-requests', json={'longitude': 121, 'latitude': 31})
|
||||
assert response.status_code == status and response.json()['detail'] == '请重新取点'
|
||||
|
||||
Reference in New Issue
Block a user