222 lines
8.0 KiB
Python
222 lines
8.0 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
from src.people_flow.manage_api import create_app
|
|
|
|
|
|
def build_client(project_root: Path):
|
|
config_path = project_root / "config" / "local.yaml"
|
|
config_path.parent.mkdir(parents=True, exist_ok=True)
|
|
config_path.write_text(
|
|
"runtime:\n"
|
|
" rtsp_url: rtsp://before-update\n"
|
|
" output_dir: outputs\n"
|
|
"rtsp:\n"
|
|
" output_subdir: rtsp_stream\n"
|
|
"queue:\n"
|
|
" source_id: queue_cam_01\n"
|
|
" queue_time_threshold_seconds: 300\n"
|
|
" crowded_count_threshold: 5\n"
|
|
" normal_count_threshold: 2\n"
|
|
"webhook:\n"
|
|
" url: https://example.test/webhook\n"
|
|
" event_log_path: outputs/rtsp_stream/webhook_events.jsonl\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
rtsp_dir = project_root / "outputs" / "rtsp_stream"
|
|
windows_dir = rtsp_dir / "windows"
|
|
windows_dir.mkdir(parents=True, exist_ok=True)
|
|
latest_payload = {
|
|
"event": "half_hour_report",
|
|
"source_type": "rtsp",
|
|
"source_id": "queue_cam_01",
|
|
"window_start": "2026-04-16T09:30:00+08:00",
|
|
"window_end": "2026-04-16T10:00:00+08:00",
|
|
"total_people": 7,
|
|
"age_counts": {"minor": 1, "adult": 5, "senior": 1},
|
|
"gender_counts": {"male": 4, "female": 3},
|
|
"unknown_attributes": 2,
|
|
"queue_metrics": {
|
|
"queue_time_threshold_seconds": 300,
|
|
"over_threshold_count": 6,
|
|
"under_threshold_count": 2,
|
|
"queue_level": "crowded",
|
|
"queue_level_label": "人多",
|
|
"previous_queue_level": "normal",
|
|
"previous_queue_level_label": "人数正常",
|
|
"status_change": "queue_increased",
|
|
"status_change_label": "人数变多",
|
|
},
|
|
"tracks": [
|
|
{"track_id": 1, "direction": "in"},
|
|
{"track_id": 2, "direction": "out"},
|
|
{"track_id": 3, "direction": "in"},
|
|
],
|
|
}
|
|
(rtsp_dir / "latest.json").write_text(
|
|
json.dumps(latest_payload),
|
|
encoding="utf-8",
|
|
)
|
|
(windows_dir / "stats_2026-04-16_09-00-00.json").write_text(
|
|
json.dumps(
|
|
{
|
|
"window_start": "2026-04-16T09:00:00+08:00",
|
|
"window_end": "2026-04-16T09:30:00+08:00",
|
|
"total_people": 5,
|
|
"queue_metrics": {
|
|
"queue_time_threshold_seconds": 300,
|
|
"over_threshold_count": 2,
|
|
"under_threshold_count": 1,
|
|
"queue_level": "normal",
|
|
"queue_level_label": "人数正常",
|
|
"previous_queue_level": None,
|
|
"previous_queue_level_label": "",
|
|
"status_change": "initial",
|
|
"status_change_label": "初始",
|
|
},
|
|
"age_counts": {"minor": 0, "adult": 4, "senior": 1},
|
|
"gender_counts": {"male": 2, "female": 3},
|
|
"unknown_attributes": 1,
|
|
}
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
(windows_dir / "stats_2026-04-16_09-30-00.json").write_text(
|
|
json.dumps(latest_payload),
|
|
encoding="utf-8",
|
|
)
|
|
(project_root / "outputs" / "rtsp_run.log").write_text(
|
|
"rtsp ok\n", encoding="utf-8"
|
|
)
|
|
(rtsp_dir / "webhook_events.jsonl").write_text(
|
|
json.dumps(latest_payload) + "\n", encoding="utf-8"
|
|
)
|
|
|
|
app = create_app(config_path)
|
|
app.testing = True
|
|
return app.test_client(), config_path
|
|
|
|
|
|
def test_get_manage_health(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/health")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["status"] == "ok"
|
|
assert response.json["project_type"] == "people_flow_project"
|
|
assert response.json["runtime_status"] == "running"
|
|
|
|
|
|
def test_get_manage_config(tmp_path: Path):
|
|
client, config_path = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/config")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["config_path"] == str(config_path)
|
|
assert response.json["runtime"]["rtsp_url"] == "rtsp://before-update"
|
|
assert response.json["rtsp"]["output_subdir"] == "rtsp_stream"
|
|
assert response.json["queue"]["source_id"] == "queue_cam_01"
|
|
assert response.json["webhook"]["url"] == "https://example.test/webhook"
|
|
|
|
|
|
def test_put_manage_config_updates_rtsp_url(tmp_path: Path):
|
|
client, config_path = build_client(tmp_path)
|
|
|
|
response = client.put(
|
|
"/api/manage/config",
|
|
json={"rtsp_url": "rtsp://after-update"},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["runtime"]["rtsp_url"] == "rtsp://after-update"
|
|
|
|
saved = yaml.safe_load(config_path.read_text(encoding="utf-8"))
|
|
assert saved["runtime"]["rtsp_url"] == "rtsp://after-update"
|
|
|
|
|
|
def test_get_manage_summary(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/summary")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["result_type"] == "people_flow_project"
|
|
assert response.json["last_result_time"] == "2026-04-16T10:00:00+08:00"
|
|
assert response.json["metrics"]["total_people"] == 7
|
|
assert response.json["metrics"]["queue_level"] == "crowded"
|
|
assert response.json["metrics"]["queue_level_label"] == "人多"
|
|
assert response.json["metrics"]["previous_queue_level"] == "normal"
|
|
assert response.json["metrics"]["previous_queue_level_label"] == "人数正常"
|
|
assert response.json["metrics"]["over_threshold_count"] == 6
|
|
assert response.json["metrics"]["under_threshold_count"] == 2
|
|
assert response.json["metrics"]["status_change"] == "queue_increased"
|
|
assert response.json["metrics"]["status_change_label"] == "人数变多"
|
|
assert response.json["metrics"]["direction_counts"] == {"in": 2, "out": 1}
|
|
assert (
|
|
response.json["metrics"]["recent_window_stats"][0]["window_end"]
|
|
== "2026-04-16T10:00:00+08:00"
|
|
)
|
|
|
|
|
|
def test_get_manage_windows(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/windows?page=1&page_size=1")
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["total"] == 2
|
|
assert response.json["page"] == 1
|
|
assert response.json["page_size"] == 1
|
|
assert response.json["items"][0]["window_end"] == "2026-04-16T10:00:00+08:00"
|
|
assert response.json["items"][0]["total_people"] == 7
|
|
assert response.json["items"][0]["queue_level"] == "crowded"
|
|
assert response.json["items"][0]["queue_level_label"] == "人多"
|
|
assert response.json["items"][0]["previous_queue_level"] == "normal"
|
|
assert response.json["items"][0]["previous_queue_level_label"] == "人数正常"
|
|
assert response.json["items"][0]["status_change"] == "queue_increased"
|
|
assert response.json["items"][0]["status_change_label"] == "人数变多"
|
|
|
|
|
|
def test_get_manage_files(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/files")
|
|
|
|
assert response.status_code == 200
|
|
assert {item["path"] for item in response.json["files"]} == {
|
|
"outputs/rtsp_run.log",
|
|
"outputs/rtsp_stream/latest.json",
|
|
"outputs/rtsp_stream/webhook_events.jsonl",
|
|
"outputs/rtsp_stream/windows/stats_2026-04-16_09-00-00.json",
|
|
"outputs/rtsp_stream/windows/stats_2026-04-16_09-30-00.json",
|
|
}
|
|
|
|
|
|
def test_get_manage_files_preview(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get(
|
|
"/api/manage/files/preview?path=outputs/rtsp_stream/latest.json&lines=1"
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
assert response.json["path"] == "outputs/rtsp_stream/latest.json"
|
|
assert response.json["count"] == 1
|
|
assert "total_people" in response.json["lines"][0]
|
|
|
|
|
|
def test_get_manage_files_download(tmp_path: Path):
|
|
client, _ = build_client(tmp_path)
|
|
|
|
response = client.get("/api/manage/files/download?path=outputs/rtsp_run.log")
|
|
|
|
assert response.status_code == 200
|
|
assert response.data == b"rtsp ok\n"
|