1 files changed, 91 insertions, 0 deletions
diff --git a/packages/meshbay-node/tests/test_chat_store.py b/packages/meshbay-node/tests/test_chat_store.py
new file mode 100644
index 0000000..d74310d
--- /dev/null
+++ b/packages/meshbay-node/tests/test_chat_store.py
@@ -0,0 +1,91 @@
+"""
+Tests for the SQLite-backed chat message store.
+"""
+
+import pytest
+import pytest_asyncio
+from pathlib import Path
+
+from meshbay_node.chat.store import ChatStore
+
+
+@pytest_asyncio.fixture
+async def store(tmp_path):
+    s = ChatStore(db_path=tmp_path / "test_chat.db")
+    await s.open()
+    yield s
+    await s.close()
+
+
+@pytest.mark.asyncio
+async def test_save_and_retrieve(store):
+    row_id = await store.save_message(
+        sender_id="alice", iteration=0, payload=b"hello",
+    )
+    assert row_id == 1
+
+    msgs = await store.get_messages()
+    assert len(msgs) == 1
+    assert msgs[0].sender_id == "alice"
+    assert msgs[0].iteration == 0
+    assert msgs[0].payload == b"hello"
+    assert msgs[0].thread_id is None
+
+
+@pytest.mark.asyncio
+async def test_message_count(store):
+    assert await store.message_count() == 0
+    await store.save_message("alice", 0, b"msg1")
+    await store.save_message("bob", 1, b"msg2")
+    assert await store.message_count() == 2
+
+
+@pytest.mark.asyncio
+async def test_get_messages_since(store):
+    await store.save_message("alice", 0, b"old")
+    all_msgs = await store.get_messages()
+    cutoff = all_msgs[0].timestamp
+    await store.save_message("bob", 1, b"new")
+
+    msgs = await store.get_messages(since=cutoff)
+    assert len(msgs) == 1
+    assert msgs[0].sender_id == "bob"
+
+
+@pytest.mark.asyncio
+async def test_thread_messages(store):
+    await store.save_message("alice", 0, b"root", thread_id="t1")
+    await store.save_message("bob", 1, b"reply", thread_id="t1")
+    await store.save_message("carol", 2, b"other")
+
+    thread = await store.get_thread("t1")
+    assert len(thread) == 2
+    assert thread[0].sender_id == "alice"
+    assert thread[1].sender_id == "bob"
+
+
+@pytest.mark.asyncio
+async def test_message_ordering(store):
+    for i in range(5):
+        await store.save_message(f"user-{i}", i, f"msg-{i}".encode())
+
+    msgs = await store.get_messages()
+    assert len(msgs) == 5
+    for i, m in enumerate(msgs):
+        assert m.sender_id == f"user-{i}"
+
+
+@pytest.mark.asyncio
+async def test_limit(store):
+    for i in range(10):
+        await store.save_message("alice", i, f"msg-{i}".encode())
+
+    msgs = await store.get_messages(limit=3)
+    assert len(msgs) == 3
+
+
+@pytest.mark.asyncio
+async def test_context_manager(tmp_path):
+    async with ChatStore(db_path=tmp_path / "ctx_test.db") as store:
+        await store.save_message("alice", 0, b"test")
+        assert await store.message_count() == 1