LLM/pentestagent
1
0
Fork 0
mirror of https://github.com/GH05TCREW/pentestagent.git synced 2026-03-07 14:23:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(rag): assert persisted index is loaded (mtime unchanged)

Browse Source
This commit is contained in:
giveen
2026-01-19 10:33:38 -07:00
parent 870cc4a84a
commit 2c82a30b16
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_rag_workspace_integration.py
View File

@@ -40,8 +40,14 @@ def test_rag_and_indexer_use_workspace(tmp_path, monkeypatch):
assert rag.get_chunk_count() >= 1
assert rag.get_document_count() >= 1
# Now create a new RAGEngine and ensure it loads persisted index automatically
# Now create a new RAGEngine and ensure it loads the persisted index instead of re-indexing
# Record the persisted index mtime so we can assert it is not overwritten by a re-index
mtime_before = emb_path.stat().st_mtime
rag2 = RAGEngine(use_local_embeddings=True)
# If load-on-init doesn't run, calling index() should load from saved file
# If load-on-init doesn't run, calling index() would re-index and rewrite the file
rag2.index()
assert rag2.get_chunk_count() >= 1
mtime_after = emb_path.stat().st_mtime
assert mtime_after == mtime_before, "Expected persisted index to be loaded, not re-written"