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),