counterweight/drogon
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix in throttling test so it doesn't fail around midnight.

Browse source
This commit is contained in:
pablo 2021-01-02 23:49:10 +01:00
parent def858ef6a
commit e34a34acaf
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/throttling_test.py
View file

@ -11,8 +11,8 @@ from core.throttling_utils import (
def test_working_hours_throttling_rule_checks(): def test_working_hours_throttling_rule_checks():
working_hours_rule = WorkingHoursThrottlingRule( working_hours_rule = WorkingHoursThrottlingRule(
working_hours={ working_hours={
"start": datetime.datetime.now().time(), "start": (datetime.datetime.now() + datetime.timedelta(seconds=-5)).time(),
"end": (datetime.datetime.now() + datetime.timedelta(hours=1)).time(), "end": (datetime.datetime.now() + datetime.timedelta(seconds=5)).time(),
} }
) )
@ -86,8 +86,10 @@ def test_throttle_manager_checks_rules():
some_rules = [ some_rules = [
WorkingHoursThrottlingRule( WorkingHoursThrottlingRule(
working_hours={ working_hours={
"start": datetime.datetime.now().time(), "start": (
"end": (datetime.datetime.now() + datetime.timedelta(hours=1)).time(), datetime.datetime.now() + datetime.timedelta(seconds=-5)
).time(),
"end": (datetime.datetime.now() + datetime.timedelta(seconds=5)).time(),
} }
), ),
CooldownThrottlingRule(cooldown_time_generator=lambda: 0), CooldownThrottlingRule(cooldown_time_generator=lambda: 0),