92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
import asyncio
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
from sqlalchemy import delete, select
|
|
|
|
from app.core.contracts import AgentRequest, RequestContext
|
|
from app.core.errors import IdempotencyConflictError
|
|
from app.infrastructure.db import SessionFactory
|
|
from app.model.conversation import ConversationMessage
|
|
from app.model.platform import AgentRun, RequestIdempotency
|
|
from app.service.agent_run_application_service import AgentRunApplicationService
|
|
|
|
pytestmark = pytest.mark.usefixtures("acceptance_registry")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_accept_is_idempotent_and_persists_outbox() -> None:
|
|
trace_id = str(uuid4())
|
|
session_id = f"integration-{uuid4()}"
|
|
key = f"key-{uuid4()}"
|
|
context = RequestContext(user_id="1", trace_id=trace_id,
|
|
roles=("customer",), permissions=("agent:run",))
|
|
request = AgentRequest(
|
|
agent_type="customer_service", message="integration test", session_id=session_id,
|
|
idempotency_key=key,
|
|
)
|
|
first = None
|
|
async with SessionFactory() as session:
|
|
try:
|
|
service = AgentRunApplicationService(session)
|
|
first = await service.accept(request, context)
|
|
second = await service.accept(request, context)
|
|
assert first.run_id == second.run_id
|
|
assert second.status == "queued"
|
|
with pytest.raises(IdempotencyConflictError):
|
|
await service.accept(request.model_copy(update={"message": "different"}), context)
|
|
finally:
|
|
run = await session.scalar(
|
|
select(AgentRun).where(AgentRun.run_id == first.run_id)
|
|
) if first is not None else None
|
|
if run is not None:
|
|
await session.execute(delete(AgentRun).where(AgentRun.id == run.id))
|
|
await session.execute(
|
|
delete(RequestIdempotency).where(RequestIdempotency.id == run.idempotency_id)
|
|
)
|
|
await session.execute(
|
|
delete(ConversationMessage).where(ConversationMessage.session_id == session_id)
|
|
)
|
|
await session.commit()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_concurrent_same_key_creates_at_most_one_run() -> None:
|
|
session_id = f"concurrent-{uuid4()}"
|
|
key = f"concurrent-key-{uuid4()}"
|
|
request = AgentRequest(
|
|
agent_type="customer_service", message="concurrent test", session_id=session_id,
|
|
idempotency_key=key,
|
|
)
|
|
|
|
async def submit() -> str | None:
|
|
async with SessionFactory() as session:
|
|
try:
|
|
result = await AgentRunApplicationService(session).accept(
|
|
request, RequestContext(user_id="1", trace_id=str(uuid4()),
|
|
roles=("customer",), permissions=("agent:run",))
|
|
)
|
|
return result.run_id
|
|
except (IdempotencyConflictError, Exception):
|
|
return None
|
|
|
|
results = await asyncio.gather(*(submit() for _ in range(5)))
|
|
run_ids = {run_id for run_id in results if run_id is not None}
|
|
assert len(run_ids) == 1
|
|
|
|
async with SessionFactory() as session:
|
|
rows = await session.scalars(
|
|
select(AgentRun).where(AgentRun.session_id == session_id)
|
|
)
|
|
runs = list(rows)
|
|
assert len(runs) == 1
|
|
if runs:
|
|
await session.execute(delete(AgentRun).where(AgentRun.id == runs[0].id))
|
|
await session.execute(
|
|
delete(RequestIdempotency).where(RequestIdempotency.id == runs[0].idempotency_id)
|
|
)
|
|
await session.execute(
|
|
delete(ConversationMessage).where(ConversationMessage.session_id == session_id)
|
|
)
|
|
await session.commit()
|