mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
Renn F
64c48356d0
Service-level tests now exercise provider, permissions, project, journal, messaging, work_session, metrics, kanban, extraction, learning, notification, dashboard, llm_routing, a2a, task, repository_base, audit, db_seed, branch_name, indexed_document, query_helpers, agent. API route tests cover provider, journal, project, sessions, dashboard, work_session, tasks, a2a, groups, notifications, agents, channels, messages, kanban, api_resources. Pure-function helpers covered: handlers, deps_helpers, middleware, middleware_docs, transcription, pr templates, agents_config, errors, logging, journal/notification/channel/a2a access, task_lifecycle, streaming, converters, crypto, schemas (common + websocket), events, permissions extras. pyproject ruff per-file-ignores extended for tests so PLR2004 (status code magic values), PLC0415 (lazy imports), PLR0913 (fixture params), ARG001 (unused fixture deps), SIM105, and E501 don't fight test idioms.
109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
"""Messages API route coverage."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
from fastapi import FastAPI
|
|
from httpx import ASGITransport, AsyncClient
|
|
from roboco.api.deps import get_current_agent_id, get_db
|
|
from roboco.api.routes.messages import router as messages_router
|
|
from roboco.db.tables import AgentTable
|
|
from roboco.models import AgentRole, AgentStatus, Team
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import AsyncIterator
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def messages_client(
|
|
db_session: AsyncSession,
|
|
) -> AsyncIterator[dict]:
|
|
agent = AgentTable(
|
|
id=uuid4(),
|
|
name="Dev",
|
|
slug=f"be-dev-{uuid4().hex[:8]}",
|
|
role=AgentRole.DEVELOPER,
|
|
team=Team.BACKEND,
|
|
status=AgentStatus.ACTIVE,
|
|
model_config={},
|
|
system_prompt="dev",
|
|
capabilities=[],
|
|
permissions={},
|
|
metrics={},
|
|
)
|
|
db_session.add(agent)
|
|
await db_session.flush()
|
|
|
|
app = FastAPI()
|
|
app.include_router(messages_router, prefix="/api/messages")
|
|
|
|
async def _override_db():
|
|
yield db_session
|
|
|
|
async def _override_agent_id():
|
|
return agent.id
|
|
|
|
app.dependency_overrides[get_db] = _override_db
|
|
app.dependency_overrides[get_current_agent_id] = _override_agent_id
|
|
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
yield {"client": client, "agent": agent}
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
_HDR = {"X-Agent-ID": str(uuid4()), "X-Agent-Role": "developer"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_messages_unknown_session(messages_client: dict) -> None:
|
|
client = messages_client["client"]
|
|
response = await client.get(f"/api/messages?session_id={uuid4()}", headers=_HDR)
|
|
assert response.status_code == 404
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_message_not_found(messages_client: dict) -> None:
|
|
client = messages_client["client"]
|
|
response = await client.get(f"/api/messages/{uuid4()}", headers=_HDR)
|
|
assert response.status_code == 404
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_send_message_unknown_session(messages_client: dict) -> None:
|
|
client = messages_client["client"]
|
|
response = await client.post(
|
|
"/api/messages",
|
|
json={
|
|
"session_id": str(uuid4()),
|
|
"content": "hello",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
# 400 if session not found / 422 if validation
|
|
assert response.status_code in (400, 404, 422)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_edit_message_not_found(messages_client: dict) -> None:
|
|
client = messages_client["client"]
|
|
response = await client.patch(
|
|
f"/api/messages/{uuid4()}",
|
|
json={"new_content": "edited"},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code in (404, 422)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_delete_message_not_found(messages_client: dict) -> None:
|
|
client = messages_client["client"]
|
|
response = await client.delete(f"/api/messages/{uuid4()}", headers=_HDR)
|
|
assert response.status_code in (204, 404)
|