vigilar/tests/unit/test_package_schema.py

22 lines
609 B
Python

from vigilar.constants import EventType
from vigilar.storage.schema import package_events
def test_package_delivered_event_type():
assert EventType.PACKAGE_DELIVERED == "PACKAGE_DELIVERED"
def test_package_reminder_event_type():
assert EventType.PACKAGE_REMINDER == "PACKAGE_REMINDER"
def test_package_collected_event_type():
assert EventType.PACKAGE_COLLECTED == "PACKAGE_COLLECTED"
def test_package_events_table_columns():
col_names = [c.name for c in package_events.columns]
assert "camera_id" in col_names
assert "detected_at" in col_names
assert "status" in col_names