def test_1_weeks_mon():
now = timezone.localize(datetime(2020, 2, 1, 0, 0))
trigger = IntervalCronTrigger(weeks=1, day_of_week='mon', hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 3, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
def test_start_date():
now = timezone.localize(datetime(2020, 2, 1, 2, 0))
trigger = IntervalCronTrigger(days=3, hour=7, minute=3, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 4, 7, 3))
actual = trigger.get_next_fire_time(timezone.localize(datetime(2020, 2, 1, 7, 7)), timezone.localize(datetime(2020, 2, 1, 7, 10)))
assert expected == actual
def test_3_months_1stmon():
now = timezone.localize(datetime(2020, 2, 1, 3, 0))
trigger = IntervalCronTrigger(months=3, day='1st mon', hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 3, 2, 0))
actual = trigger.get_next_fire_time(now, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 5, 4, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_1_months_13th():
now = timezone.localize(datetime(2020, 2, 1, 0, 0))
trigger = IntervalCronTrigger(months=1, day='13', hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 13, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 3, 13, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_now():
start = timezone.localize(datetime.now())
trigger = IntervalCronTrigger(days=1, start_date=start, timezone=timezone)
expected = timezone.localize(datetime.combine(datetime.now().date(), time())) + timedelta(days=1)
actual = trigger.get_next_fire_time(None, start)
assert expected == actual
expected = timezone.localize(datetime.combine(start.date(), time())) + timedelta(days=2)
actual = trigger.get_next_fire_time(actual, actual + timedelta(minutes=5))
assert expected == actual
def test_large_gap_1():
start = timezone.localize(datetime(2020, 2, 1, 1, 0))
trigger = IntervalCronTrigger(days=1, hour=2, start_date=start, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 2, 2, 0))
actual = trigger.get_next_fire_time(None, start)
assert expected == actual
expected = timezone.localize(datetime(2020, 2, 9, 2, 0))
actual = trigger.get_next_fire_time(timezone.localize(datetime(2020, 2, 2, 2, 3)), timezone.localize(datetime(2020, 2, 8, 2, 9)))
assert expected == actual
def test_next_iter_only():
now = timezone.localize(datetime(2020, 2, 1, 2, 0))
trigger = IntervalCronTrigger(days=1, hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 2, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 2, 3, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_1_months_firstday_2():
now = timezone.localize(datetime(2020, 2, 1, 0, 0))
trigger = IntervalCronTrigger(months=1, day='1', start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 3, 1, 0, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 4, 1, 0, 0))
actual = trigger.get_next_fire_time(timezone.localize(datetime(2020, 3, 1, 0, 5)), timezone.localize(datetime(2020, 3, 1, 2, 5)))
assert expected == actual
def test_large_gap_2():
start = timezone.localize(datetime(2020, 2, 1, 1, 0))
trigger = IntervalCronTrigger(weeks=1, day_of_week='mon', hour=2, start_date=start, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 3, 2, 0))
actual = trigger.get_next_fire_time(None, start)
assert expected == actual
expected = timezone.localize(datetime(2020, 2, 10, 2, 0))
actual = trigger.get_next_fire_time(timezone.localize(datetime(2020, 2, 3, 2, 3)), timezone.localize(datetime(2020, 2, 5, 7, 9)))
assert expected == actual
expected = timezone.localize(datetime(2020, 2, 17, 2, 0))
actual = trigger.get_next_fire_time(timezone.localize(datetime(2020, 2, 3, 2, 3)), timezone.localize(datetime(2020, 2, 10, 7, 9)))
assert expected == actual
def test_every_2yrs_on_1st_mon():
now = timezone.localize(datetime(2000, 1, 1, 0, 0))
trigger = IntervalCronTrigger(months=24, day="1st mon", hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2000, 1, 3, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2002, 1, 7, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2004, 1, 5, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_every_3mo_on_last_fri():
now = timezone.localize(datetime(2020, 1, 1, 0, 0))
trigger = IntervalCronTrigger(months=3, day="last fri", hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 1, 31, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 4, 24, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2020, 7, 31, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_1_months_lastday():
now = timezone.localize(datetime(2020, 2, 13, 0, 0))
trigger = IntervalCronTrigger(months=1, day='last', start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 2, 29, 0, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 3, 31, 0, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2020, 4, 30, 0, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2020, 5, 31, 0, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
def test_every_3_day():
now = timezone.localize(datetime(2020, 3, 7, 0, 0))
trigger = IntervalCronTrigger(days=3, hour=2, start_date=now, timezone=timezone)
expected = timezone.localize(datetime(2020, 3, 10, 2, 0))
actual = trigger.get_next_fire_time(None, now)
assert expected == actual
expected = timezone.localize(datetime(2020, 3, 13, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2020, 3, 16, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual
expected = timezone.localize(datetime(2020, 3, 19, 2, 0))
actual = trigger.get_next_fire_time(actual, actual)
assert expected == actual