Files
replypier/backend/tests/repositories/test_chat_sentiment.py
T
WangShen007 54b949b392 chore(brand): rename the product to ReplyPier
The upstream project name is gone from code, assets, docs and build
output: brand strings, domains (replypier.site and its app/api/docs/help
subdomains), env prefixes, image and network names, the widget's public
JS API (window.ReplyPier, replypierId/BaseUrl/Config/Token), CSS tokens
(--rp-*), file and directory names, and the WordPress plugin package.

- widget loader is now webclient/replypier.min.js; both bundles rebuilt
  from source (backend/assets/widget.{js,css}, frontend/public/webclient)
- NOTICE, llms.txt, CONTRIBUTING, HELP_CENTER_INFRA and config.dev.js
  rewritten rather than string-swapped; the demo config no longer ships
  the previous vendor's Firebase/Razorpay/Google keys
- new migration replypier_ai_model_001 adds REPLYPIER to the managed
  AIModelType and rewrites pre-rename rows
- local compose project, database, role and volume renamed; LOCAL_DEPLOY
  documents carrying the old local data over
2026-10-26 00:51:03 +08:00

128 lines
5.3 KiB
Python

"""
Copyright 2024-2026 ReplyPier
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