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.
283 lines
8.0 KiB
Python
283 lines
8.0 KiB
Python
"""WorkSession 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_agent_context, get_db
|
|
from roboco.api.routes.work_session import router as ws_router
|
|
from roboco.db.tables import AgentTable, ProjectTable, TaskTable
|
|
from roboco.models import AgentRole, AgentStatus, Team
|
|
from roboco.models.base import (
|
|
TaskNature,
|
|
TaskStatus,
|
|
TaskType,
|
|
)
|
|
from roboco.models.permissions import AgentContext
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import AsyncIterator
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def ws_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()
|
|
project = ProjectTable(
|
|
id=uuid4(),
|
|
name="WS-Proj",
|
|
slug=f"ws-proj-{uuid4().hex[:6]}",
|
|
git_url="https://example.com/r.git",
|
|
assigned_cell=Team.BACKEND,
|
|
created_by=agent.id,
|
|
)
|
|
db_session.add(project)
|
|
await db_session.flush()
|
|
task = TaskTable(
|
|
id=uuid4(),
|
|
title="t",
|
|
description="d",
|
|
acceptance_criteria=["ac"],
|
|
status=TaskStatus.PENDING,
|
|
priority=2,
|
|
task_type=TaskType.CODE,
|
|
nature=TaskNature.TECHNICAL,
|
|
project_id=project.id,
|
|
created_by=agent.id,
|
|
team=Team.BACKEND,
|
|
)
|
|
db_session.add(task)
|
|
await db_session.flush()
|
|
|
|
app = FastAPI()
|
|
app.include_router(ws_router, prefix="/api/work-sessions")
|
|
|
|
async def _override_db():
|
|
yield db_session
|
|
|
|
async def _override_agent() -> AgentContext:
|
|
return AgentContext(
|
|
agent_id=agent.id, role=AgentRole.DEVELOPER, team=Team.BACKEND
|
|
)
|
|
|
|
app.dependency_overrides[get_db] = _override_db
|
|
app.dependency_overrides[get_agent_context] = _override_agent
|
|
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
yield {
|
|
"client": client,
|
|
"agent": agent,
|
|
"project": project,
|
|
"task": task,
|
|
}
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
_HDR = {"X-Agent-ID": str(uuid4()), "X-Agent-Role": "developer"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_list_sessions_empty(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
response = await client.get("/api/work-sessions", headers=_HDR)
|
|
assert response.status_code == 200
|
|
assert isinstance(response.json(), list)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_session_not_found(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
response = await client.get(f"/api/work-sessions/{uuid4()}", headers=_HDR)
|
|
assert response.status_code == 404
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_session(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
response = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/x",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 201
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_session_by_id(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/y",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.get(f"/api/work-sessions/{sid}", headers=_HDR)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_commit(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/c",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.post(
|
|
f"/api/work-sessions/{sid}/commits",
|
|
json={"commit_sha": "abc123def"},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_commit_session_not_found(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
response = await client.post(
|
|
f"/api/work-sessions/{uuid4()}/commits",
|
|
json={"commit_sha": "abc"},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 404
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_files(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/f",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.post(
|
|
f"/api/work-sessions/{sid}/files",
|
|
json={"file_paths": ["a.py", "b.py"]},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_create_pr(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/p",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.post(
|
|
f"/api/work-sessions/{sid}/pr",
|
|
json={"pr_number": 42, "pr_url": "https://github.com/x/y/pull/42"},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_complete_session(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/cmpl",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.post(f"/api/work-sessions/{sid}/complete", headers=_HDR)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_abandon_session(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
create = await client.post(
|
|
"/api/work-sessions",
|
|
json={
|
|
"project_id": str(ws_client["project"].id),
|
|
"task_id": str(ws_client["task"].id),
|
|
"branch_name": "feature/ab",
|
|
"base_branch": "main",
|
|
"target_branch": "main",
|
|
},
|
|
headers=_HDR,
|
|
)
|
|
sid = create.json()["id"]
|
|
response = await client.post(
|
|
f"/api/work-sessions/{sid}/abandon",
|
|
params={"reason": "scrapped"},
|
|
headers=_HDR,
|
|
)
|
|
assert response.status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_active_for_task_returns_null(ws_client: dict) -> None:
|
|
client = ws_client["client"]
|
|
response = await client.get(
|
|
f"/api/work-sessions/task/{ws_client['task'].id}", headers=_HDR
|
|
)
|
|
# Returns null body (200 with None) when there's no active session.
|
|
assert response.status_code == 200
|