54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
|
from datetime import UTC, datetime
|
||
|
|
from decimal import Decimal
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from app.core.fund_contracts import FundQuoteQuery
|
||
|
|
from app.infrastructure.fund_quote_cache import FundQuoteCache
|
||
|
|
from app.infrastructure.memory_cache import CacheReadResult
|
||
|
|
from app.service.fund_quote_service import FundQuoteService
|
||
|
|
|
||
|
|
|
||
|
|
class MemoryClient:
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.values: dict[str, str] = {}
|
||
|
|
self.ttls: dict[str, int] = {}
|
||
|
|
|
||
|
|
async def get(self, key: str) -> CacheReadResult:
|
||
|
|
return CacheReadResult(self.values.get(key))
|
||
|
|
|
||
|
|
async def set(self, key: str, value: str, ttl_seconds: int) -> bool:
|
||
|
|
self.values[key] = value
|
||
|
|
self.ttls[key] = ttl_seconds
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
class Adapter:
|
||
|
|
calls = 0
|
||
|
|
|
||
|
|
async def fetch_names(self, codes: list[str]) -> dict[str, str]:
|
||
|
|
self.calls += 1
|
||
|
|
return {code: "测试基金" for code in codes}
|
||
|
|
|
||
|
|
async def fetch_quotes(self, codes: list[str]) -> dict[str, dict[str, object]]:
|
||
|
|
self.calls += 1
|
||
|
|
return {code: {"nav": Decimal("1.2")} for code in codes}
|
||
|
|
|
||
|
|
async def fetch_history(self, code: str, target_date: object) -> dict[str, object]:
|
||
|
|
self.calls += 1
|
||
|
|
return {"nav": Decimal("1.1"), "nav_date": target_date, "degraded": False}
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_service_reads_and_writes_cache() -> None:
|
||
|
|
client = MemoryClient()
|
||
|
|
adapter = Adapter()
|
||
|
|
service = FundQuoteService(adapter, FundQuoteCache(client))
|
||
|
|
query = FundQuoteQuery(fund_codes=("159511",))
|
||
|
|
first = await service.query(query, now=datetime(2026, 9, 9, 8, tzinfo=UTC))
|
||
|
|
second = await service.query(query, now=datetime(2026, 9, 9, 8, tzinfo=UTC))
|
||
|
|
assert first[0].quote_source == "eastmoney"
|
||
|
|
assert second[0].quote_source == "cache"
|
||
|
|
assert adapter.calls == 2
|
||
|
|
assert next(iter(client.ttls.values())) == 900
|