Minor fixes

This commit is contained in:
Rui Alves 2024-11-24 15:16:05 +00:00
parent 6759b5c9bc
commit 77f64fef8c
2 changed files with 6 additions and 5 deletions

View File

@ -146,8 +146,8 @@ class BaseTestHttp(unittest.TestCase):
def insert_mock_review_segment(
self,
id: str,
start_time: datetime.datetime = datetime.datetime.now().timestamp(),
end_time: datetime.datetime = datetime.datetime.now().timestamp() + 20,
start_time: float = datetime.datetime.now().timestamp(),
end_time: float = datetime.datetime.now().timestamp() + 20,
severity: SeverityEnum = SeverityEnum.alert,
has_been_reviewed: bool = False,
) -> Event:

View File

@ -251,6 +251,7 @@ class TestHttpReview(BaseTestHttp):
five_days_ago = datetime.today() - timedelta(days=5)
twenty_days_ago = datetime.today() - timedelta(days=20)
one_month_ago = datetime.today() - timedelta(days=30)
one_month_ago_ts = one_month_ago.timestamp()
with TestClient(self.app) as client:
super().insert_mock_review_segment("123456.random", now.timestamp())
@ -259,7 +260,7 @@ class TestHttpReview(BaseTestHttp):
)
super().insert_mock_review_segment(
"123458.random",
twenty_days_ago.timestamp(),
one_month_ago_ts,
None,
SeverityEnum.detection,
)
@ -270,9 +271,9 @@ class TestHttpReview(BaseTestHttp):
None,
SeverityEnum.detection,
)
# This won't appear in the output since it's within last month start_time clause (review.start_time > month_ago)
# This won't appear in the output since it's not within last month start_time clause (review.start_time > month_ago)
super().insert_mock_review_segment(
"123450.random", one_month_ago.timestamp() + 0
"123450.random", one_month_ago_ts
)
review_summary_request = client.get("/review/summary")
assert review_summary_request.status_code == 200