def test_get_additional_occurrences(self):
occ_replacer = OccurrenceReplacer([self.occ])
other_occ = Occurrence(event=self.event2,
start=self.start + datetime.timedelta(days=5),
end=self.end,
original_start=self.start,
original_end=self.end)
other_occ.save()
res = occ_replacer.get_additional_occurrences(self.start, self.end)
assert [self.occ] == res
def test_has_occurrence_with_other_event(self):
other_occ = Occurrence(event=self.event2,
start=self.start,
end=self.end,
original_start=self.start,
original_end=self.end)
other_occ.save()
occ_replacer = OccurrenceReplacer([self.occ])
assert occ_replacer.has_occurrence(self.occ)
assert not occ_replacer.has_occurrence(other_occ)
def setUp(self):
weekly = Rule.objects.create(frequency="WEEKLY")
daily = Rule.objects.create(frequency="DAILY")
cal = Calendar.objects.create(name="MyCal")
self.default_tzinfo = timezone.get_default_timezone()
self.start = datetime.datetime.now() - datetime.timedelta(days=10)
self.end = self.start + datetime.timedelta(days=300)
self.event1 = Event(
**{
'title': 'Weekly Event',
'start': self.start,
'end': self.end,
'end_recurring_period': self.end,
'rule': weekly,
'calendar': cal
})
self.event1.save()
self.occ = Occurrence(event=self.event1,
start=self.start,
end=self.end,
original_start=self.start,
original_end=self.end)
self.occ.save()
self.event2 = Event(
**{
'title':
'Recent Event',
'start':
datetime.datetime(2008, 1, 5, 9, 0,
tzinfo=self.default_tzinfo),
'end':
datetime.datetime(
2008, 1, 5, 10, 0, tzinfo=self.default_tzinfo),
'end_recurring_period':
datetime.datetime(2009, 5, 5, 0, 0,
tzinfo=self.default_tzinfo),
'rule':
daily,
'calendar':
cal
})
self.event2.save()
class TestOccurrenceReplacer(TestCase):
def setUp(self):
weekly = Rule.objects.create(frequency="WEEKLY")
daily = Rule.objects.create(frequency="DAILY")
cal = Calendar.objects.create(name="MyCal")
self.default_tzinfo = timezone.get_default_timezone()
self.start = datetime.datetime.now() - datetime.timedelta(days=10)
self.end = self.start + datetime.timedelta(days=300)
self.event1 = Event(
**{
'title': 'Weekly Event',
'start': self.start,
'end': self.end,
'end_recurring_period': self.end,
'rule': weekly,
'calendar': cal
})
self.event1.save()
self.occ = Occurrence(event=self.event1,
start=self.start,
end=self.end,
original_start=self.start,
original_end=self.end)
self.occ.save()
self.event2 = Event(
**{
'title':
'Recent Event',
'start':
datetime.datetime(2008, 1, 5, 9, 0,
tzinfo=self.default_tzinfo),
'end':
datetime.datetime(
2008, 1, 5, 10, 0, tzinfo=self.default_tzinfo),
'end_recurring_period':
datetime.datetime(2009, 5, 5, 0, 0,
tzinfo=self.default_tzinfo),
'rule':
daily,
'calendar':
cal
})
self.event2.save()
def test_has_occurrence(self):
other_occ = Occurrence(event=self.event1,
start=self.start,
end=self.end,
original_start=self.start,
original_end=self.end)
other_occ.save()
occ_replacer = OccurrenceReplacer([self.occ])
assert occ_replacer.has_occurrence(self.occ)
assert occ_replacer.has_occurrence(other_occ)
def test_has_occurrence_with_other_event(self):
other_occ = Occurrence(event=self.event2,
start=self.start,
end=self.end,
original_start=self.start,
original_end=self.end)
other_occ.save()
occ_replacer = OccurrenceReplacer([self.occ])
assert occ_replacer.has_occurrence(self.occ)
assert not occ_replacer.has_occurrence(other_occ)
def test_get_additional_occurrences(self):
occ_replacer = OccurrenceReplacer([self.occ])
other_occ = Occurrence(event=self.event2,
start=self.start + datetime.timedelta(days=5),
end=self.end,
original_start=self.start,
original_end=self.end)
other_occ.save()
res = occ_replacer.get_additional_occurrences(self.start, self.end)
assert [self.occ] == res
def test_get_additional_occurrences_cancelled(self):
occ_replacer = OccurrenceReplacer([self.occ])
self.occ.cancelled = True
self.occ.save()
res = occ_replacer.get_additional_occurrences(self.start, self.end)
assert [] == res
def test_get_occurrence(self):
# self.occ is a persisted Occurrence
occ_replacer = OccurrenceReplacer([self.occ])
res = occ_replacer.get_occurrence(self.occ)
assert res == self.occ
res = occ_replacer.get_occurrence(self.occ)
assert res == self.occ
def test_get_occurrence_works_for_event_like_object(self):
# get_occurrence method checks the duck
expected_ok = False
occ_replacer = OccurrenceReplacer([self.occ])
try:
occ_replacer.get_occurrence(int)
except AttributeError:
expected_ok = True
assert expected_ok