fix: improve maintenance and synchronize live conversations

This commit is contained in:
WangShen007 committed 2026-10-29 09:46:21 +08:00
1 parent bb541bd212
commit ef86a5d3af
297 files changed
+2983 -1274

No files matched your search

@@ -18,7 +18,6 @@ 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
@@ -16,7 +16,7 @@ limitations under the License.
import json
from datetime import datetime, timedelta, timezone
from unittest.mock import MagicMock, patch
from unittest.mock import patch
import pytest
@@ -26,7 +26,6 @@ from app.models.role import Role
from app.models.permission import Permission, role_permissions
from app.models.agent import Agent, AgentType
from app.models.jira import AgentJiraConfig, JiraToken
from app.models.organization import Organization
from app.models.user import UserGroup
from app.core.security import get_password_hash
from app.repositories.jira import JiraRepository
@@ -461,7 +460,7 @@ class TestJiraRepositoryPerformance:
mock_query.side_effect = query_side_effect
result = jira_repo.get_agent_with_jira_config(str(test_agent.id))
jira_repo.get_agent_with_jira_config(str(test_agent.id))
# Verify that query was called (optimization check)
assert mock_query.called
@@ -15,10 +15,9 @@ limitations under the License.
"""
import pytest
import json
from datetime import datetime, timedelta
from uuid import UUID, uuid4
from unittest.mock import patch, MagicMock
from unittest.mock import patch
from sqlalchemy.orm import Session
from sqlalchemy.exc import SQLAlchemyError
@@ -385,13 +384,13 @@ class TestSessionQueries:
def test_get_agent_sessions(self, session_repo, test_agent, test_organization_id, test_customer):
"""Test retrieving all sessions for an agent"""
# Create multiple sessions
session1 = session_repo.create_session(
session_repo.create_session(
session_id=uuid4(),
agent_id=test_agent.id,
customer_id=test_customer.id,
organization_id=test_organization_id
)
session2 = session_repo.create_session(
session_repo.create_session(
session_id=uuid4(),
agent_id=test_agent.id,
customer_id=test_customer.id,
@@ -462,7 +461,7 @@ class TestSessionQueries:
def test_get_customer_sessions(self, session_repo, test_session, test_customer):
"""Test retrieving all sessions for a customer"""
# Create another session for the same customer
session2 = session_repo.create_session(
session_repo.create_session(
session_id=uuid4(),
agent_id=test_session.agent_id,
customer_id=test_customer.id,
@@ -494,7 +493,7 @@ class TestSessionQueries:
def test_get_agent_customer_sessions(self, session_repo, test_session, test_agent, test_customer):
"""Test retrieving sessions between specific agent and customer"""
# Create another session for the same agent-customer pair
session2 = session_repo.create_session(
session_repo.create_session(
session_id=uuid4(),
agent_id=test_agent.id,
customer_id=test_customer.id,
@@ -15,7 +15,7 @@ limitations under the License.
"""
import pytest
from unittest.mock import MagicMock, patch, ANY
from unittest.mock import MagicMock, patch
from sqlalchemy.orm import Session
from app.repositories.shopify_shop_repository import ShopifyShopRepository
from app.models.shopify import ShopifyShop
@@ -67,7 +67,7 @@ def test_get_shop(repository, mock_db, mock_shop, sample_shop_data):
shop_id = sample_shop_data["id"]
query_mock = MagicMock()
filter_mock = MagicMock()
first_mock = MagicMock(return_value=mock_shop)
MagicMock(return_value=mock_shop)
mock_db.query.return_value = query_mock
query_mock.filter.return_value = filter_mock
@@ -90,7 +90,7 @@ def test_get_shop_by_domain(repository, mock_db, mock_shop, sample_shop_data):
shop_domain = sample_shop_data["shop_domain"]
query_mock = MagicMock()
filter_mock = MagicMock()
first_mock = MagicMock(return_value=mock_shop)
MagicMock(return_value=mock_shop)
mock_db.query.return_value = query_mock
query_mock.filter.return_value = filter_mock
@@ -113,7 +113,7 @@ def test_get_shop_not_found(repository, mock_db):
shop_id = str(uuid.uuid4())
query_mock = MagicMock()
filter_mock = MagicMock()
first_mock = MagicMock(return_value=None)
MagicMock(return_value=None)
mock_db.query.return_value = query_mock
query_mock.filter.return_value = filter_mock
@@ -140,7 +140,7 @@ def test_get_shops(repository, mock_db):
query_mock = MagicMock()
offset_mock = MagicMock()
limit_mock = MagicMock()
all_mock = MagicMock(return_value=shops)
MagicMock(return_value=shops)
mock_db.query.return_value = query_mock
query_mock.offset.return_value = offset_mock
@@ -176,7 +176,7 @@ def test_get_shops_by_organization(repository, mock_db):
filter_mock = MagicMock()
offset_mock = MagicMock()
limit_mock = MagicMock()
all_mock = MagicMock(return_value=shops)
MagicMock(return_value=shops)
mock_db.query.return_value = query_mock
query_mock.filter.return_value = filter_mock
@@ -16,7 +16,7 @@ limitations under the License.
import pytest
from uuid import uuid4
from unittest.mock import Mock, MagicMock, patch, PropertyMock
from unittest.mock import Mock, MagicMock, patch
from datetime import datetime
from app.repositories.widget_app import WidgetAppRepository
@@ -533,7 +533,6 @@ class TestAPIKeyRegeneration:
self, mock_hash, mock_generate, widget_app_repo, mock_db, mock_widget_app, org_id
):
"""Test API key regeneration."""
old_hash = mock_widget_app.api_key_hash
mock_generate.return_value = "wak_new_key_12345678901234567890123456789012345"
mock_hash.return_value = "$2b$12$newhash"
@@ -627,7 +626,7 @@ class TestOrganizationIsolation:
def test_cannot_access_other_org_app(self, widget_app_repo, mock_db):
"""Test that one org cannot access another org's app."""
org1 = uuid4()
uuid4()
org2 = uuid4()
mock_query = MagicMock()