forked from Rockachopa/Timmy-time-dashboard
4e11dd2490
Move 97 test files from flat tests/ into 13 subdirectories: tests/dashboard/ (8 files — routes, mobile, mission control) tests/swarm/ (17 files — coordinator, docker, routing, tasks) tests/timmy/ (12 files — agent, backends, CLI, tools) tests/self_coding/ (14 files — git safety, indexer, self-modify) tests/lightning/ (3 files — L402, LND, interface) tests/creative/ (8 files — assembler, director, image/music/video) tests/integrations/ (10 files — chat bridge, telegram, voice, websocket) tests/mcp/ (4 files — bootstrap, discovery, executor) tests/spark/ (3 files — engine, tools, events) tests/hands/ (3 files — registry, oracle, phase5) tests/scripture/ (1 file) tests/infrastructure/ (3 files — router cascade, API) tests/security/ (3 files — XSS, regression) Fix Path(__file__) reference in test_mobile_scenarios.py for new depth. Add __init__.py to all test subdirectories. Tests: 1503 passed, 9 failed (pre-existing), 53 errors (pre-existing) https://claude.ai/code/session_019oMFNvD8uSGSSmBMGkBfQN
86 lines
2.0 KiB
Python
86 lines
2.0 KiB
Python
"""Tests for timmy_serve/inter_agent.py — agent-to-agent messaging."""
|
|
|
|
from timmy_serve.inter_agent import InterAgentMessenger
|
|
|
|
|
|
def test_send_message():
|
|
m = InterAgentMessenger()
|
|
msg = m.send("alice", "bob", "hello")
|
|
assert msg.from_agent == "alice"
|
|
assert msg.to_agent == "bob"
|
|
assert msg.content == "hello"
|
|
|
|
|
|
def test_receive_messages():
|
|
m = InterAgentMessenger()
|
|
m.send("alice", "bob", "msg1")
|
|
m.send("alice", "bob", "msg2")
|
|
msgs = m.receive("bob")
|
|
assert len(msgs) == 2
|
|
|
|
|
|
def test_pop_message():
|
|
m = InterAgentMessenger()
|
|
m.send("alice", "bob", "first")
|
|
m.send("alice", "bob", "second")
|
|
msg = m.pop("bob")
|
|
assert msg.content == "first"
|
|
remaining = m.receive("bob")
|
|
assert len(remaining) == 1
|
|
|
|
|
|
def test_pop_empty():
|
|
m = InterAgentMessenger()
|
|
assert m.pop("nobody") is None
|
|
|
|
|
|
def test_pop_all():
|
|
m = InterAgentMessenger()
|
|
m.send("a", "b", "1")
|
|
m.send("a", "b", "2")
|
|
msgs = m.pop_all("b")
|
|
assert len(msgs) == 2
|
|
assert m.receive("b") == []
|
|
|
|
|
|
def test_broadcast():
|
|
m = InterAgentMessenger()
|
|
# Create queues by sending initial messages
|
|
m.send("system", "agent1", "init")
|
|
m.send("system", "agent2", "init")
|
|
m.pop_all("agent1")
|
|
m.pop_all("agent2")
|
|
count = m.broadcast("system", "announcement")
|
|
assert count == 2
|
|
|
|
|
|
def test_history():
|
|
m = InterAgentMessenger()
|
|
m.send("a", "b", "1")
|
|
m.send("b", "a", "2")
|
|
history = m.history()
|
|
assert len(history) == 2
|
|
|
|
|
|
def test_clear_specific():
|
|
m = InterAgentMessenger()
|
|
m.send("a", "b", "msg")
|
|
m.clear("b")
|
|
assert m.receive("b") == []
|
|
|
|
|
|
def test_clear_all():
|
|
m = InterAgentMessenger()
|
|
m.send("a", "b", "msg")
|
|
m.clear()
|
|
assert m.history() == []
|
|
|
|
|
|
def test_max_queue_size():
|
|
m = InterAgentMessenger(max_queue_size=3)
|
|
for i in range(5):
|
|
m.send("a", "b", f"msg{i}")
|
|
msgs = m.receive("b")
|
|
assert len(msgs) == 3
|
|
assert msgs[0].content == "msg2" # oldest dropped
|