feat: overhaul UI styling, improve templates, enhance services and tests
This commit is contained in:
+13
-1
@@ -21,6 +21,7 @@ class TestEmbedderInit:
|
||||
"""CHROMA_ENABLED=false 时不初始化。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
emb.init_chroma()
|
||||
assert emb._chroma._client is None
|
||||
@@ -31,6 +32,7 @@ class TestEmbedderInit:
|
||||
monkeypatch.setattr(settings, "CHROMA_DIR", str(tmp_path / "chroma"))
|
||||
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
emb.init_chroma()
|
||||
|
||||
@@ -44,6 +46,7 @@ class TestEmbedderInit:
|
||||
"""CHROMA_ENABLED=false 时 get_collection 返回 None。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
assert emb.get_collection() is None
|
||||
|
||||
@@ -60,6 +63,7 @@ class TestEmbedderIndexing:
|
||||
"""CHROMA_ENABLED=false 时 index_paper 返回 False。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
assert emb.index_paper("test-id") is False
|
||||
|
||||
@@ -71,6 +75,7 @@ class TestEmbedderIndexing:
|
||||
monkeypatch.setattr(settings, "EMBED_MODEL", "")
|
||||
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
emb.init_chroma()
|
||||
|
||||
@@ -83,6 +88,7 @@ class TestEmbedderIndexing:
|
||||
"""CHROMA_ENABLED=false 时 index_batch 返回全失败。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
result = emb.index_batch(["a", "b"])
|
||||
assert result["success"] == 0
|
||||
@@ -92,6 +98,7 @@ class TestEmbedderIndexing:
|
||||
"""空列表时返回 0。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
result = emb.index_batch([])
|
||||
assert result["total"] == 0
|
||||
|
||||
@@ -99,6 +106,7 @@ class TestEmbedderIndexing:
|
||||
"""CHROMA_ENABLED=false 时 delete_paper 返回 False。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
assert emb.delete_paper("test-id") is False
|
||||
|
||||
@@ -106,6 +114,7 @@ class TestEmbedderIndexing:
|
||||
"""CHROMA_ENABLED=false 时 search_similar 返回空列表。"""
|
||||
monkeypatch.setattr(settings, "CHROMA_ENABLED", False)
|
||||
import app.services.embedder as emb
|
||||
|
||||
emb._chroma.reset()
|
||||
assert emb.search_similar("test query") == []
|
||||
|
||||
@@ -123,6 +132,7 @@ class TestEmbeddingApi:
|
||||
monkeypatch.setattr(settings, "EMBED_API_BASE", "")
|
||||
monkeypatch.setattr(settings, "EMBED_MODEL", "")
|
||||
import app.services.embedder as emb
|
||||
|
||||
assert emb._get_embedding("test") is None
|
||||
|
||||
def test_dimension_mismatch_returns_none(self, monkeypatch):
|
||||
@@ -158,6 +168,8 @@ class TestEmbeddingApi:
|
||||
with patch("httpx.Client") as mock_client:
|
||||
mock_client.return_value.__enter__ = MagicMock()
|
||||
mock_client.return_value.__exit__ = MagicMock(return_value=False)
|
||||
mock_client.return_value.__enter__.return_value.post.side_effect = Exception("timeout")
|
||||
mock_client.return_value.__enter__.return_value.post.side_effect = (
|
||||
Exception("timeout")
|
||||
)
|
||||
result = emb._get_embedding("test")
|
||||
assert result is None
|
||||
|
||||
Reference in New Issue
Block a user