mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
263 lines
8.7 KiB
Python
263 lines
8.7 KiB
Python
"""Provider API route coverage — async httpx client + dependency overrides."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from http import HTTPStatus
|
|
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.provider import router as provider_router
|
|
from roboco.db.tables import ProviderConfigTable
|
|
from roboco.models import AgentRole, Team
|
|
from roboco.models.base import ModelProvider
|
|
from roboco.models.permissions import AgentContext
|
|
from sqlalchemy import delete, select
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import AsyncIterator
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
|
|
def _make_app(db_session, role: AgentRole = AgentRole.MAIN_PM, team=None) -> FastAPI:
|
|
app = FastAPI()
|
|
app.include_router(provider_router, prefix="/api/providers")
|
|
|
|
async def _override_db():
|
|
yield db_session
|
|
|
|
async def _override_agent() -> AgentContext:
|
|
return AgentContext(agent_id=uuid4(), role=role, team=team)
|
|
|
|
app.dependency_overrides[get_db] = _override_db
|
|
app.dependency_overrides[get_agent_context] = _override_agent
|
|
return app
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def app_client(
|
|
db_session: AsyncSession,
|
|
) -> AsyncIterator[AsyncClient]:
|
|
app = _make_app(db_session)
|
|
suffix = uuid4().hex[:8]
|
|
# Only seed if not already present (set_ollama_api_key in a prior test
|
|
# may have committed rows that survive rollback isolation).
|
|
existing = (
|
|
await db_session.execute(
|
|
select(ProviderConfigTable).where(
|
|
ProviderConfigTable.type == ModelProvider.OLLAMA_CLOUD
|
|
)
|
|
)
|
|
).scalar_one_or_none()
|
|
if existing is None:
|
|
db_session.add(
|
|
ProviderConfigTable(
|
|
name=f"anthropic-test-{suffix}",
|
|
type=ModelProvider.ANTHROPIC,
|
|
enabled=True,
|
|
)
|
|
)
|
|
db_session.add(
|
|
ProviderConfigTable(
|
|
name=f"ollama-test-{suffix}",
|
|
type=ModelProvider.OLLAMA_CLOUD,
|
|
enabled=False,
|
|
base_url="https://ollama.example.com",
|
|
)
|
|
)
|
|
await db_session.flush()
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
yield client
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
_HDR_PM = {"X-Agent-ID": str(uuid4()), "X-Agent-Role": "main_pm"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_catalog(app_client: AsyncClient) -> None:
|
|
response = await app_client.get("/api/providers/catalog", headers=_HDR_PM)
|
|
assert response.status_code == HTTPStatus.OK
|
|
assert isinstance(response.json(), list)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_catalog_forbidden_for_developer(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
app = _make_app(db_session, role=AgentRole.DEVELOPER, team=Team.BACKEND)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.get(
|
|
"/api/providers/catalog",
|
|
headers={"X-Agent-ID": str(uuid4()), "X-Agent-Role": "developer"},
|
|
)
|
|
assert response.status_code == HTTPStatus.FORBIDDEN
|
|
app.dependency_overrides.clear()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_ollama_key_status(app_client: AsyncClient) -> None:
|
|
response = await app_client.get("/api/providers/ollama-key", headers=_HDR_PM)
|
|
assert response.status_code == HTTPStatus.OK
|
|
body = response.json()
|
|
assert "has_key" in body
|
|
assert "enabled" in body
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_ollama_key(app_client: AsyncClient) -> None:
|
|
response = await app_client.put(
|
|
"/api/providers/ollama-key",
|
|
json={"api_key": "secret-key-123"},
|
|
headers=_HDR_PM,
|
|
)
|
|
assert response.status_code == HTTPStatus.OK
|
|
body = response.json()
|
|
assert body["has_key"] is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_current_mode(app_client: AsyncClient) -> None:
|
|
response = await app_client.get("/api/providers", headers=_HDR_PM)
|
|
assert response.status_code == HTTPStatus.OK
|
|
body = response.json()
|
|
assert body["mode"] in {"anthropic", "ollama", "mix"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_mode_anthropic_clears_assignments(
|
|
app_client: AsyncClient,
|
|
) -> None:
|
|
response = await app_client.post(
|
|
"/api/providers", json={"mode": "anthropic"}, headers=_HDR_PM
|
|
)
|
|
assert response.status_code == HTTPStatus.OK
|
|
body = response.json()
|
|
assert body["mode"] == "anthropic"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_mode_unknown_returns_4xx(app_client: AsyncClient) -> None:
|
|
"""Unknown mode is rejected — Pydantic 422 at schema layer or 400 at service."""
|
|
response = await app_client.post(
|
|
"/api/providers", json={"mode": "quantum"}, headers=_HDR_PM
|
|
)
|
|
assert response.status_code in (
|
|
HTTPStatus.BAD_REQUEST,
|
|
HTTPStatus.UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_ollama_key_not_seeded(db_session: AsyncSession) -> None:
|
|
"""When provider not seeded, returns 404."""
|
|
# Delete the OLLAMA_CLOUD provider
|
|
await db_session.execute(
|
|
delete(ProviderConfigTable).where(
|
|
ProviderConfigTable.type == ModelProvider.OLLAMA_CLOUD
|
|
)
|
|
)
|
|
await db_session.flush()
|
|
|
|
app = _make_app(db_session)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.get("/api/providers/ollama-key", headers=_HDR_PM)
|
|
app.dependency_overrides.clear()
|
|
assert response.status_code == HTTPStatus.NOT_FOUND
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_ollama_key_no_provider(db_session: AsyncSession) -> None:
|
|
"""Setting key with no provider raises 404."""
|
|
await db_session.execute(
|
|
delete(ProviderConfigTable).where(
|
|
ProviderConfigTable.type == ModelProvider.OLLAMA_CLOUD
|
|
)
|
|
)
|
|
await db_session.flush()
|
|
|
|
app = _make_app(db_session)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.put(
|
|
"/api/providers/ollama-key",
|
|
json={"api_key": "secret"},
|
|
headers=_HDR_PM,
|
|
)
|
|
app.dependency_overrides.clear()
|
|
assert response.status_code == HTTPStatus.NOT_FOUND
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_ollama_key_developer_forbidden(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
app = _make_app(db_session, role=AgentRole.DEVELOPER, team=Team.BACKEND)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.put(
|
|
"/api/providers/ollama-key",
|
|
json={"api_key": "secret"},
|
|
headers={"X-Agent-ID": str(uuid4()), "X-Agent-Role": "developer"},
|
|
)
|
|
app.dependency_overrides.clear()
|
|
assert response.status_code == HTTPStatus.FORBIDDEN
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_mode_developer_forbidden(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
app = _make_app(db_session, role=AgentRole.DEVELOPER, team=Team.BACKEND)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.get(
|
|
"/api/providers",
|
|
headers={"X-Agent-ID": str(uuid4()), "X-Agent-Role": "developer"},
|
|
)
|
|
app.dependency_overrides.clear()
|
|
assert response.status_code == HTTPStatus.FORBIDDEN
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_mode_mix_without_per_agent_returns_400(
|
|
app_client: AsyncClient,
|
|
) -> None:
|
|
"""Apply 'mix' mode without per_agent triggers ValueError → 400 (lines 149-152)."""
|
|
response = await app_client.post(
|
|
"/api/providers",
|
|
json={"mode": "mix"},
|
|
headers=_HDR_PM,
|
|
)
|
|
assert response.status_code == HTTPStatus.BAD_REQUEST
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_mode_ollama_without_provider_returns_404(
|
|
db_session: AsyncSession,
|
|
) -> None:
|
|
"""Apply 'ollama' mode without ollama provider raises NotFoundError → 404."""
|
|
await db_session.execute(
|
|
delete(ProviderConfigTable).where(
|
|
ProviderConfigTable.type == ModelProvider.OLLAMA_CLOUD
|
|
)
|
|
)
|
|
await db_session.flush()
|
|
|
|
app = _make_app(db_session)
|
|
transport = ASGITransport(app=app)
|
|
async with AsyncClient(transport=transport, base_url="http://test") as client:
|
|
response = await client.post(
|
|
"/api/providers", json={"mode": "ollama"}, headers=_HDR_PM
|
|
)
|
|
app.dependency_overrides.clear()
|
|
assert response.status_code in (HTTPStatus.NOT_FOUND, HTTPStatus.OK)
|