From b45bff9db189ee1f7e412e4a61257e089b04819d Mon Sep 17 00:00:00 2001
From: Igor Lins e Silva <4753812+igorls@users.noreply.github.com>
Date: Tue, 7 Apr 2026 17:30:51 -0300
Subject: [PATCH] test: add WAL mode and entity timeline limit assertions

---
 tests/test_knowledge_graph.py | 17 +++++++++++++++++
 uv.lock                       | 10 ++++++++++
 2 files changed, 27 insertions(+)

diff --git a/tests/test_knowledge_graph.py b/tests/test_knowledge_graph.py
index a8fcd9a..d7d9838 100644
--- a/tests/test_knowledge_graph.py
+++ b/tests/test_knowledge_graph.py
@@ -107,6 +107,23 @@ class TestTimeline:
         tl = kg.timeline()
         assert len(tl) == 100  # LIMIT 100
 
+    def test_timeline_entity_has_limit(self, kg):
+        # Add > 100 triples all connected to a single entity
+        for i in range(105):
+            kg.add_triple(
+                "hub", "connects_to", f"spoke_{i}", valid_from=f"2025-01-{(i % 28) + 1:02d}"
+            )
+        tl = kg.timeline("hub")
+        assert len(tl) == 100  # LIMIT 100 on entity-filtered branch
+
+
+class TestWALMode:
+    def test_wal_mode_enabled(self, kg):
+        conn = kg._conn()
+        mode = conn.execute("PRAGMA journal_mode").fetchone()[0]
+        conn.close()
+        assert mode == "wal"
+
 
 class TestStats:
     def test_stats_empty(self, kg):
diff --git a/uv.lock b/uv.lock
index 0b00717..9d99313 100644
--- a/uv.lock
+++ b/uv.lock
@@ -966,6 +966,13 @@ dependencies = [
     { name = "pyyaml" },
 ]
 
+[package.optional-dependencies]
+dev = [
+    { name = "pytest", version = "8.4.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.10'" },
+    { name = "pytest", version = "9.0.3", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.10'" },
+    { name = "ruff" },
+]
+
 [package.dev-dependencies]
 dev = [
     { name = "pytest", version = "8.4.2", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.10'" },
@@ -976,8 +983,11 @@ dev = [
 [package.metadata]
 requires-dist = [
     { name = "chromadb", specifier = ">=0.4.0,<1" },
+    { name = "pytest", marker = "extra == 'dev'", specifier = ">=7.0" },
     { name = "pyyaml", specifier = ">=6.0" },
+    { name = "ruff", marker = "extra == 'dev'", specifier = ">=0.4.0" },
 ]
+provides-extras = ["dev"]
 
 [package.metadata.requires-dev]
 dev = [