Continued unit tests for GET /review/event/{event_id} Endpoint

This commit is contained in:
Rui Alves 2025-01-12 16:06:35 +00:00
parent 5487058c0c
commit ea364aced3
2 changed files with 12 additions and 9 deletions

View File

@ -124,7 +124,8 @@ class BaseTestHttp(unittest.TestCase):
def insert_mock_event( def insert_mock_event(
self, self,
id: str, id: str,
start_time: datetime.datetime = datetime.datetime.now().timestamp(), start_time: float = datetime.datetime.now().timestamp(),
end_time: float = datetime.datetime.now().timestamp() + 20,
) -> Event: ) -> Event:
"""Inserts a basic event model with a given id.""" """Inserts a basic event model with a given id."""
return Event.insert( return Event.insert(
@ -132,7 +133,7 @@ class BaseTestHttp(unittest.TestCase):
label="Mock", label="Mock",
camera="front_door", camera="front_door",
start_time=start_time, start_time=start_time,
end_time=start_time + 20, end_time=end_time,
top_score=100, top_score=100,
false_positive=False, false_positive=False,
zones=list(), zones=list(),

View File

@ -682,7 +682,7 @@ class TestHttpReview(BaseTestHttp):
response_json, response_json,
) )
def test_review_event_specific_not_found_in_data(self): def test_review_event_not_found_in_data(self):
now = datetime.now().timestamp() now = datetime.now().timestamp()
with TestClient(self.app) as client: with TestClient(self.app) as client:
@ -696,27 +696,29 @@ class TestHttpReview(BaseTestHttp):
response_json, response_json,
) )
def test_review_event_specific(self): def test_review_get_specific_event(self):
now = datetime.now().timestamp() now = datetime.now().timestamp()
with TestClient(self.app) as client: with TestClient(self.app) as client:
id = "123456.random" event_id = "123456.event.random"
super().insert_mock_event(event_id)
review_id = "123456.review.random"
super().insert_mock_review_segment( super().insert_mock_review_segment(
id, now + 1, now + 2, data={"detections": {"event_id": id}} review_id, now + 1, now + 2, data={"detections": {"event_id": event_id}}
) )
response = client.get(f"/review/event/{id}") response = client.get(f"/review/event/{event_id}")
assert response.status_code == 200 assert response.status_code == 200
response_json = response.json() response_json = response.json()
self.assertDictEqual( self.assertDictEqual(
{ {
"id": id, "id": review_id,
"camera": "front_door", "camera": "front_door",
"start_time": now + 1, "start_time": now + 1,
"end_time": now + 2, "end_time": now + 2,
"has_been_reviewed": False, "has_been_reviewed": False,
"severity": SeverityEnum.alert, "severity": SeverityEnum.alert,
"thumb_path": 'False', "thumb_path": 'False',
"data": {"detections": {"event_id": id}}, "data": {"detections": {"event_id": event_id}},
}, },
response_json, response_json,
) )