mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-11-12 14:00:35 +00:00
91f4fb4570
Signed-off-by: PeaceMaker-best <221849497+PeaceMaker-best@users.noreply.github.com> Co-authored-by: PeaceMaker-best <221849497+PeaceMaker-best@users.noreply.github.com>
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""Operational startup failures must not be treated as index corruption."""
|
|
|
|
import sqlite3
|
|
|
|
import pytest
|
|
|
|
from deerflow.agents.memory.backends.deermem.deermem.config import DeerMemConfig
|
|
from deerflow.agents.memory.backends.deermem.deermem.core import retrieval
|
|
|
|
|
|
@pytest.mark.parametrize("code", [sqlite3.SQLITE_BUSY, sqlite3.SQLITE_LOCKED, sqlite3.SQLITE_READONLY, sqlite3.SQLITE_IOERR, sqlite3.SQLITE_ERROR])
|
|
def test_operational_startup_failure_preserves_existing_index(tmp_path, monkeypatch, code):
|
|
config = DeerMemConfig(storage_path=str(tmp_path))
|
|
adapter = retrieval.create_fts5_retrieval(config)
|
|
adapter._engine.index_fact("existing", "remember deployment preference")
|
|
adapter.close()
|
|
index = tmp_path / ".retrieval" / "memory-fts5.sqlite3"
|
|
original = index.read_bytes()
|
|
attempts = []
|
|
|
|
def unavailable(self):
|
|
attempts.append(self)
|
|
error = sqlite3.OperationalError("temporary startup failure")
|
|
error.sqlite_errorcode = code
|
|
raise error
|
|
|
|
with monkeypatch.context() as patched:
|
|
patched.setattr(retrieval.FTS5Retrieval, "_init_schema", unavailable)
|
|
assert retrieval.create_fts5_retrieval(config) is None
|
|
|
|
assert index.read_bytes() == original
|
|
assert len(attempts) == 1
|
|
recovered = retrieval.FTS5Retrieval(index)
|
|
try:
|
|
assert [fact["id"] for fact in recovered.search("deployment")] == ["existing"]
|
|
finally:
|
|
recovered.close()
|