Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-02-05 00:15:51 +01:00
Code Issues Projects Releases Wiki Activity

Added unit tests for GET /review/{review_id}/viewed Endpoint

Browse Source
This commit is contained in:
Rui Alves 2025-01-12 17:53:40 +00:00
parent 056d722e93
commit bd746cdaf7
2 changed files with 38 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/api/review.py
View File

@ -490,8 +490,6 @@ def set_not_reviewed(review_id: str):
review.save() review.save()
return JSONResponse( return JSONResponse(
content=( content=({"success": True, "message": f"Set Review {review_id} as not viewed"}),
{"success": True, "message": "Set Review " + review_id + " as not viewed"}
),
status_code=200, status_code=200,
) )

39
frigate/test/http_api/test_http_review.py
View File

@ -717,7 +717,7 @@ class TestHttpReview(BaseTestHttp):
"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": event_id}}, "data": {"detections": {"event_id": event_id}},
}, },
response_json, response_json,
@ -753,9 +753,44 @@ class TestHttpReview(BaseTestHttp):
"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": {}, "data": {},
}, },
response_json, response_json,
) )
####################################################################################################################
################################### DELETE /review/{review_id}/viewed Endpoint ##################################
####################################################################################################################
def test_delete_review_viewed_review_not_found(self):
with TestClient(self.app) as client:
review_id = "123456.random"
response = client.delete(f"/review/{review_id}/viewed")
assert response.status_code == 404
response_json = response.json()
self.assertDictEqual(
{"success": False, "message": f"Review {review_id} not found"},
response_json,
)
def test_delete_review_viewed(self):
now = datetime.now().timestamp()
with TestClient(self.app) as client:
review_id = "123456.review.random"
super().insert_mock_review_segment(
review_id, now + 1, now + 2, has_been_reviewed=True
)
review_before = ReviewSegment.get(ReviewSegment.id == review_id)
assert review_before.has_been_reviewed == True
response = client.delete(f"/review/{review_id}/viewed")
assert response.status_code == 200
response_json = response.json()
self.assertDictEqual(
{"success": True, "message": f"Set Review {review_id} as not viewed"},
response_json,
)
review_after = ReviewSegment.get(ReviewSegment.id == review_id)
assert review_after.has_been_reviewed == False