93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
"""Snapshot RPC handler envelope path (REQ-028 AC-1/AC-2/AC-3)."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import date
|
|
|
|
from trex_core.application.messaging import RpcRequest
|
|
from trex_core.application.messaging_handlers import make_snapshot_rpc_handler
|
|
from trex_core.application.portfolio import (
|
|
DegradedReason,
|
|
InMemoryPauseStore,
|
|
PortfolioItem,
|
|
SnapshotStore,
|
|
)
|
|
from trex_core.domain.slot import SlotService
|
|
from trex_core.domain.slot.in_memory_repository import InMemorySlotRepository
|
|
|
|
DAY = date(2026, 6, 5)
|
|
|
|
|
|
def _wire(items=None):
|
|
repo = InMemorySlotRepository()
|
|
svc = SlotService(repo)
|
|
svc.preallocate_daily_slots(DAY, 5)
|
|
store = SnapshotStore()
|
|
store.swap(items or [
|
|
PortfolioItem(portfolio_id="pf-A", symbol="005930",
|
|
cash_balance=1_000_000, asset_quantity=0,
|
|
active_strategy_id="target-value-v1",
|
|
strategy_params={"k": 1}),
|
|
])
|
|
pause = InMemoryPauseStore()
|
|
handler = make_snapshot_rpc_handler(
|
|
repo, svc,
|
|
snapshot_store=store, pause_store=pause, default_total_slots=5,
|
|
)
|
|
return store, pause, handler
|
|
|
|
|
|
def test_envelope_shape_returned():
|
|
_, _, handler = _wire()
|
|
resp = handler(RpcRequest(
|
|
payload={"view": "all", "trading_date": DAY.isoformat()},
|
|
correlation_id="snap-1",
|
|
))
|
|
body = resp.payload
|
|
assert set(body.keys()) == {"assets", "view", "available_slots", "degraded", "reason", "message"}
|
|
assert isinstance(body["assets"], list)
|
|
assert body["view"] == "all"
|
|
assert body["degraded"] is False
|
|
assert body["reason"] is None
|
|
|
|
|
|
def test_envelope_strategy_key_filter():
|
|
_, _, handler = _wire([
|
|
PortfolioItem(portfolio_id="x", symbol="X", cash_balance=0, asset_quantity=0,
|
|
active_strategy_id="alpha"),
|
|
PortfolioItem(portfolio_id="y", symbol="Y", cash_balance=0, asset_quantity=0,
|
|
active_strategy_id="beta"),
|
|
])
|
|
resp = handler(RpcRequest(
|
|
payload={"view": "all", "strategy_key": "alpha"}, correlation_id="snap-2",
|
|
))
|
|
pids = {a["asset_id"] for a in resp.payload["assets"]}
|
|
assert pids == {"x"}
|
|
|
|
|
|
def test_degraded_signal_does_not_leak_credentials():
|
|
store, _, handler = _wire()
|
|
store.set_degraded(
|
|
DegradedReason.PORTFOLIO_HUB_UNAVAILABLE,
|
|
"GET https://hub failed: 401 Bearer secret-token-abc",
|
|
)
|
|
resp = handler(RpcRequest(payload={"view": "all"}, correlation_id="snap-3"))
|
|
body = resp.payload
|
|
assert body["degraded"] is True
|
|
assert body["reason"] == "portfolio_hub_unavailable"
|
|
assert body["message"] is not None
|
|
assert "secret-token-abc" not in body["message"]
|
|
|
|
|
|
def test_bat007_legacy_path_when_stores_unset():
|
|
"""Make sure pre-BAT-008 callers still work (slots+capacity stub shape)."""
|
|
repo = InMemorySlotRepository()
|
|
svc = SlotService(repo)
|
|
svc.preallocate_daily_slots(DAY, 3)
|
|
handler = make_snapshot_rpc_handler(repo, svc)
|
|
resp = handler(RpcRequest(
|
|
payload={"view": "all", "trading_date": DAY.isoformat()}, correlation_id="legacy",
|
|
))
|
|
body = resp.payload
|
|
assert "slots" in body and "capacity" in body
|
|
assert "assets" not in body
|