haiku.rag/tests/store
Yiorgis Gozadinos 2da294b850
Put multi-table writes behind one transaction boundary
Four call sites repeated lock, snapshot, try/except, restore. Each used
restore_table_versions, which restores in _tables() order — documents
first, contradicting RESTORE_TABLE_ORDER — and each caught Exception, so
a cancellation mid-write skipped rollback and left the earlier table
writes committed.

Store.write_transaction() holds the lock, snapshots under it, and rolls
back through _rollback_to_snapshot: RESTORE_TABLE_ORDER, shielded from
cancellation, absorbed cancellation re-delivered, rollback failure raised
with the original as cause. The two single-table update_meta sites keep
the bare lock.

The batch documents write moves inside the guarded body; it was outside
the try, so a failure there was never rolled back. Auto-vacuum is
scheduled after the transaction rather than inside it.

restore_table_versions is removed; those four sites were its only callers.
2026-08-19 14:05:40 +03:00
..
__init__.py Split meta document attributes into a document_meta table 2026-06-12 10:17:17 +03:00
legacy_documents.py Serialize and roll back the multi-table document delete 2026-06-12 10:17:18 +03:00
test_citation.py Expose all chunk metadata on search results and citations through SearchResult.chunk_meta and Citation.chunk_meta 2026-08-17 19:42:23 +02:00
test_compression.py Collapse docling compression to a single function 2026-06-24 10:43:27 +03:00
test_database_info.py Extract gather_database_info shared by the info command 2026-06-09 11:09:28 +03:00
test_document_item_grouped.py Stop the reranker fetch reading text it discards 2026-08-18 18:00:00 +03:00
test_document_items.py Remove the single-document item accessors the batching replaced 2026-08-18 17:12:10 +03:00
test_document_meta_split.py Rename document_meta identity column document_id to id 2026-07-08 16:32:55 +03:00
test_engine.py Similar fix for haiku.rag.ingester, and formatting 2026-08-12 15:20:11 +01:00
test_indexes.py Trim comments, docstrings and docs to what they need to say 2026-08-18 12:38:02 +03:00
test_migrations.py Stop eagerly importing Store from haiku.rag.store 2026-08-11 17:26:51 +01:00
test_open_path.py Read the table list and settings once per open 2026-08-18 14:58:17 +03:00
test_read_only.py Put multi-table writes behind one transaction boundary 2026-08-19 14:05:40 +03:00
test_restore.py Stop eagerly importing Store from haiku.rag.store 2026-08-11 17:26:51 +01:00
test_tags.py Stop eagerly importing Store from haiku.rag.store 2026-08-11 17:26:51 +01:00
test_v0_40_0_migration.py Similar fix for haiku.rag.ingester, and formatting 2026-08-12 15:20:11 +01:00
test_v0_45_0_migration.py Stop eagerly importing Store from haiku.rag.store 2026-08-11 17:26:51 +01:00
test_v0_48_0_migration.py Collapse docling compression to a single function 2026-06-24 10:43:27 +03:00
test_v0_50_0_migration.py Reach and enforce 100% coverage 2026-07-26 20:11:02 +03:00
test_v0_58_0_migration.py Load document blobs only when asked 2026-08-06 13:04:51 +03:00
test_v0_64_0_migration.py Rename document_meta identity column document_id to id 2026-07-08 16:32:55 +03:00
test_v0_75_0_migration.py Trim comments, docstrings and docs to what they need to say 2026-08-18 12:38:02 +03:00