AFFAANhandClaude Opus 5
19e1e84fb7
Independent FastAPI backend for the MaskanX agentic growth platform. Includes the agent runtime, MCP client integrations (Meta Ads, LinkedIn, HubSpot, Tavily, Exa, xAI, Citedy, image generation), PostgreSQL storage for chats and cron jobs, provider and secret management, and the CLI. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
84 lines
3.7 KiB
Python
84 lines
3.7 KiB
Python
"""Tests for Mission Control P2: Dashboard API data aggregation."""
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from fastapi import FastAPI
|
|
from unittest.mock import patch
|
|
from adclaw.config.config import Config, AgentsConfig, PersonaConfig, PersonaCronConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def dashboard_client():
|
|
from adclaw.app.routers.personas import router
|
|
app = FastAPI()
|
|
app.include_router(router, prefix="/api")
|
|
return TestClient(app)
|
|
|
|
|
|
class TestDashboardData:
|
|
"""Dashboard needs personas + model + skills + cron data."""
|
|
|
|
def test_personas_include_model_info(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
# Add model info to researcher
|
|
config_with_personas.agents.personas[1].model_provider = "aliyun-intl"
|
|
config_with_personas.agents.personas[1].model_name = "qwen3.5-plus"
|
|
mock.return_value = config_with_personas
|
|
|
|
resp = dashboard_client.get("/api/agents/personas")
|
|
assert resp.status_code == 200
|
|
personas = resp.json()
|
|
researcher = next(p for p in personas if p["id"] == "researcher")
|
|
assert researcher["model_provider"] == "aliyun-intl"
|
|
assert researcher["model_name"] == "qwen3.5-plus"
|
|
|
|
def test_personas_include_skills_count(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
config_with_personas.agents.personas[1].skills = ["citedy-trend-scout", "exa-search"]
|
|
mock.return_value = config_with_personas
|
|
|
|
resp = dashboard_client.get("/api/agents/personas")
|
|
personas = resp.json()
|
|
researcher = next(p for p in personas if p["id"] == "researcher")
|
|
assert len(researcher["skills"]) == 2
|
|
|
|
def test_personas_include_cron_config(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
config_with_personas.agents.personas[1].cron = PersonaCronConfig(
|
|
enabled=True,
|
|
schedule="0 9 * * *",
|
|
prompt="Find AI trends today",
|
|
output="chat",
|
|
)
|
|
mock.return_value = config_with_personas
|
|
|
|
resp = dashboard_client.get("/api/agents/personas")
|
|
personas = resp.json()
|
|
researcher = next(p for p in personas if p["id"] == "researcher")
|
|
assert researcher["cron"]["enabled"] is True
|
|
assert researcher["cron"]["schedule"] == "0 9 * * *"
|
|
|
|
def test_coordinator_marked(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
mock.return_value = config_with_personas
|
|
resp = dashboard_client.get("/api/agents/personas")
|
|
personas = resp.json()
|
|
coord = next(p for p in personas if p["is_coordinator"])
|
|
assert coord["id"] == "coordinator"
|
|
|
|
|
|
class TestDashboardFiltering:
|
|
"""Dashboard should support filtering by persona properties."""
|
|
|
|
def test_list_returns_all_personas(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
mock.return_value = config_with_personas
|
|
resp = dashboard_client.get("/api/agents/personas")
|
|
assert len(resp.json()) == 3
|
|
|
|
def test_get_single_persona_by_id(self, dashboard_client, config_with_personas):
|
|
with patch("adclaw.app.routers.personas.load_config") as mock:
|
|
mock.return_value = config_with_personas
|
|
resp = dashboard_client.get("/api/agents/personas/researcher")
|
|
assert resp.status_code == 200
|
|
assert resp.json()["name"] == "Mike"
|