From 4480bc20ac2271f2408cb3eb9ab46cf968477565 Mon Sep 17 00:00:00 2001 From: LIghtJUNction Date: Wed, 29 Apr 2026 04:18:18 +0800 Subject: [PATCH] fix: test_ensure_vec_db mock path and create kb_dir to avoid SQLite error --- tests/unit/test_kb_manager_resilience.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_kb_manager_resilience.py b/tests/unit/test_kb_manager_resilience.py index a4cfae7ee..1c086cfb5 100644 --- a/tests/unit/test_kb_manager_resilience.py +++ b/tests/unit/test_kb_manager_resilience.py @@ -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