mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-10 06:23:43 +08:00
firestar5683
14e42178df
Expose Sentry events and notification presentation through Galaxy. Co-authored-by: Prabhaav Pillai <prabhaav@msu.edu> Co-authored-by: dirwin31 <83434411+dirwin31@users.noreply.github.com>
60 lines
2.6 KiB
Python
60 lines
2.6 KiB
Python
"""Local motion records are projected without exposing storage or session details."""
|
|
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from openpilot.starpilot.galaxy.sentry_events import SentryEvents, SentryEventsUnavailable
|
|
from openpilot.starpilot.sentry_mode.storage import EventStore
|
|
|
|
|
|
class SentryEventsTest(unittest.TestCase):
|
|
def test_actual_store_projection_and_empty_store(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
store = EventStore(Path(directory).resolve() / "events")
|
|
self.assertEqual(SentryEvents(store).snapshot()["events"], [])
|
|
receipt = store.record("warning", 123, permitted=lambda: True)
|
|
result = SentryEvents(store).snapshot()
|
|
self.assertEqual(result["schemaVersion"], 1)
|
|
self.assertEqual(result["source"], "local")
|
|
self.assertEqual(result["capacity"], 512)
|
|
self.assertEqual(result["events"][0]["eventId"], receipt.event_id)
|
|
self.assertEqual(set(result["events"][0]), {"eventId", "kind", "systemTimeMs", "images"})
|
|
self.assertEqual(result["events"][0]["kind"], "warning")
|
|
self.assertEqual(result["events"][0]["images"], [])
|
|
|
|
def test_published_images_project_after_restart_and_use_same_event_id(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory).resolve() / "events"
|
|
image = b"\xff\xd8fixture\xff\xd9"
|
|
receipt = EventStore(root).record("alarm", 456, permitted=lambda: True, images={"cabin": image})
|
|
projection = SentryEvents(EventStore(root))
|
|
event = projection.snapshot()["events"][0]
|
|
self.assertEqual(event["eventId"], receipt.event_id)
|
|
self.assertEqual(event["images"], ["cabin"])
|
|
self.assertEqual(projection.image(event["eventId"], "cabin"), image)
|
|
with self.assertRaises(SentryEventsUnavailable):
|
|
projection.image(event["eventId"], "wide")
|
|
|
|
def test_invalid_or_oversized_snapshot_is_unavailable(self):
|
|
class Fixture:
|
|
def __init__(self, value):
|
|
self.value = value
|
|
|
|
def snapshot(self):
|
|
return self.value
|
|
|
|
base = {"version": 1, "incomplete": False, "capacity": 512, "events": []}
|
|
event = {"eventId": "a" * 32, "kind": "alarm", "wallTimeNs": 123_000_000}
|
|
for value in ({**base, "version": True}, {**base, "capacity": True},
|
|
{**base, "events": [event] * 513},
|
|
{**base, "events": [{**event, "kind": "unknown"}]},
|
|
{**base, "events": [{**event, "wallTimeNs": -1}]}):
|
|
with self.subTest(value=value):
|
|
with self.assertRaises(SentryEventsUnavailable):
|
|
SentryEvents(Fixture(value)).snapshot()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|