diff options
author | Alex Grönholm <alex.gronholm@nextday.fi> | 2017-12-22 23:17:53 +0200 |
---|---|---|
committer | Alex Grönholm <alex.gronholm@nextday.fi> | 2017-12-22 23:17:53 +0200 |
commit | b7475d3bf86c63c6f1a5db7997d733a40344247e (patch) | |
tree | 859fdc83f20317b6a168a71c6e6a5f0c118c06a1 /tests | |
parent | 897226ca12e3323a8d3df7a50d0557077e8462be (diff) | |
download | apscheduler-b7475d3bf86c63c6f1a5db7997d733a40344247e.tar.gz |
Added jitter support to combining triggers
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_triggers.py | 37 |
1 files changed, 31 insertions, 6 deletions
diff --git a/tests/test_triggers.py b/tests/test_triggers.py index 7c9f888..af25e38 100644 --- a/tests/test_triggers.py +++ b/tests/test_triggers.py @@ -593,18 +593,31 @@ class TestAndTrigger(object): expected = timezone.localize(expected) if expected else None assert trigger.get_next_fire_time(None, timezone.localize(start_time)) == expected - def test_repr(self, trigger): + def test_jitter(self, trigger, timezone): + trigger.jitter = 5 + start_time = timezone.localize(datetime(2017, 8, 6)) + expected = timezone.localize(datetime(2017, 8, 7)) + for _ in range(100): + next_fire_time = trigger.get_next_fire_time(None, start_time) + assert abs(expected.timestamp() - next_fire_time.timestamp()) <= 5 + + @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter']) + def test_repr(self, trigger, jitter): + trigger.jitter = jitter + jitter_part = ', jitter={}'.format(jitter) if jitter else '' assert repr(trigger) == ( "<AndTrigger([<CronTrigger (month='5-8', day='6-15', " "end_date='2017-08-10 00:00:00 CEST', timezone='Europe/Berlin')>, <CronTrigger " "(month='6-9', day='*/3', end_date='2017-09-07 00:00:00 CEST', " - "timezone='Europe/Berlin')>])>") + "timezone='Europe/Berlin')>]{})>".format(jitter_part)) def test_str(self, trigger): assert str(trigger) == "and[cron[month='5-8', day='6-15'], cron[month='6-9', day='*/3']]" - def test_pickle(self, trigger): + @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter']) + def test_pickle(self, trigger, jitter): """Test that the trigger is pickleable.""" + trigger.jitter = jitter data = pickle.dumps(trigger, 2) trigger2 = pickle.loads(data) @@ -629,18 +642,30 @@ class TestOrTrigger(object): expected = timezone.localize(expected) if expected else None assert trigger.get_next_fire_time(None, timezone.localize(start_time)) == expected - def test_repr(self, trigger): + def test_jitter(self, trigger, timezone): + trigger.jitter = 5 + start_time = expected = timezone.localize(datetime(2017, 8, 6)) + for _ in range(100): + next_fire_time = trigger.get_next_fire_time(None, start_time) + assert abs(expected.timestamp() - next_fire_time.timestamp()) <= 5 + + @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter']) + def test_repr(self, trigger, jitter): + trigger.jitter = jitter + jitter_part = ', jitter={}'.format(jitter) if jitter else '' assert repr(trigger) == ( "<OrTrigger([<CronTrigger (month='5-8', day='6-15', " "end_date='2017-08-10 00:00:00 CEST', timezone='Europe/Berlin')>, <CronTrigger " "(month='6-9', day='*/3', end_date='2017-09-07 00:00:00 CEST', " - "timezone='Europe/Berlin')>])>") + "timezone='Europe/Berlin')>]{})>".format(jitter_part)) def test_str(self, trigger): assert str(trigger) == "or[cron[month='5-8', day='6-15'], cron[month='6-9', day='*/3']]" - def test_pickle(self, trigger): + @pytest.mark.parametrize('jitter', [None, 5], ids=['nojitter', 'jitter']) + def test_pickle(self, trigger, jitter): """Test that the trigger is pickleable.""" + trigger.jitter = jitter data = pickle.dumps(trigger, 2) trigger2 = pickle.loads(data) |