Added test for review endpoint (time filter - after + before)

This commit is contained in:
Rui Alves 2024-11-19 20:17:32 +00:00
parent 654997470f
commit 3415798191

View File

@ -1,3 +1,5 @@
import datetime
from fastapi.testclient import TestClient
from frigate.models import Event, ReviewSegment
@ -8,7 +10,7 @@ class TestHttpReview(BaseTestHttp):
def setUp(self):
super().setUp([Event, ReviewSegment])
# Does not return any data point since the end time (before parameter) is not passed
# Does not return any data point since the end time (before parameter) is not passed and the review segment end_time is 20 seconds from now
def test_get_reviews_no_filters(self):
app = super().create_app()
@ -17,3 +19,18 @@ class TestHttpReview(BaseTestHttp):
reviews_response = client.get("/review")
assert reviews_response.status_code == 200
assert len(reviews_response.json()) == 0
def test_get_with_time_filter(self):
app = super().create_app()
now = datetime.datetime.now().timestamp()
with TestClient(app) as client:
super().insert_mock_review_segment("4566.random", now, now + 20)
params = {
"after": now - 1,
"before": now + 21,
}
reviews_response = client.get("/review", params=params)
assert reviews_response.status_code == 200
reviews_in_response = reviews_response.json()
assert len(reviews_in_response) == 1