fz-mirrors/AstrBot
1
0
Fork 0
mirror of https://github.com/AstrBotDevs/AstrBot synced 2026-07-15 17:30:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: test_ensure_vec_db mock path and create kb_dir to avoid SQLite error

Browse Source
This commit is contained in:
LIghtJUNction
2026-04-29 04:18:18 +08:00
parent 585b871492
commit 4480bc20ac
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/unit/test_kb_manager_resilience.py
View File

@@ -257,6 +257,7 @@ async def test_ensure_vec_db_clears_stale_init_error(
helper.chunker = MagicMock()
helper.init_error = "Previous initialization failed"
helper.kb_dir = Path("/tmp/test_kb") / mock_knowledge_base.kb_id
helper.kb_dir.mkdir(parents=True, exist_ok=True)
helper.kb_medias_dir = helper.kb_dir / "medias" / mock_knowledge_base.kb_id
helper.kb_files_dir = helper.kb_dir / "files" / mock_knowledge_base.kb_id
@@ -266,7 +267,7 @@ async def test_ensure_vec_db_clears_stale_init_error(
mock_vec_db.close = AsyncMock()
with patch(
"astrbot.core.db.vec_db.faiss_impl.vec_db.FaissVecDB",
"astrbot.core.knowledge_base.kb_helper.FaissVecDB",
return_value=mock_vec_db,
):
# Execute _ensure_vec_db