nessi
58eab24258
Add FastAPI-based monitoring dashboard with support for Codex CLI and Devin API v3 Enterprise consumption tracking. Includes Docker deployment, systemd service configuration, kiosk mode launcher, comprehensive documentation, and demo mode for testing without credentials.
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.config import Settings
|
|
from app.main import create_app
|
|
from app.models import ProviderUsage
|
|
from app.providers.base import UsageProvider
|
|
|
|
|
|
class StubProvider(UsageProvider):
|
|
def __init__(self, provider: str, status: str):
|
|
super().__init__(provider, 1, 'test')
|
|
self.status = status
|
|
|
|
async def _fetch_usage(self):
|
|
return ProviderUsage(provider=self.provider, status=self.status, source='test',
|
|
last_successful_update=datetime.now(timezone.utc) if self.status == 'ok' else None)
|
|
|
|
|
|
def test_health_index_and_demo_usage():
|
|
settings = Settings(demo_mode=True)
|
|
with TestClient(create_app(settings)) as client:
|
|
assert client.get('/health').json() == {'status': 'ok'}
|
|
index = client.get('/')
|
|
assert index.status_code == 200
|
|
assert 'AI USAGE' in index.text
|
|
asset = client.get('/static/style.css')
|
|
assert asset.status_code == 200
|
|
assert 'provider-card' in asset.text
|
|
usage = client.get('/api/usage')
|
|
assert usage.status_code == 200
|
|
assert usage.headers['cache-control'] == 'no-store'
|
|
body = usage.json()
|
|
assert body['demo_mode'] is True
|
|
assert body['codex']['status'] == 'ok'
|
|
assert body['devin']['status'] == 'ok'
|
|
datetime.fromisoformat(body['server_time'])
|
|
|
|
|
|
def test_provider_errors_are_independent():
|
|
settings = Settings(codex_timeout=1, devin_timeout=1)
|
|
providers = (StubProvider('codex', 'ok'), StubProvider('devin', 'error'))
|
|
with TestClient(create_app(settings, providers)) as client:
|
|
body = client.get('/api/usage').json()
|
|
assert body['codex']['status'] == 'ok'
|
|
assert body['devin']['status'] == 'error'
|
|
assert body['devin']['error'] == 'Unable to retrieve usage information'
|
|
|
|
|
|
def test_disabled_provider_is_normalized():
|
|
settings = Settings(demo_mode=True, codex_enabled=False)
|
|
with TestClient(create_app(settings)) as client:
|
|
body = client.get('/api/usage').json()
|
|
assert body['codex']['status'] == 'disabled'
|
|
assert body['codex']['limits'] == []
|