"""roboco.services.cockpit + route — read-only company summary (mocked deps).""" from __future__ import annotations from http import HTTPStatus from typing import Any from unittest.mock import AsyncMock, MagicMock from uuid import uuid4 import pytest from fastapi import HTTPException from roboco.api.routes import cockpit as croute from roboco.models import AgentRole from roboco.models.permissions import AgentContext from roboco.services import cockpit as cm from roboco.services.cockpit import CockpitService from roboco.services.strategy_engine import StrategyObservation _IN_PROGRESS = 2 _CLAIMED = 1 _BLOCKED = 3 _BUDGET = 100.0 _SPEND_30D = 150.0 def _agent(role: AgentRole) -> AgentContext: return AgentContext(agent_id=uuid4(), role=role, team=None) def _patch(monkeypatch: pytest.MonkeyPatch) -> None: goals = { "north_star": "Win the market", "objectives": [{"metric": "NPS"}], "operating_policy": {"monthly_budget_cap": _BUDGET}, } monkeypatch.setattr( cm, "get_company_goals_service", lambda _s: MagicMock(get=AsyncMock(return_value=goals)), ) counts = { "in_progress": _IN_PROGRESS, "claimed": _CLAIMED, "blocked": _BLOCKED, "awaiting_ceo_approval": 1, } monkeypatch.setattr( cm, "get_task_service", lambda _s: MagicMock(count_by_status=AsyncMock(return_value=counts)), ) usage = MagicMock( get_summary=AsyncMock(return_value={"total_cost_usd": _SPEND_30D}), get_projection=AsyncMock(return_value={"projected_monthly_cost_usd": 200.0}), ) monkeypatch.setattr(cm, "get_usage_service", lambda _s: usage) monkeypatch.setattr( cm, "get_strategy_engine", lambda _s: MagicMock( assess=AsyncMock( return_value=[StrategyObservation(kind="idle", summary="s", detail="d")] ) ), ) proposed = MagicMock() proposed.status = "proposed" done = MagicMock() done.status = "provisioned" monkeypatch.setattr( cm, "get_pitch_service", lambda _s: MagicMock(list_pitches=AsyncMock(return_value=[proposed, done])), ) @pytest.mark.asyncio async def test_summary_aggregates(monkeypatch: pytest.MonkeyPatch) -> None: _patch(monkeypatch) out = await CockpitService(MagicMock()).summary() assert out["basis"] == "proxy" assert out["north_star"] == "Win the market" assert out["delivery"]["in_flight"] == _IN_PROGRESS + _CLAIMED assert out["delivery"]["blocked"] == _BLOCKED assert out["spend"]["spend_30d_usd"] == _SPEND_30D assert out["spend"]["over_budget"] is True assert out["pending_pitches"] == 1 assert out["signals"][0]["kind"] == "idle" @pytest.mark.asyncio async def test_route_forbidden_for_developer() -> None: with pytest.raises(HTTPException) as exc: await croute.cockpit_summary(MagicMock(), _agent(AgentRole.DEVELOPER)) assert exc.value.status_code == HTTPStatus.FORBIDDEN @pytest.mark.asyncio async def test_route_ok_for_ceo(monkeypatch: pytest.MonkeyPatch) -> None: summary: dict[str, Any] = { "basis": "proxy", "north_star": "Win", "objectives": [], "delivery": { "task_counts": {}, "in_flight": 0, "blocked": 0, "awaiting_ceo": 0, }, "spend": { "spend_30d_usd": 0.0, "projected_monthly_usd": None, "monthly_budget_cap_usd": None, "over_budget": False, }, "pending_pitches": 0, "signals": [], } svc = MagicMock(summary=AsyncMock(return_value=summary)) monkeypatch.setattr(croute, "get_cockpit_service", lambda _db: svc) resp = await croute.cockpit_summary(MagicMock(), _agent(AgentRole.CEO)) assert resp.basis == "proxy" assert resp.spend.over_budget is False @pytest.mark.asyncio async def test_signals_returns_only_strategy_signals( monkeypatch: pytest.MonkeyPatch, ) -> None: # The lightweight slice returns ONLY the strategy signals — none of the # summary fan-out (goals / spend / counts / pitches). _patch(monkeypatch) out = await CockpitService(MagicMock()).signals() assert list(out.keys()) == ["signals"] assert out["signals"][0]["kind"] == "idle" assert out["signals"][0]["summary"] == "s" @pytest.mark.asyncio async def test_signals_route_forbidden_for_developer() -> None: with pytest.raises(HTTPException) as exc: await croute.cockpit_signals(MagicMock(), _agent(AgentRole.DEVELOPER)) assert exc.value.status_code == HTTPStatus.FORBIDDEN @pytest.mark.asyncio async def test_signals_route_ok_for_ceo(monkeypatch: pytest.MonkeyPatch) -> None: svc = MagicMock( signals=AsyncMock( return_value={"signals": [{"kind": "idle", "summary": "s", "detail": "d"}]} ) ) monkeypatch.setattr(croute, "get_cockpit_service", lambda _db: svc) resp = await croute.cockpit_signals(MagicMock(), _agent(AgentRole.CEO)) assert resp.signals[0].kind == "idle"