def test_daily_never_ends_different_time_zone(self):
data = {
'freq': rrule.DAILY,
'interval': 1,
'dtstart': '6/1/2017',
'byhour': '22',
'time_zone': 'US/Eastern',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rrule_model.next_occurrence,
datetime.datetime(2017, 6, 2, 2))
self.assertEqual(rule[0].replace(tzinfo=None),
datetime.datetime(2017, 6, 1, 22))
self.assertEqual(rule[1].replace(tzinfo=None),
datetime.datetime(2017, 6, 2, 22))
self.assertEqual(rule[365].replace(tzinfo=None),
datetime.datetime(2018, 6, 1, 22))
# Advance the occurrence
rrule_model.update_next_occurrence()
self.assertEqual(rrule_model.next_occurrence,
datetime.datetime(2017, 6, 3, 2))
def test_weekly_mwf_every_2_weeks_with_end(self):
data = {
'freq': rrule.WEEKLY,
'interval': 2,
'dtstart': '6/1/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.ON,
'until': '7/10/2017',
'byweekday': '[0, 2, 4]',
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule.count(), 8)
self.assertEqual(rule[0], datetime.datetime(2017, 6, 2))
self.assertEqual(rule[1], datetime.datetime(2017, 6, 12))
self.assertEqual(rule[2], datetime.datetime(2017, 6, 14))
self.assertEqual(rule[3], datetime.datetime(2017, 6, 16))
self.assertEqual(rule[4], datetime.datetime(2017, 6, 26))
self.assertEqual(rule[5], datetime.datetime(2017, 6, 28))
self.assertEqual(rule[6], datetime.datetime(2017, 6, 30))
self.assertEqual(rule[7], datetime.datetime(2017, 7, 10))
def test_monthly_last_day_of_month(self):
data = {
'freq': rrule.MONTHLY,
'interval': 1,
'dtstart': '6/4/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
'repeat_by': 'DAY_OF_THE_MONTH_END',
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule[0], datetime.datetime(2017, 6, 30))
self.assertEqual(rule[1], datetime.datetime(2017, 7, 31))
self.assertEqual(rule[2], datetime.datetime(2017, 8, 31))
self.assertEqual(rule[3], datetime.datetime(2017, 9, 30))
self.assertEqual(rule[4], datetime.datetime(2017, 10, 31))
self.assertEqual(rule[5], datetime.datetime(2017, 11, 30))
self.assertEqual(rule[6], datetime.datetime(2017, 12, 31))
self.assertEqual(rule[7], datetime.datetime(2018, 1, 31))
self.assertEqual(rule[8], datetime.datetime(2018, 2, 28))
self.assertEqual(rule[9], datetime.datetime(2018, 3, 31))
self.assertEqual(rule[10], datetime.datetime(2018, 4, 30))
self.assertEqual(rule[11], datetime.datetime(2018, 5, 31))
self.assertEqual(rule[12], datetime.datetime(2018, 6, 30))
self.assertEqual(rule[24], datetime.datetime(2019, 6, 30))
self.assertEqual(rule[32], datetime.datetime(2020, 2, 29))
def test_update_from_id_field(self):
"""
Verifies that an existing rrule object will get updated from passing an id in the form data and that
the next recurrence is refreshed
"""
data = {
'freq': rrule.DAILY,
'interval': 1,
'dtstart': '6/4/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save(
occurrence_handler_path='ambition_utils.rrule.handler.OccurrenceHandler',
)
self.assertEqual(RRule.objects.count(), 1)
self.assertEqual(rrule_model.next_occurrence, datetime.datetime(2017, 6, 4))
# Handle update
data = {
'rrule': str(rrule_model.id),
'freq': rrule.DAILY,
'interval': 1,
'dtstart': '6/7/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save(
occurrence_handler_path='ambition_utils.rrule.handler.OccurrenceHandler',
)
self.assertEqual(RRule.objects.count(), 1)
self.assertEqual(rrule_model.next_occurrence, datetime.datetime(2017, 6, 7))
def test_daily_never_ends(self):
data = {
'freq': rrule.DAILY,
'interval': 1,
'dtstart': '6/1/2017',
'byhour': '3',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule[0].replace(tzinfo=None), datetime.datetime(2017, 6, 1, 3))
self.assertEqual(rule[1].replace(tzinfo=None), datetime.datetime(2017, 6, 2, 3))
self.assertEqual(rule[365].replace(tzinfo=None), datetime.datetime(2018, 6, 1, 3))
def test_yearly(self):
data = {
'freq': rrule.YEARLY,
'interval': 1,
'dtstart': '6/4/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule[0], datetime.datetime(2017, 6, 4))
self.assertEqual(rule[1], datetime.datetime(2018, 6, 4))
self.assertEqual(rule[2], datetime.datetime(2019, 6, 4))
def test_monthly_every_month_day_of_month(self):
data = {
'freq': rrule.MONTHLY,
'interval': 1,
'dtstart': '6/4/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
'repeat_by': 'DAY_OF_THE_MONTH',
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule[0], datetime.datetime(2017, 6, 4))
self.assertEqual(rule[1], datetime.datetime(2017, 7, 4))
self.assertEqual(rule[2], datetime.datetime(2017, 8, 4))
def test_daily_with_end_date(self):
data = {
'freq': rrule.DAILY,
'interval': 6,
'dtstart': '6/1/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.ON,
'until': '6/10/2017',
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule.count(), 2)
self.assertEqual(rule[0], datetime.datetime(2017, 6, 1))
self.assertEqual(rule[1], datetime.datetime(2017, 6, 7))
def test_daily_every_4_days_3_times(self):
data = {
'freq': rrule.DAILY,
'interval': 4,
'dtstart': '6/1/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.AFTER,
'count': 3,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule.count(), 3)
self.assertEqual(rule[0], datetime.datetime(2017, 6, 1))
self.assertEqual(rule[1], datetime.datetime(2017, 6, 5))
self.assertEqual(rule[2], datetime.datetime(2017, 6, 9))
def test_kwargs_setting(self):
"""
Verifies that kwargs passed to the save method get set or ignored if they don't exist on the model
"""
data = {
'freq': rrule.YEARLY,
'interval': 1,
'dtstart': '6/4/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save(
occurrence_handler_path='ambition_utils.rrule.handler.OccurrenceHandler',
fake_recurrence=form
)
self.assertEqual(rrule_model.occurrence_handler_path, 'ambition_utils.rrule.handler.OccurrenceHandler')
def test_monthly_every_month_day_of_week_from_end_of_month(self):
"""
Second Monday from end of month every other month
"""
data = {
'freq': rrule.MONTHLY,
'interval': 2,
'dtstart': '6/11/2017',
'byhour': '0',
'time_zone': 'UTC',
'ends': RecurrenceEnds.NEVER,
'repeat_by': 'DAY_OF_THE_WEEK_END',
'bynweekday': '[[0, -2]]'
}
form = RecurrenceForm(data=data)
self.assertTrue(form.is_valid())
rrule_model = form.save()
rule = rrule_model.get_rrule()
self.assertEqual(rule[0], datetime.datetime(2017, 6, 19))
self.assertEqual(rule[1], datetime.datetime(2017, 8, 21))
self.assertEqual(rule[2], datetime.datetime(2017, 10, 23))