259e78e175
Defines RelayTransport (lifecycle/handshake/inbound/outbound/interrupt) as the gateway<->connector wire contract; RelayAdapter.connect now registers an inbound handler that bridges connector-delivered MessageEvents into handle_message. Adds an in-memory StubConnector under tests/ and an E2E round-trip proving: connect registers the handler, inbound events reach the adapter, guild_id drives build_session_key isolation (two guilds -> two keys; same guild/channel/user -> one), outbound send round-trips, get_chat_info is proxied. Phase 1, Task 1.2 of the gateway-relay plan.
67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
"""Test-only in-memory stub connector implementing RelayTransport.
|
|
|
|
MUST stay under tests/ — never under plugins/ or gateway/ (a CI guard in
|
|
test_no_stub_leak.py asserts this). It lets Phase 1 prove the gateway side of
|
|
the relay end-to-end with zero dependency on the real (Node) connector.
|
|
|
|
The stub:
|
|
- hands back a fixed CapabilityDescriptor at handshake,
|
|
- lets a test push synthetic inbound MessageEvents (push_inbound),
|
|
- records every outbound action (sent/interrupts) for assertions,
|
|
- answers get_chat_info from a small fixture map.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from gateway.platforms.base import MessageEvent
|
|
from gateway.relay.descriptor import CapabilityDescriptor
|
|
from gateway.relay.transport import InboundHandler
|
|
|
|
|
|
class StubConnector:
|
|
"""In-memory RelayTransport for tests."""
|
|
|
|
def __init__(self, descriptor: CapabilityDescriptor) -> None:
|
|
self._descriptor = descriptor
|
|
self._inbound: Optional[InboundHandler] = None
|
|
self.connected = False
|
|
self.sent: List[Dict[str, Any]] = []
|
|
self.interrupts: List[Dict[str, Any]] = []
|
|
self.chat_info: Dict[str, Dict[str, Any]] = {}
|
|
# Canned result for the next send_outbound (override per-test).
|
|
self.next_send_result: Dict[str, Any] = {"success": True, "message_id": "m1"}
|
|
|
|
async def connect(self) -> bool:
|
|
self.connected = True
|
|
return True
|
|
|
|
async def disconnect(self) -> None:
|
|
self.connected = False
|
|
|
|
async def handshake(self) -> CapabilityDescriptor:
|
|
return self._descriptor
|
|
|
|
def set_inbound_handler(self, handler: InboundHandler) -> None:
|
|
self._inbound = handler
|
|
|
|
async def send_outbound(self, action: Dict[str, Any]) -> Dict[str, Any]:
|
|
self.sent.append(action)
|
|
if action.get("op") == "send":
|
|
return dict(self.next_send_result)
|
|
return {"success": True}
|
|
|
|
async def get_chat_info(self, chat_id: str) -> Dict[str, Any]:
|
|
return self.chat_info.get(chat_id, {"name": chat_id, "type": "dm"})
|
|
|
|
async def send_interrupt(self, session_key: str, reason: Optional[str] = None) -> None:
|
|
self.interrupts.append({"session_key": session_key, "reason": reason})
|
|
|
|
# ── test driver ──────────────────────────────────────────────────────
|
|
async def push_inbound(self, event: MessageEvent) -> None:
|
|
"""Simulate the connector delivering a normalized inbound event."""
|
|
if self._inbound is None:
|
|
raise RuntimeError("no inbound handler registered (call adapter.connect first)")
|
|
await self._inbound(event)
|