mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 01:40:15 +08:00
21 lines
660 B
Python
21 lines
660 B
Python
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
|
|
from astrbot.core.db.vec_db.faiss_impl.vec_db import FaissVecDB
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_insert_batch_skips_empty_contents() -> None:
|
|
vec_db = FaissVecDB.__new__(FaissVecDB)
|
|
vec_db.embedding_provider = AsyncMock()
|
|
vec_db.document_storage = AsyncMock()
|
|
vec_db.embedding_storage = AsyncMock()
|
|
|
|
result = await FaissVecDB.insert_batch(vec_db, [])
|
|
|
|
assert result == []
|
|
vec_db.embedding_provider.get_embeddings_batch.assert_not_awaited()
|
|
vec_db.document_storage.insert_documents_batch.assert_not_awaited()
|
|
vec_db.embedding_storage.insert_batch.assert_not_awaited()
|