74 lines
3.0 KiB
Python
74 lines
3.0 KiB
Python
"""Phase 1: HTTP-inbound /p/<profile>/ routing for the webhook adapter."""
|
|
import pytest
|
|
|
|
from gateway.config import GatewayConfig, Platform
|
|
from gateway.session import SessionSource, build_session_key
|
|
|
|
|
|
class TestSessionSourceProfileField:
|
|
def test_profile_roundtrips(self):
|
|
s = SessionSource(
|
|
platform=Platform.WEBHOOK if hasattr(Platform, "WEBHOOK") else Platform.TELEGRAM,
|
|
chat_id="c1",
|
|
chat_type="webhook",
|
|
profile="coder",
|
|
)
|
|
restored = SessionSource.from_dict(s.to_dict())
|
|
assert restored.profile == "coder"
|
|
|
|
def test_profile_absent_not_serialized(self):
|
|
s = SessionSource(platform=Platform.TELEGRAM, chat_id="c1", chat_type="dm")
|
|
assert "profile" not in s.to_dict()
|
|
|
|
def test_source_profile_drives_session_key_namespace(self):
|
|
s = SessionSource(platform=Platform.TELEGRAM, chat_id="99", chat_type="dm")
|
|
# build_session_key takes profile explicitly; the adapter passes
|
|
# source.profile through. Verify the namespace follows it.
|
|
assert build_session_key(s, profile="coder") == "agent:coder:telegram:dm:99"
|
|
|
|
|
|
class TestWebhookProfileResolution:
|
|
"""_resolve_request_profile validates the /p/<profile>/ prefix."""
|
|
|
|
def _adapter(self, multiplex: bool, served=("default", "coder")):
|
|
from gateway.platforms.webhook import WebhookAdapter, _PROFILE_REJECTED
|
|
|
|
class _FakeReq:
|
|
def __init__(self, profile):
|
|
self.match_info = {"profile": profile} if profile is not None else {}
|
|
|
|
cfg = GatewayConfig(multiplex_profiles=multiplex)
|
|
|
|
class _Runner:
|
|
config = cfg
|
|
|
|
# Construct minimally; we only call _resolve_request_profile.
|
|
adapter = WebhookAdapter.__new__(WebhookAdapter)
|
|
adapter.gateway_runner = _Runner()
|
|
return adapter, _FakeReq, _PROFILE_REJECTED, served
|
|
|
|
def test_no_prefix_returns_none(self):
|
|
adapter, Req, _REJ, _ = self._adapter(multiplex=True)
|
|
assert adapter._resolve_request_profile(Req(None)) is None
|
|
|
|
def test_prefix_ignored_when_multiplex_off(self):
|
|
adapter, Req, _REJ, _ = self._adapter(multiplex=False)
|
|
# Even a bogus profile is ignored (not 404'd) when multiplexing is off.
|
|
assert adapter._resolve_request_profile(Req("anything")) is None
|
|
|
|
def test_known_profile_accepted(self, monkeypatch):
|
|
adapter, Req, _REJ, served = self._adapter(multiplex=True)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.profiles.profiles_to_serve",
|
|
lambda multiplex: [(n, None) for n in served],
|
|
)
|
|
assert adapter._resolve_request_profile(Req("coder")) == "coder"
|
|
|
|
def test_unknown_profile_rejected(self, monkeypatch):
|
|
adapter, Req, REJ, served = self._adapter(multiplex=True)
|
|
monkeypatch.setattr(
|
|
"hermes_cli.profiles.profiles_to_serve",
|
|
lambda multiplex: [(n, None) for n in served],
|
|
)
|
|
assert adapter._resolve_request_profile(Req("ghost")) is REJ
|