Skip to content

Commit 20193ba

Browse files
Add mock implementations for ChromaDB and SentenceTransformer in tests
1 parent fb318c4 commit 20193ba

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

tests/conftest.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,19 @@
1+
import sys
2+
from unittest.mock import MagicMock
3+
4+
mock_chroma = MagicMock()
5+
mock_chroma.HttpClient.return_value = MagicMock() # Returns a fake client
6+
sys.modules["chromadb"] = mock_chroma
7+
8+
mock_st = MagicMock()
9+
mock_st.SentenceTransformer.return_value = MagicMock()
10+
sys.modules["sentence_transformers"] = mock_st
11+
112
import pytest
213
from fastapi.testclient import TestClient
314
from sqlmodel import Session, SQLModel, create_engine, pool
415
from app.main import app
516
from app.db.session import get_session
6-
from app.services.llm_factory import LLMOrchestrator
717

818
TEST_DATABASE_URL = "sqlite:///:memory:"
919
engine = create_engine(

0 commit comments

Comments
 (0)