yuxuanhui
57 lines
2.5 KiB
Python
57 lines
2.5 KiB
Python
"""Settings metadata retains platform choices, numeric bounds and scope dependencies."""
|
|
|
|
import pytest
|
|
from fastapi import HTTPException
|
|
|
|
from app.catalog.research_metadata import ResearchMetadata, setting_rows
|
|
from tests.catalog_fake import platform_response
|
|
|
|
|
|
def settings_response():
|
|
data = platform_response()
|
|
children = data["actions"]["POST"]["settings"]["children"]
|
|
children["neutralization"] = {"choices": [{"value": "INDUSTRY"}, {"value": "NONE"}]}
|
|
children["decay"] = {"min_value": 0, "max_value": 500}
|
|
children["truncation"] = {"choices": [{"value": 0.05}, {"value": 0.08}]}
|
|
children["visualization"] = {"choices": [{"value": False}, {"value": True}]}
|
|
children["nanHandling"] = {"choices": {"instrumentType": {"EQUITY": {"region": {
|
|
region: {"delay": {str(delay): [{"value": "OFF" if delay == 0 else "ON"}] for delay in [0, 1]}}
|
|
for region in ["USA", "CHN", "IND"]
|
|
}}}}}
|
|
return data
|
|
|
|
|
|
def test_metadata_preserves_types_and_resolves_delay():
|
|
rows = setting_rows(settings_response())
|
|
first = next(r for r in rows if r["region"] == "USA" and r["delay"] == 0)
|
|
second = next(r for r in rows if r["region"] == "USA" and r["delay"] == 1)
|
|
assert first["fields"]["decay"] == {"minimum": 0, "maximum": 500}
|
|
assert first["fields"]["truncation"]["choices"] == [0.05, 0.08]
|
|
assert first["fields"]["visualization"]["choices"] == [False, True]
|
|
assert first["fields"]["nanHandling"]["choices"] == ["OFF"]
|
|
assert second["fields"]["nanHandling"]["choices"] == ["ON"]
|
|
assert first["neutralizations"] == ["INDUSTRY", "NONE"]
|
|
|
|
|
|
async def test_refresh_publishes_fields_and_keeps_previous_snapshot_on_invalid_options(app):
|
|
class Client:
|
|
async def research_setting_options(self):
|
|
return settings_response()
|
|
|
|
async with app.state.sessions.begin() as db:
|
|
service = ResearchMetadata(db, Client())
|
|
previous = await service.refresh_settings()
|
|
assert previous["content"]["items"][0]["fields"]["visualization"]["choices"] == [False, True]
|
|
|
|
class InvalidClient:
|
|
async def research_setting_options(self):
|
|
data = settings_response()
|
|
data["actions"]["POST"]["settings"]["children"]["nanHandling"]["choices"] = {"unsupported": []}
|
|
return data
|
|
|
|
service.client = InvalidClient()
|
|
with pytest.raises(HTTPException) as error:
|
|
await service.refresh_settings()
|
|
assert error.value.status_code == 502
|
|
assert (await service.get("settings"))["content"] == previous["content"]
|