test: 补充知识处理、检索与 Agent 回归用例

覆盖切块、向量检索、工具调用和历史预算；模型相关用例使用桩。

backend/tests/repositories/test_chat_sentiment.py

@@ -0,0 +1,127 @@
+"""
+Copyright 2024-2026 ChatterMate
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import pytest
+from uuid import uuid4
+from unittest.mock import patch
+from app.repositories.chat import ChatRepository
+from app.models.chat_history import ChatHistory
+from app.models.session_to_agent import SessionToAgent, SessionStatus
+
+
+class TestChatSentimentIntegration:
+    """Tests for sentiment analysis integration in ChatRepository."""
+
+    @pytest.fixture
+    def chat_repo(self, db):
+        return ChatRepository(db)
+
+    @pytest.fixture
+    def test_session(self, db, test_organization, test_agent, test_customer):
+        session_id = uuid4()
+        session = SessionToAgent(
+            session_id=session_id,
+            agent_id=test_agent.id,
+            customer_id=test_customer.id,
+            organization_id=test_organization.id,
+            status=SessionStatus.OPEN
+        )
+        db.add(session)
+        db.commit()
+        db.refresh(session)
+        return session
+
+    def test_create_user_message_adds_sentiment(self, chat_repo, test_organization, test_agent, test_customer, test_session):
+        """User messages should have sentiment analysis applied."""
+        message = chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "This is great, thank you so much!",
+            "message_type": "user"
+        })
+        assert message.sentiment_label is not None
+        assert message.sentiment_label in ("positive", "neutral", "negative")
+        assert message.sentiment_score is not None
+
+    def test_create_bot_message_no_sentiment(self, chat_repo, test_organization, test_agent, test_customer, test_session):
+        """Bot messages should NOT have sentiment analysis."""
+        message = chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "How can I help you today?",
+            "message_type": "bot"
+        })
+        assert message.sentiment_label is None
+        assert message.sentiment_score is None
+
+    def test_session_sentiment_updated_after_user_message(self, chat_repo, test_organization, test_agent, test_customer, test_session, db):
+        """Session sentiment should be updated after a user message."""
+        chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "This is terrible and broken, I hate it",
+            "message_type": "user"
+        })
+        db.refresh(test_session)
+        assert test_session.sentiment_label is not None
+        assert test_session.sentiment_score is not None
+
+    def test_session_sentiment_averages_multiple_messages(self, chat_repo, test_organization, test_agent, test_customer, test_session, db):
+        """Session sentiment should average across multiple user messages."""
+        # Send a positive message
+        chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "thanks great awesome wonderful",
+            "message_type": "user"
+        })
+        # Send a negative message
+        chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "terrible awful horrible worst",
+            "message_type": "user"
+        })
+        db.refresh(test_session)
+        # With one positive and one negative, session should be near neutral
+        assert test_session.sentiment_label is not None
+        assert test_session.sentiment_score is not None
+
+    @patch("app.repositories.chat.analyze_sentiment", side_effect=Exception("analysis failed"))
+    def test_sentiment_error_does_not_break_message_creation(self, mock_analyze, chat_repo, test_organization, test_agent, test_customer, test_session):
+        """Sentiment analysis errors should not prevent message creation."""
+        message = chat_repo.create_message({
+            "organization_id": test_organization.id,
+            "agent_id": test_agent.id,
+            "customer_id": test_customer.id,
+            "session_id": test_session.session_id,
+            "message": "Hello there",
+            "message_type": "user"
+        })
+        assert message is not None
+        assert message.message == "Hello there"
+        # Sentiment should be None since analysis failed
+        assert message.sentiment_label is None