From e34a34acaffba4d93a12a0a5d1d216bbd76d3207 Mon Sep 17 00:00:00 2001 From: pablo Date: Sat, 2 Jan 2021 23:49:10 +0100 Subject: [PATCH] Fix in throttling test so it doesn't fail around midnight. --- tests/throttling_test.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/throttling_test.py b/tests/throttling_test.py index 3a9f916..0c555af 100644 --- a/tests/throttling_test.py +++ b/tests/throttling_test.py @@ -11,8 +11,8 @@ from core.throttling_utils import ( def test_working_hours_throttling_rule_checks(): working_hours_rule = WorkingHoursThrottlingRule( working_hours={ - "start": datetime.datetime.now().time(), - "end": (datetime.datetime.now() + datetime.timedelta(hours=1)).time(), + "start": (datetime.datetime.now() + datetime.timedelta(seconds=-5)).time(), + "end": (datetime.datetime.now() + datetime.timedelta(seconds=5)).time(), } ) @@ -86,8 +86,10 @@ def test_throttle_manager_checks_rules(): some_rules = [ WorkingHoursThrottlingRule( working_hours={ - "start": datetime.datetime.now().time(), - "end": (datetime.datetime.now() + datetime.timedelta(hours=1)).time(), + "start": ( + datetime.datetime.now() + datetime.timedelta(seconds=-5) + ).time(), + "end": (datetime.datetime.now() + datetime.timedelta(seconds=5)).time(), } ), CooldownThrottlingRule(cooldown_time_generator=lambda: 0),