5f3169bad4
Adds ResolutionOutputGenerator service that generates PSA ticket notes, knowledge base article draft, and client summary on session resolve, plus integration tests for generate_all and edit_output. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
"""Integration tests for ResolutionOutputGenerator."""
|
|
import pytest
|
|
from unittest.mock import AsyncMock, patch
|
|
from httpx import AsyncClient
|
|
|
|
from app.models.ai_session import AISession
|
|
from app.models.session_resolution_output import SessionResolutionOutput
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@patch("app.services.resolution_output_generator._call_ai")
|
|
async def test_generate_all_creates_three_outputs(
|
|
mock_call_ai, client: AsyncClient, test_user, auth_headers, test_db
|
|
):
|
|
"""generate_all creates PSA notes, KB article, and client summary."""
|
|
mock_call_ai.return_value = ("Generated content here", 100, 50)
|
|
|
|
session = AISession(
|
|
user_id=test_user["user_data"]["id"],
|
|
account_id=test_user["user_data"]["account_id"],
|
|
session_type="guided",
|
|
intake_type="free_text",
|
|
intake_content={"text": "test"},
|
|
status="resolved",
|
|
confidence_tier="guided",
|
|
conversation_messages=[
|
|
{"role": "user", "content": "DNS not working"},
|
|
{"role": "assistant", "content": "Fixed by flushing DNS cache"},
|
|
],
|
|
resolution_summary="Flushed DNS cache",
|
|
)
|
|
test_db.add(session)
|
|
await test_db.flush()
|
|
|
|
from app.services.resolution_output_generator import ResolutionOutputGenerator
|
|
gen = ResolutionOutputGenerator(test_db)
|
|
outputs = await gen.generate_all(session.id)
|
|
|
|
assert len(outputs) == 3
|
|
types = {o.output_type for o in outputs}
|
|
assert types == {"psa_ticket_notes", "knowledge_base", "client_summary"}
|
|
assert all(o.status == "draft" for o in outputs)
|
|
assert mock_call_ai.call_count == 3
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_edit_output(client: AsyncClient, test_user, auth_headers, test_db):
|
|
"""Editing an output stores edited_content."""
|
|
session = AISession(
|
|
user_id=test_user["user_data"]["id"],
|
|
account_id=test_user["user_data"]["account_id"],
|
|
session_type="guided",
|
|
intake_type="free_text",
|
|
intake_content={"text": "test"},
|
|
status="resolved",
|
|
confidence_tier="guided",
|
|
conversation_messages=[],
|
|
resolution_summary="Fixed it",
|
|
)
|
|
test_db.add(session)
|
|
await test_db.flush()
|
|
|
|
output = SessionResolutionOutput(
|
|
session_id=session.id,
|
|
output_type="psa_ticket_notes",
|
|
generated_content="Original notes",
|
|
status="draft",
|
|
generated_by_model="claude-sonnet-4-6",
|
|
)
|
|
test_db.add(output)
|
|
await test_db.flush()
|
|
|
|
from app.services.resolution_output_generator import ResolutionOutputGenerator
|
|
gen = ResolutionOutputGenerator(test_db)
|
|
edited = await gen.edit_output(output.id, "My edited notes")
|
|
|
|
assert edited.edited_content == "My edited notes"
|