feat: persist reviewed chunks and index publication versions

This commit is contained in:
Backend Contributors
2026-10-16 08:44:30 +08:00
parent f9827feb1b
commit 2722a59335
20 changed files with 1027 additions and 16 deletions
+27 -2
View File
@@ -16,7 +16,7 @@ def test_startup_requires_migration(database):
def test_startup_rejects_previous_schema(database):
migrate(database, "0003_retrieval_runs")
migrate(database, "0004_auth_sessions")
with pytest.raises(RuntimeError, match="alembic upgrade head"):
with TestClient(create_app(database)):
pass
@@ -42,7 +42,7 @@ def test_upgrade_preserves_original_rows_and_repeated_upgrade(database):
assert connection.scalar(text("SELECT count(*) FROM audit_events")) == 2
assert (
connection.scalar(text("SELECT version_num FROM alembic_version"))
== "0004_auth_sessions"
== "0005_chunk_indexes"
)
assert compare_metadata(MigrationContext.configure(connection), Base.metadata) == []
migrate(database, "0001_baseline", downgrade=True)
@@ -73,3 +73,28 @@ def test_auth_upgrade_and_downgrade_preserve_knowledge_and_retrieval(database):
assert c.scalar(text("SELECT content FROM knowledge_items")) == "保留的占位正文"
assert c.scalar(text("SELECT count(*) FROM retrieval_runs")) == 1
migrate(database)
def test_chunk_index_downgrade_preserves_users_sessions_knowledge_and_retrieval(
client, migrated_database
):
from test_indexing import build, publish
from test_retrieval import add_item, search
item = add_item(client)
run = search(client)
version = build(client)
assert publish(client, version).status_code == 200
migrate(migrated_database, "0004_auth_sessions", downgrade=True)
with migrated_database.connect() as c:
assert c.scalar(text("SELECT count(*) FROM users")) == 1
assert c.scalar(text("SELECT count(*) FROM auth_sessions")) == 1
assert (
c.scalar(text("SELECT content FROM knowledge_items WHERE id=:id"), {"id": item["id"]})
== item["content"]
)
assert c.scalar(text("SELECT id FROM retrieval_runs")) == run["run_id"]
migrate(migrated_database)
assert client.get("/api/auth/me").status_code == 200
assert client.get("/api/indexes").json()["total"] == 0
assert client.get("/api/indexes/active").json()["generation"] == 0