this is a push

This commit is contained in:
2026-08-13 15:03:34 +08:00
parent 0f55259fff
commit ca61c132f2
109 changed files with 44036 additions and 198 deletions

58
test/test_events.py Normal file
View File

@@ -0,0 +1,58 @@
from __future__ import annotations
import unittest
from backend.events import EventStore
class EventStoreTests(unittest.TestCase):
def setUp(self) -> None:
self.store = EventStore(max_events=2)
def create_event(self, session_id: str = "session") -> dict:
return self.store.create(
session_id=session_id,
classes=["fire"],
detections=[{"class": "fire", "confidence": 0.82}],
notification_channels={"wechat": True, "feishu": False},
)
def test_create_and_summarize_event(self) -> None:
event = self.create_event()
summary = self.store.summary()
self.assertEqual(event["status"], "pending")
self.assertEqual(event["max_confidence"], 0.82)
self.assertEqual(summary["total"], 1)
self.assertEqual(summary["pending"], 1)
self.assertEqual(summary["fire"], 1)
def test_update_event_status(self) -> None:
event = self.create_event()
updated = self.store.update(event["id"], "resolved")
self.assertIsNotNone(updated)
self.assertEqual(updated["status"], "resolved")
self.assertIsNotNone(updated["handled_at"])
def test_store_respects_maximum_size(self) -> None:
first = self.create_event("first")
self.create_event("second")
self.create_event("third")
event_ids = {event["id"] for event in self.store.list()}
self.assertEqual(len(event_ids), 2)
self.assertNotIn(first["id"], event_ids)
def test_invalid_status_is_rejected(self) -> None:
event = self.create_event()
with self.assertRaisesRegex(ValueError, "Unsupported"):
self.store.update(event["id"], "invalid")
if __name__ == "__main__":
unittest.main()