feat: add ops (validation + has_unread/join/delete)

This commit is contained in:
lda
2026-09-01 01:47:55 +07:00 Verified
parent 6da8bdbf50
commit 07f650cafa
2 changed files with 92 additions and 0 deletions
+36
View File
@@ -0,0 +1,36 @@
from agentmsgs.stores.memory import InMemoryStore
from agentmsgs.core import ops
def test_ops_validates_sender_must_be_in_thread():
s = InMemoryStore()
a = ops.get_or_create_agent(s, "A"); b = ops.get_or_create_agent(s, "B"); c = ops.get_or_create_agent(s, "C")
t = ops.create_thread(s, {a,b})
try:
ops.append_message(s, t.id, c, "oops")
assert False, "should raise"
except ValueError as e:
assert "not in thread" in str(e)
def test_ops_has_unread_exclude_own():
s = InMemoryStore()
a = ops.get_or_create_agent(s, "A"); b = ops.get_or_create_agent(s, "B")
t = ops.create_thread(s, {a,b})
ops.append_message(s, t.id, a, "a1")
# b has unread, a's own message shouldn't count if exclude_own
assert ops.has_unread(s, t.id, b) is True
assert ops.has_unread(s, t.id, a, exclude_own=True) is False
assert ops.has_unread(s, t.id, a, exclude_own=False) is True
ops.mark_read(s, t.id, b, 1)
assert ops.has_unread(s, t.id, b) is False
def test_ops_join_and_delete():
s = InMemoryStore()
a = ops.get_or_create_agent(s, "A"); b = ops.get_or_create_agent(s, "B"); c = ops.get_or_create_agent(s, "C")
t = ops.create_thread(s, {a,b})
t2 = ops.join_thread(s, t.id, c)
assert c in t2.participants
ops.delete_agent(s, c.id)
assert s.get_agent_by_id(c.id) is None