62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
"""Snapshot scheduler tests (REQ-030 AC-1)."""
|
|
from __future__ import annotations
|
|
|
|
from trex_core.adapters.broker.paper import PaperBroker
|
|
from trex_core.adapters.portfolio.stub_read import StubPortfolioRead
|
|
from trex_core.application.portfolio import PortfolioItem, SnapshotStore
|
|
from trex_core.application.snapshot_scheduler import SnapshotScheduler
|
|
|
|
|
|
def test_run_once_fetches_universe_and_swaps_store():
|
|
read = StubPortfolioRead(daemon_id="d-1", items=[
|
|
PortfolioItem(portfolio_id="pf-X", symbol="X",
|
|
cash_balance=100, asset_quantity=0),
|
|
])
|
|
store = SnapshotStore()
|
|
sched = SnapshotScheduler(read_port=read, store=store, broker=PaperBroker())
|
|
assert sched.run_once() is True
|
|
pids = {a.item.portfolio_id for a in store.list_assets()}
|
|
assert pids == {"pf-X"}
|
|
# universe membership + strategy params come from fetch (baseline)
|
|
# cash/quantity are baseline values (option D will overwrite later)
|
|
a = store.get("pf-X")
|
|
assert a.item.cash_balance == 100
|
|
assert a.warning is False
|
|
|
|
|
|
def test_run_once_failure_sets_degraded_signal():
|
|
class BadRead:
|
|
def whoami(self): return "d-2"
|
|
def get_portfolios(self, daemon_id):
|
|
from trex_core.application.portfolio import PortfolioHubUnavailableError
|
|
raise PortfolioHubUnavailableError("simulated")
|
|
store = SnapshotStore()
|
|
sched = SnapshotScheduler(read_port=BadRead(), store=store, broker=PaperBroker())
|
|
assert sched.run_once() is False
|
|
reason, msg = store.degraded_state()
|
|
assert reason is not None and reason.value == "portfolio_hub_unavailable"
|
|
|
|
|
|
def test_swap_preserves_warning_for_carried_over_portfolio():
|
|
store = SnapshotStore()
|
|
store.swap([PortfolioItem(portfolio_id="pf-X", symbol="X",
|
|
cash_balance=100, asset_quantity=0)])
|
|
store.set_warning("pf-X", True)
|
|
# second swap includes the same portfolio with refreshed cash
|
|
store.swap([PortfolioItem(portfolio_id="pf-X", symbol="X",
|
|
cash_balance=200, asset_quantity=0)])
|
|
a = store.get("pf-X")
|
|
assert a.warning is True
|
|
assert a.item.cash_balance == 200
|
|
|
|
|
|
def test_option_d_update_cash_position_overrides_baseline():
|
|
"""REQ-030 wiring — when *_after lands we update the authoritative values."""
|
|
store = SnapshotStore()
|
|
store.swap([PortfolioItem(portfolio_id="pf-X", symbol="X",
|
|
cash_balance=100, asset_quantity=0)])
|
|
store.update_cash_position("pf-X", cash_balance=92.5, asset_quantity=1)
|
|
a = store.get("pf-X")
|
|
assert a.item.cash_balance == 92.5
|
|
assert a.item.asset_quantity == 1
|