add more coverage

tests/test_utils.py

@@ -3,8 +3,8 @@
 from django.test import TestCase
 from django.utils import timezone
 
-from schedule.models import Event, Rule, Calendar
-from schedule.utils import EventListManager
+from schedule.models import Event, Rule, Calendar, Occurrence
+from schedule.utils import EventListManager, OccurrenceReplacer
 
 
 class TestEventListManager(TestCase):
@@ -47,3 +47,86 @@ def test_occurrences_after(self):
         self.assertEqual(next(occurrences).event, self.event1)
         occurrences = eml.occurrences_after()
         self.assertEqual(list(occurrences), [])
+
+
+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