test: cover chat core and session storage

tests/unit/test_sqlite_store.py

@@ -0,0 +1,58 @@
+import tempfile
+import unittest
+from pathlib import Path
+
+from javis.memory.sqlite_store import SessionNotFoundError, SQLiteSessionStore
+
+
+class SQLiteSessionStoreTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self.temporary_directory = tempfile.TemporaryDirectory()
+        database = Path(self.temporary_directory.name) / "sessions.sqlite3"
+        self.store = SQLiteSessionStore(database)
+
+    def tearDown(self) -> None:
+        self.temporary_directory.cleanup()
+
+    def test_create_list_and_load_session(self) -> None:
+        created = self.store.create_session("ollama", "test-model")
+        loaded = self.store.get_session(created.id)
+
+        self.assertEqual(loaded.id, created.id)
+        self.assertEqual(loaded.provider, "ollama")
+        self.assertEqual(loaded.model, "test-model")
+        self.assertEqual(self.store.list_sessions()[0].id, created.id)
+
+    def test_messages_keep_exchange_order(self) -> None:
+        session = self.store.create_session("ollama", "test-model")
+        self.store.append_exchange(session.id, "eins", "zwei")
+        self.store.append_exchange(session.id, "drei", "vier")
+
+        messages = self.store.get_messages(session.id)
+
+        self.assertEqual(
+            [(message.role, message.content) for message in messages],
+            [
+                ("user", "eins"),
+                ("assistant", "zwei"),
+                ("user", "drei"),
+                ("assistant", "vier"),
+            ],
+        )
+
+    def test_unknown_session_raises_clear_error(self) -> None:
+        with self.assertRaises(SessionNotFoundError):
+            self.store.get_session("missing")
+
+    def test_clear_keeps_session_and_removes_messages(self) -> None:
+        session = self.store.create_session("ollama", "test-model")
+        self.store.append_exchange(session.id, "eins", "zwei")
+
+        self.store.clear_messages(session.id)
+
+        self.assertEqual(self.store.get_messages(session.id), [])
+        self.assertEqual(self.store.get_session(session.id).message_count, 0)
+
+
+if __name__ == "__main__":
+    unittest.main()