diff --git a/mcp-server/tests/test_idempotent_ingest.py b/mcp-server/tests/test_idempotent_ingest.py index a78b754..7f732a5 100644 --- a/mcp-server/tests/test_idempotent_ingest.py +++ b/mcp-server/tests/test_idempotent_ingest.py @@ -110,6 +110,9 @@ def test_ingest_calls_recompute_searchable(monkeypatch, tmp_path): async def _recompute(cid): calls["recompute"].append(cid) monkeypatch.setattr(ingest.db, "recompute_searchable", _recompute) + + async def _mark_indexed(cid): return None + monkeypatch.setattr(ingest.db, "mark_indexed", _mark_indexed) monkeypatch.setattr(ingest.config, "PARENT_DOC_RETRIEVAL_ENABLED", False) monkeypatch.setattr(ingest.config, "MULTIMODAL_ENABLED", False) diff --git a/mcp-server/tests/test_unified_ingest.py b/mcp-server/tests/test_unified_ingest.py index 0456e51..d6a78af 100644 --- a/mcp-server/tests/test_unified_ingest.py +++ b/mcp-server/tests/test_unified_ingest.py @@ -87,6 +87,7 @@ def patched(monkeypatch, tmp_path): monkeypatch.setattr(db, "set_case_law_extraction_status", _set_status) monkeypatch.setattr(db, "set_case_law_halacha_status", _set_status) monkeypatch.setattr(db, "recompute_searchable", _recompute_searchable) + monkeypatch.setattr(db, "mark_indexed", _set_status) # Force flat chunking + multimodal OFF unless a test flips it. monkeypatch.setattr(config, "PARENT_DOC_RETRIEVAL_ENABLED", False) monkeypatch.setattr(config, "MULTIMODAL_ENABLED", False)