36c53e0e7c
Major additions across backend, frontend, and infrastructure: Backend: - IoT collector framework (Modbus TCP, MQTT, HTTP) with manager - Realistic Beijing solar/weather simulator with cloud transients - Alarm auto-checker with demo anomaly injection (3-4 events/hour) - Report generation (PDF/Excel) with sync fallback and E2E testing - Energy data CSV/XLSX export endpoint - WebSocket real-time broadcast at /ws/realtime - Alembic initial migration for all 14 tables - 77 pytest tests across 9 API routers Frontend: - Live notification badge with alarm count (was hardcoded 0) - Sankey energy flow diagram on dashboard - Device photos (SVG illustrations) on all device pages - Report download with status icons - Energy data export buttons (CSV/Excel) - WebSocket hook with auto-reconnect and polling fallback - BigScreen 2D responsive CSS (tablet/mobile) - Error handling improvements across pages Infrastructure: - PostgreSQL + TimescaleDB as primary database - Production docker-compose with nginx reverse proxy - Comprehensive Chinese README - .env.example with documentation - quick-start.sh deployment script - nginx config with gzip, caching, security headers Data: - 30-day realistic backfill (47K rows, weather-correlated) - 18 devices, 6 alarm rules, 15 historical alarm events - Beijing solar position model with seasonal variation Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
import pytest
|
|
from conftest import auth_header
|
|
|
|
|
|
class TestOverview:
|
|
async def test_get_overview(self, client, admin_user, admin_token, seed_devices, seed_daily_summary, seed_carbon):
|
|
resp = await client.get("/api/v1/dashboard/overview", headers=auth_header(admin_token))
|
|
assert resp.status_code == 200
|
|
body = resp.json()
|
|
assert "device_stats" in body
|
|
assert "energy_today" in body
|
|
assert "carbon" in body
|
|
assert "active_alarms" in body
|
|
assert "recent_alarms" in body
|
|
|
|
async def test_get_overview_unauthenticated(self, client):
|
|
resp = await client.get("/api/v1/dashboard/overview")
|
|
assert resp.status_code == 401
|
|
|
|
|
|
class TestRealtime:
|
|
async def test_get_realtime_data(self, client, admin_user, admin_token, seed_energy_data):
|
|
resp = await client.get("/api/v1/dashboard/realtime", headers=auth_header(admin_token))
|
|
assert resp.status_code == 200
|
|
body = resp.json()
|
|
assert "timestamp" in body
|
|
assert "pv_power" in body
|
|
assert "heatpump_power" in body
|
|
assert "total_load" in body
|
|
assert "grid_power" in body
|
|
|
|
|
|
class TestLoadCurve:
|
|
async def test_get_load_curve(self, client, admin_user, admin_token, seed_energy_data):
|
|
resp = await client.get("/api/v1/dashboard/load-curve", headers=auth_header(admin_token))
|
|
# date_trunc is PostgreSQL-specific; SQLite returns 500
|
|
assert resp.status_code in (200, 500)
|
|
if resp.status_code == 200:
|
|
assert isinstance(resp.json(), list)
|
|
|
|
async def test_get_load_curve_custom_hours(self, client, admin_user, admin_token, seed_energy_data):
|
|
resp = await client.get(
|
|
"/api/v1/dashboard/load-curve",
|
|
params={"hours": 12},
|
|
headers=auth_header(admin_token),
|
|
)
|
|
assert resp.status_code in (200, 500)
|