Files
trex-core/tests/adapters/messaging/test_instance_topology.py
T
genichinandClaude Opus 4.8 f0ad37747b feat(rel-007): dcy-025/024 + BAT-035/036 — slots.req 즉시 교체 + safety_sell_enabled
BAT-035 (dcy-025): trex.snapshot.req → trex.slots.req 즉시 교체
- topology: SNAPSHOT_REQ_QUEUE 제거, SLOTS_REQ_QUEUE = "trex.slots.req" 추가
- portfolio.py: View/build_snapshot_envelope/SnapshotEnvelope/_asset_to_dict 제거;
  slot_to_dict 시그니처 확장(snapshot_store/watchdog_store/pause_store/watchdog_timeout_sec)
  — asset_quantity(hub)/cash_balance/cash_estimated/warned/watchdog{}/paused 병합 (REQ-074)
- messaging_handlers.py: make_snapshot_rpc_handler 제거, make_slots_rpc_handler 추가
  — filter(all/inactive/pending/holding/error) + spy(bool) 처리; strategy 필드 제외 (REQ-073/074)
- daemon.py: _snapshot_rpc_loop → _slots_rpc_loop, make_slots_rpc_handler 연결
- cli.py: universe/slots → slots_queue + spy=True + server-side filter; buyable→inactive 매핑 (REQ-063)
- rabbitmq.py: snapshot_queue → slots_queue
- docs: interface-contract.md + runtime-flows.md slots.req 계약으로 업데이트 (REQ-075)
- REQ-028: Implemented → Deprecated (코드 삭제 완료)
- 테스트: test_snapshot_handler/selection/rpc_watchdog/messaging_snapshot_consumer 삭제;
  test_slots_handler.py(신규, 13 cases), test_killswitch_masking_resume 재작성,
  test_ledger_option_d._apply_cash_estimate → slots handler 경유, test_topology/cli_operator 갱신

BAT-036 (dcy-024): per-instance safety_sell_enabled 플래그
- config.py: DaemonInstanceConfig + DaemonConfig 에 safety_sell_enabled: bool = True 추가;
  from_toml 파싱, from_config 전파
- daemon.py: start() workers 조건부 — safety_sell_enabled=false 시 trex-safety-sell 미기동 + WARNING
- tests: test_instance_config.py 에 safety_sell_enabled 파싱 3개 케이스 추가

REQ: 073/074/075/021/034/063/048/005 Approved → Implemented

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-11 14:45:45 +09:00

115 lines
4.0 KiB
Python

"""Multi-instance topology tests (REQ-049/050 / BAT-013).
Builder/routing-key tests run always. declare_topology + directed/broadcast
routing tests are gated on TREX_RABBITMQ_TEST_URL.
"""
from __future__ import annotations
import os
import time
import uuid
import pytest
from trex_core.adapters.messaging import topology
from trex_core.adapters.messaging.rabbitmq import RabbitMQTransport
# ---- builders (always run) --------------------------------------------------
def test_named_instance_suffix():
"""REQ-049 AC-1."""
assert topology.intent_queue("inst-1") == "trex.intent.req.inst-1"
assert topology.slots_queue("inst-1") == "trex.slots.req.inst-1"
assert topology.control_queue("inst-1") == "trex.control.req.inst-1"
def test_default_is_literal():
"""REQ-049 AC-2: default keeps legacy literal names (backward-compat)."""
assert topology.intent_queue("default") == "trex.intent.req"
assert topology.slots_queue(None) == "trex.slots.req"
assert topology.control_queue() == "trex.control.req"
def test_direct_routing_key():
"""REQ-050 AC-3."""
assert topology.direct_routing_key("inst-1", "control") == "inst-1.control"
assert topology.direct_routing_key("inst-2", "intent") == "inst-2.intent"
# ---- RabbitMQ topology (gated) ----------------------------------------------
@pytest.fixture
def mq_url() -> str:
url = os.environ.get("TREX_RABBITMQ_TEST_URL")
if not url:
pytest.skip("TREX_RABBITMQ_TEST_URL not set; skipping RabbitMQ integration test")
return url
def _delete_instance_queues(url, instance_key):
import pika
conn = pika.BlockingConnection(pika.URLParameters(url))
try:
ch = conn.channel()
for q in (topology.intent_queue(instance_key), topology.slots_queue(instance_key),
topology.control_queue(instance_key)):
try:
ch.queue_delete(queue=q)
except Exception: # noqa: BLE001
pass
finally:
conn.close()
def test_directed_routing_isolation(mq_url):
"""REQ-050 AC-4 (S-2): inst-1 routing key reaches only inst-1's queue."""
import pika
k1 = "it1-" + uuid.uuid4().hex[:6]
k2 = "it2-" + uuid.uuid4().hex[:6]
t1 = RabbitMQTransport(mq_url)
t1.declare_topology(k1)
t2 = RabbitMQTransport(mq_url)
t2.declare_topology(k2)
try:
conn = pika.BlockingConnection(pika.URLParameters(mq_url))
ch = conn.channel()
ch.basic_publish(exchange=topology.DIRECT_EXCHANGE,
routing_key=topology.direct_routing_key(k1, "intent"), body=b"x")
time.sleep(0.4)
m1 = ch.basic_get(topology.intent_queue(k1), auto_ack=True)
m2 = ch.basic_get(topology.intent_queue(k2), auto_ack=True)
conn.close()
assert m1[0] is not None, "inst-1 intent queue should receive the directed message"
assert m2[0] is None, "inst-2 intent queue must NOT receive it"
finally:
t1.close()
t2.close()
_delete_instance_queues(mq_url, k1)
_delete_instance_queues(mq_url, k2)
def test_broadcast_dual_binding(mq_url):
"""REQ-050 AC-5 (S-3): broadcast reaches every instance's control queue."""
import pika
k1 = "bc1-" + uuid.uuid4().hex[:6]
k2 = "bc2-" + uuid.uuid4().hex[:6]
t1 = RabbitMQTransport(mq_url)
t1.declare_topology(k1)
t2 = RabbitMQTransport(mq_url)
t2.declare_topology(k2)
try:
conn = pika.BlockingConnection(pika.URLParameters(mq_url))
ch = conn.channel()
ch.basic_publish(exchange=topology.BROADCAST_EXCHANGE, routing_key="", body=b"halt")
time.sleep(0.4)
m1 = ch.basic_get(topology.control_queue(k1), auto_ack=True)
m2 = ch.basic_get(topology.control_queue(k2), auto_ack=True)
conn.close()
assert m1[0] is not None and m2[0] is not None, "both control queues should receive broadcast"
finally:
t1.close()
t2.close()
_delete_instance_queues(mq_url, k1)
_delete_instance_queues(mq_url, k2)