mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-03 13:00:37 +00:00
Rohit C PrasadandDevika
2b45018ffa
Imported from andrewyng/aisuite@1b4bbf303e (contents of its platform/ directory, hoisted to the repo root). Development history prior to this commit lives in that repository. Co-authored-by: Devika <devikaverma11@gmail.com>
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
"""Phase 3 wiring — inbound Slack/Telegram replies correlate to Inbox items via the gateway.
|
|
|
|
An inbound message carrying an `[ocw:<id>]` token is consumed as an Inbox reply (resolving the
|
|
item + releasing any suspended agent), not routed to the super-agent as a new turn. A normal
|
|
message still goes to the handler."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from coworker.connectors import ConnectorSettings, FakeAdapter, Gateway, MessageEvent
|
|
from coworker.inbox import InboxStore
|
|
from coworker.inbox_routing import resolve_from_reply
|
|
|
|
|
|
async def test_inbound_reply_resolves_item_and_is_not_routed(tmp_path):
|
|
inbox = InboxStore(tmp_path / "inbox.json")
|
|
item = inbox.add_approval("s1", "Restart service?", inbox="ops")
|
|
|
|
routed: list[MessageEvent] = []
|
|
|
|
async def handler(ev: MessageEvent) -> None:
|
|
routed.append(ev)
|
|
|
|
def reply_resolver(ev: MessageEvent) -> bool:
|
|
return resolve_from_reply(ev.text, inbox.resolve) is not None
|
|
|
|
settings = {"fake": ConnectorSettings("fake", enabled=True, allowed_users={"u1"})}
|
|
gw = Gateway(settings=settings, handler=handler, reply_resolver=reply_resolver)
|
|
fake = FakeAdapter()
|
|
gw.register(fake)
|
|
await gw.start()
|
|
|
|
# An inbound approval reply: resolves the item, NOT routed to the handler.
|
|
await fake.inject(f"approve [ocw:{item.id}]", user_id="u1")
|
|
assert inbox.get(item.id).resolution == "allow"
|
|
assert routed == []
|
|
|
|
# A normal message (no token) still goes to the handler.
|
|
await fake.inject("hey, what's up?", user_id="u1")
|
|
assert len(routed) == 1 and routed[0].text == "hey, what's up?"
|
|
|
|
await gw.stop()
|
|
|
|
|
|
async def test_freetext_answer_to_question_is_consumed(tmp_path):
|
|
inbox = InboxStore(tmp_path / "inbox.json")
|
|
q = inbox.add_question("s1", "Which region?", inbox="ops")
|
|
routed: list = []
|
|
|
|
async def handler(ev):
|
|
routed.append(ev)
|
|
|
|
settings = {"fake": ConnectorSettings("fake", enabled=True, allow_all=True)}
|
|
gw = Gateway(
|
|
settings=settings,
|
|
handler=handler,
|
|
reply_resolver=lambda ev: resolve_from_reply(ev.text, inbox.resolve)
|
|
is not None,
|
|
)
|
|
fake = FakeAdapter()
|
|
gw.register(fake)
|
|
await gw.start()
|
|
|
|
await fake.inject(f"us-west-2 [ocw:{q.id}]", user_id="anyone")
|
|
assert inbox.get(q.id).resolution == "us-west-2"
|
|
assert routed == []
|
|
await gw.stop()
|