Merge pull request #36 from fourblocks2020/feat/slot-filled-no-route
fix(slot-filled): 매도 체결 이벤트 NO_ROUTE 라우팅 갭 해소 (M3)
This commit is contained in:
@@ -9,7 +9,7 @@ from trex_core.adapters.messaging import topology
|
||||
from trex_core.adapters.messaging.in_memory import InMemoryTransport
|
||||
from trex_core.adapters.persistence.in_memory_intent_store import InMemoryIntentStore
|
||||
from trex_core.application.execution import ExecutionService
|
||||
from trex_core.application.intents import BuyIntent, IntentService
|
||||
from trex_core.application.intents import BuyIntent, IntentService, SellIntent
|
||||
from trex_core.application.ledger_outbox import InMemoryLedgerOutbox, LedgerWriter
|
||||
from trex_core.application.messaging_handlers import slot_transition_event_payload
|
||||
from trex_core.domain.slot import SlotService, SlotState
|
||||
@@ -27,8 +27,11 @@ def _wire(transport):
|
||||
writer = LedgerWriter(outbox, ledger)
|
||||
repo.reconcile_universe("default", {"PF-1": "SYM"})
|
||||
|
||||
def emitter(event_type: str, slot) -> None:
|
||||
rk = topology.event_routing_key(event_type, slot.strategy_id or "_", slot.symbol or "_")
|
||||
def emitter(event_type: str, slot, routing_strategy_id: str | None = None) -> None:
|
||||
# Use routing_strategy_id hint when supplied (e.g. sell fills clear
|
||||
# slot.strategy_id via INACTIVE transition — NO_ROUTE fix).
|
||||
effective_sid = routing_strategy_id or getattr(slot, "strategy_id", None) or "_"
|
||||
rk = topology.event_routing_key(event_type, effective_sid, slot.symbol or "_")
|
||||
payload = slot_transition_event_payload(slot, event_type=event_type)
|
||||
try:
|
||||
transport.publish_event(rk, payload)
|
||||
@@ -124,3 +127,94 @@ def test_publish_failure_does_not_break_execution_loop():
|
||||
assert r.ledger_enqueued == 1
|
||||
pending = [s for s in repo.list_slots("default") if s.status is SlotState.HOLDING]
|
||||
assert len(pending) == 1
|
||||
|
||||
|
||||
def test_sell_fill_emits_slot_filled_sell_with_correct_strategy_routing():
|
||||
"""Regression: slot_filled_sell must route to strategy queue even after
|
||||
PENDING_SELL->INACTIVE clears strategy_id from the slot row.
|
||||
|
||||
M2 REL-009 feedback: slot_filled_sell was NO_ROUTEd (DLQ) because
|
||||
_on_filled used the post-transition slot (strategy_id=None) for the
|
||||
routing key, producing `slot_filled_sell._.{symbol}` which does not match
|
||||
the `*.{strategy_id}.#` binding pattern of trex.events.strategy.{sid}.
|
||||
"""
|
||||
transport = InMemoryTransport()
|
||||
received_sell: list[tuple[str, dict]] = []
|
||||
stop = threading.Event()
|
||||
th = threading.Thread(
|
||||
target=transport.consume_events_forever,
|
||||
args=("trex.events.strategy.A", topology.strategy_event_binding_pattern("A"),
|
||||
lambda rk, p: received_sell.append((rk, p)), stop),
|
||||
daemon=True,
|
||||
)
|
||||
th.start()
|
||||
time.sleep(0.05)
|
||||
|
||||
repo, svc, intents, execution, _ = _wire(transport)
|
||||
# Buy with strategy_id="A"
|
||||
d = intents.submit_buy(BuyIntent(
|
||||
intent_id="b1", qty=10, strategy_id="A", portfolio_id="PF-1",
|
||||
))
|
||||
execution.run_once() # buy fill -> holding
|
||||
slot = repo.fetch_slot(d.slot_id)
|
||||
assert slot.status is SlotState.HOLDING
|
||||
assert slot.strategy_id == "A"
|
||||
|
||||
# Sell — strategy_id_for_emit must be captured before INACTIVE clears it.
|
||||
intents.submit_sell(SellIntent(intent_id="s1", slot_id=d.slot_id))
|
||||
execution.run_once() # sell fill -> inactive
|
||||
|
||||
time.sleep(0.05)
|
||||
stop.set()
|
||||
th.join(timeout=1.0)
|
||||
|
||||
# The slot_filled_sell event must reach the strategy A queue.
|
||||
sell_events = [(rk, p) for rk, p in received_sell if "sell" in rk]
|
||||
assert len(sell_events) == 1, (
|
||||
f"Expected 1 slot_filled_sell event on strategy-A queue, got {len(sell_events)}. "
|
||||
f"All received: {[rk for rk, _ in received_sell]}"
|
||||
)
|
||||
rk, payload = sell_events[0]
|
||||
# Routing key must use strategy_id="A", not "_" (NO_ROUTE sentinel).
|
||||
assert rk == "slot_filled_sell.A.005930", (
|
||||
f"NO_ROUTE regression: routing key is '{rk}', expected 'slot_filled_sell.A.005930'"
|
||||
)
|
||||
assert payload["event_type"] == "slot_filled_sell"
|
||||
|
||||
|
||||
def test_sell_fill_dlq_sentinel_absent_when_strategy_id_set():
|
||||
"""Routing key must NOT contain '_' as strategy_id token when a real
|
||||
strategy_id was set on the slot (no DLQ sentinel in the key)."""
|
||||
transport = InMemoryTransport()
|
||||
all_published: list[str] = []
|
||||
|
||||
def emitter(event_type: str, slot, routing_strategy_id: str | None = None) -> None:
|
||||
effective_sid = routing_strategy_id or getattr(slot, "strategy_id", None) or "_"
|
||||
rk = topology.event_routing_key(event_type, effective_sid, slot.symbol or "_")
|
||||
all_published.append(rk)
|
||||
payload = slot_transition_event_payload(slot, event_type=event_type)
|
||||
try:
|
||||
transport.publish_event(rk, payload)
|
||||
except Exception: # noqa: BLE001
|
||||
pass
|
||||
|
||||
repo = InMemorySlotRepository()
|
||||
svc = SlotService(repo)
|
||||
intents = IntentService(svc, repo, InMemoryIntentStore(),
|
||||
resolve_symbol=lambda pid: "005930")
|
||||
from trex_core.application.ledger_outbox import InMemoryLedgerOutbox
|
||||
outbox = InMemoryLedgerOutbox()
|
||||
execution = ExecutionService(svc, repo, PaperBroker(fill_after_polls=0),
|
||||
ledger_outbox=outbox, event_emitter=emitter)
|
||||
repo.reconcile_universe("default", {"PF-1": "SYM"})
|
||||
|
||||
d = intents.submit_buy(BuyIntent(intent_id="b1", qty=10, strategy_id="orb-v1", portfolio_id="PF-1"))
|
||||
execution.run_once()
|
||||
intents.submit_sell(SellIntent(intent_id="s1", slot_id=d.slot_id))
|
||||
execution.run_once()
|
||||
|
||||
# Neither buy nor sell events should use "_" as strategy token.
|
||||
bad_keys = [rk for rk in all_published if "._.".encode() in rk.encode() or rk.split(".")[1] == "_"]
|
||||
assert bad_keys == [], (
|
||||
f"DLQ-sentinel routing keys found (strategy_id=None leak): {bad_keys}"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user