Пример #1
0
    def setUp(self):
        weekly = Rule(frequency = "WEEKLY")
        weekly.save()
        daily = Rule(frequency = "DAILY")
        daily.save()
        cal = Room(name="MyCal")
        cal.save()

        self.reservation1 = Reservation(**{
                'title': 'Weekly Reservation',
                'start': datetime.datetime(2009, 4, 1, 8, 0),
                'end': datetime.datetime(2009, 4, 1, 9, 0),
                'end_recurring_period' : datetime.datetime(2009, 10, 5, 0, 0),
                'rule': weekly,
                'room': cal
               })
        self.reservation1.save()
        self.reservation2 = Reservation(**{
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 9, 0),
                'end': datetime.datetime(2008, 1, 5, 10, 0),
                'end_recurring_period' : datetime.datetime(2009, 5, 5, 0, 0),
                'rule': daily,
                'room': cal
               })
        self.reservation2.save()
class TestOccurrencePool(TestCase):
    def setUp(self):
        rule = Rule(frequency="WEEKLY")
        rule.save()
        cal = Room(name="MyCal")
        cal.save()
        data = {
            "title": "Recent Reservation",
            "start": datetime.datetime(2008, 1, 5, 8, 0),
            "end": datetime.datetime(2008, 1, 5, 9, 0),
            "end_recurring_period": datetime.datetime(2008, 5, 5, 0, 0),
            "rule": rule,
            "room": cal,
        }
        self.recurring_reservation = Reservation(**data)
        self.recurring_reservation.save()

    def testPeriodFromPool(self):
        """
            Test that period initiated with occurrence_pool returns the same occurrences as "straigh" period
            in a corner case whereby a period's start date is equal to the occurrence's end date
        """
        start = datetime.datetime(2008, 1, 5, 9, 0)
        end = datetime.datetime(2008, 1, 5, 10, 0)
        parent_period = Period(Reservation.objects.all(), start, end)
        period = Period(
            parent_period.reservations, start, end, parent_period.get_persisted_occurrences(), parent_period.occurrences
        )
        self.assertEquals(parent_period.occurrences, period.occurrences)
Пример #3
0
    def setUp(self):
        weekly = Rule(frequency="WEEKLY")
        weekly.save()
        daily = Rule(frequency="DAILY")
        daily.save()
        cal = Room(name="MyCal")
        cal.save()

        self.reservation1 = Reservation(
            **{
                'title': 'Weekly Reservation',
                'start': datetime.datetime(2009, 4, 1, 8, 0),
                'end': datetime.datetime(2009, 4, 1, 9, 0),
                'end_recurring_period': datetime.datetime(2009, 10, 5, 0, 0),
                'rule': weekly,
                'room': cal
            })
        self.reservation1.save()
        self.reservation2 = Reservation(
            **{
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 9, 0),
                'end': datetime.datetime(2008, 1, 5, 10, 0),
                'end_recurring_period': datetime.datetime(2009, 5, 5, 0, 0),
                'rule': daily,
                'room': cal
            })
        self.reservation2.save()
Пример #4
0
class TestOccurrencePool(TestCase):
    def setUp(self):
        rule = Rule(frequency="WEEKLY")
        rule.save()
        cal = Room(name="MyCal")
        cal.save()
        data = {
            'title': 'Recent Reservation',
            'start': datetime.datetime(2008, 1, 5, 8, 0),
            'end': datetime.datetime(2008, 1, 5, 9, 0),
            'end_recurring_period': datetime.datetime(2008, 5, 5, 0, 0),
            'rule': rule,
            'room': cal
        }
        self.recurring_reservation = Reservation(**data)
        self.recurring_reservation.save()

    def testPeriodFromPool(self):
        """
            Test that period initiated with occurrence_pool returns the same occurrences as "straigh" period
            in a corner case whereby a period's start date is equal to the occurrence's end date
        """
        start = datetime.datetime(2008, 1, 5, 9, 0)
        end = datetime.datetime(2008, 1, 5, 10, 0)
        parent_period = Period(Reservation.objects.all(), start, end)
        period = Period(parent_period.reservations, start, end,
                        parent_period.get_persisted_occurrences(),
                        parent_period.occurrences)
        self.assertEquals(parent_period.occurrences, period.occurrences)
Пример #5
0
 def make_occurrences(self, rule):
     cal = Room.objects.get(name="MyCal")
     recurring_data = {
             'title': 'Recent Reservation',
             'start': datetime.datetime(2010, 1, 1, 8, 0),
             'end': datetime.datetime(2010, 1, 1, 9, 0),
             'end_recurring_period' : datetime.datetime(2010, 2, 1, 0, 0),
             'rule': rule,
             'room': cal
            }
     recurring_reservation = Reservation(**recurring_data)
     occurrences = recurring_reservation.get_occurrences(start=datetime.datetime(2010, 1, 1, 0, 0),
                                 end=datetime.datetime(2010, 2, 1, 0, 0))
     return [o.start.day for o in occurrences]
Пример #6
0
 def setUp(self):
     rule = Rule(frequency="WEEKLY")
     rule.save()
     cal = Room(name="MyCal")
     cal.save()
     data = {
         'title': 'Recent Reservation',
         'start': datetime.datetime(2008, 1, 5, 8, 0),
         'end': datetime.datetime(2008, 1, 5, 9, 0),
         'end_recurring_period': datetime.datetime(2008, 5, 5, 0, 0),
         'rule': rule,
         'room': cal
     }
     self.recurring_reservation = Reservation(**data)
     self.recurring_reservation.save()
Пример #7
0
 def make_occurrences(self, rule):
     cal = Room.objects.get(name="MyCal")
     recurring_data = {
         'title': 'Recent Reservation',
         'start': datetime.datetime(2010, 1, 1, 8, 0),
         'end': datetime.datetime(2010, 1, 1, 9, 0),
         'end_recurring_period': datetime.datetime(2010, 2, 1, 0, 0),
         'rule': rule,
         'room': cal
     }
     recurring_reservation = Reservation(**recurring_data)
     occurrences = recurring_reservation.get_occurrences(
         start=datetime.datetime(2010, 1, 1, 0, 0),
         end=datetime.datetime(2010, 2, 1, 0, 0))
     return [o.start.day for o in occurrences]
 def setUp(self):
     rule = Rule(frequency="WEEKLY")
     rule.save()
     cal = Room(name="MyCal")
     cal.save()
     data = {
         "title": "Recent Reservation",
         "start": datetime.datetime(2008, 1, 5, 8, 0),
         "end": datetime.datetime(2008, 1, 5, 9, 0),
         "end_recurring_period": datetime.datetime(2008, 5, 5, 0, 0),
         "rule": rule,
         "room": cal,
     }
     recurring_reservation = Reservation(**data)
     recurring_reservation.save()
     self.month = Month(reservations=Reservation.objects.all(), date=datetime.datetime(2008, 2, 7, 9, 0))
Пример #9
0
 def setUp(self):
     rule = Rule(frequency="WEEKLY")
     rule.save()
     cal = Room(name="MyCal")
     cal.save()
     data = {
         'title': 'Recent Reservation',
         'start': datetime.datetime(2008, 1, 5, 8, 0),
         'end': datetime.datetime(2008, 1, 5, 9, 0),
         'end_recurring_period': datetime.datetime(2008, 5, 5, 0, 0),
         'rule': rule,
         'room': cal
     }
     recurring_reservation = Reservation(**data)
     recurring_reservation.save()
     self.period = Period(reservations=Reservation.objects.all(),
                          start=datetime.datetime(2008, 1, 4, 7, 0),
                          end=datetime.datetime(2008, 1, 21, 7, 0))
Пример #10
0
class TestReservationListManager(TestCase):
    def setUp(self):
        weekly = Rule(frequency="WEEKLY")
        weekly.save()
        daily = Rule(frequency="DAILY")
        daily.save()
        cal = Room(name="MyCal")
        cal.save()

        self.reservation1 = Reservation(
            **{
                'title': 'Weekly Reservation',
                'start': datetime.datetime(2009, 4, 1, 8, 0),
                'end': datetime.datetime(2009, 4, 1, 9, 0),
                'end_recurring_period': datetime.datetime(2009, 10, 5, 0, 0),
                'rule': weekly,
                'room': cal
            })
        self.reservation1.save()
        self.reservation2 = Reservation(
            **{
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 9, 0),
                'end': datetime.datetime(2008, 1, 5, 10, 0),
                'end_recurring_period': datetime.datetime(2009, 5, 5, 0, 0),
                'rule': daily,
                'room': cal
            })
        self.reservation2.save()

    def test_occurrences_after(self):
        eml = ReservationListManager([self.reservation1, self.reservation2])
        occurrences = eml.occurrences_after(datetime.datetime(
            2009, 4, 1, 0, 0))
        self.assertEqual(occurrences.next().reservation, self.reservation1)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation1)
Пример #11
0
class TestReservationListManager(TestCase):
    def setUp(self):
        weekly = Rule(frequency = "WEEKLY")
        weekly.save()
        daily = Rule(frequency = "DAILY")
        daily.save()
        cal = Room(name="MyCal")
        cal.save()

        self.reservation1 = Reservation(**{
                'title': 'Weekly Reservation',
                'start': datetime.datetime(2009, 4, 1, 8, 0),
                'end': datetime.datetime(2009, 4, 1, 9, 0),
                'end_recurring_period' : datetime.datetime(2009, 10, 5, 0, 0),
                'rule': weekly,
                'room': cal
               })
        self.reservation1.save()
        self.reservation2 = Reservation(**{
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 9, 0),
                'end': datetime.datetime(2008, 1, 5, 10, 0),
                'end_recurring_period' : datetime.datetime(2009, 5, 5, 0, 0),
                'rule': daily,
                'room': cal
               })
        self.reservation2.save()

    def test_occurrences_after(self):
        eml = ReservationListManager([self.reservation1, self.reservation2])
        occurrences = eml.occurrences_after(datetime.datetime(2009,4,1,0,0))
        self.assertEqual(occurrences.next().reservation, self.reservation1)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation2)
        self.assertEqual(occurrences.next().reservation, self.reservation1)
Пример #12
0
 def test_get_occurrence(self):
     reservation = Reservation(**self.recurring_data)
     reservation.save()
     occurrence = reservation.get_occurrence(datetime.datetime(2008, 1, 5, 8, 0))
     self.assertEqual(occurrence.start, datetime.datetime(2008,1,5,8))
     occurrence.save()
     occurrence = reservation.get_occurrence(datetime.datetime(2008, 1, 5, 8, 0))
     self.assertTrue(occurrence.pk is not None)
Пример #13
0
 def test_reservation_get_occurrences_after(self):
     recurring_reservation=Reservation(**self.recurring_data)
     recurring_reservation.save()
     occurrences = recurring_reservation.get_occurrences(start=datetime.datetime(2008, 1, 5),
         end = datetime.datetime(2008, 1, 6))
     occurrence = occurrences[0]
     occurrence2 = recurring_reservation.occurrences_after(datetime.datetime(2008,1,5)).next()
     self.assertEqual(occurrence, occurrence2)
Пример #14
0
 def setUp(self):
     rule = Rule(frequency = "WEEKLY")
     rule.save()
     cal = Room(name="MyCal")
     cal.save()
     self.recurring_data = {
             'title': 'Recent Reservation',
             'start': datetime.datetime(2008, 1, 5, 8, 0),
             'end': datetime.datetime(2008, 1, 5, 9, 0),
             'end_recurring_period' : datetime.datetime(2008, 5, 5, 0, 0),
             'rule': rule,
             'room': cal
            }
     self.data = {
             'title': 'Recent Reservation',
             'start': datetime.datetime(2008, 1, 5, 8, 0),
             'end': datetime.datetime(2008, 1, 5, 9, 0),
             'end_recurring_period' : datetime.datetime(2008, 5, 5, 0, 0),
             'room': cal
            }
     self.recurring_reservation = Reservation(**self.recurring_data)
     self.recurring_reservation.save()
     self.start = datetime.datetime(2008, 1, 12, 0, 0)
     self.end = datetime.datetime(2008, 1, 27, 0, 0)
Пример #15
0
class TestOccurrence(TestCase):
    def setUp(self):
        rule = Rule(frequency = "WEEKLY")
        rule.save()
        cal = Room(name="MyCal")
        cal.save()
        self.recurring_data = {
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 8, 0),
                'end': datetime.datetime(2008, 1, 5, 9, 0),
                'end_recurring_period' : datetime.datetime(2008, 5, 5, 0, 0),
                'rule': rule,
                'room': cal
               }
        self.data = {
                'title': 'Recent Reservation',
                'start': datetime.datetime(2008, 1, 5, 8, 0),
                'end': datetime.datetime(2008, 1, 5, 9, 0),
                'end_recurring_period' : datetime.datetime(2008, 5, 5, 0, 0),
                'room': cal
               }
        self.recurring_reservation = Reservation(**self.recurring_data)
        self.recurring_reservation.save()
        self.start = datetime.datetime(2008, 1, 12, 0, 0)
        self.end = datetime.datetime(2008, 1, 27, 0, 0)

    def test_presisted_occurrences(self):
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        persisted_occurrence = occurrences[0]
        persisted_occurrence.save()
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertTrue(occurrences[0].pk)
        self.assertFalse(occurrences[1].pk)

    def test_moved_occurrences(self):
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        moved_occurrence = occurrences[1]
        span_pre = (moved_occurrence.start, moved_occurrence.end)
        span_post = [x + datetime.timedelta(hours=2) for x in span_pre]
        # check has_occurrence on both periods
        period_pre = Period([self.recurring_reservation], span_pre[0], span_pre[1])
        period_post = Period([self.recurring_reservation], span_post[0], span_post[1])
        self.assertTrue(period_pre.has_occurrences())
        self.assertFalse(period_post.has_occurrences())
        # move occurrence
        moved_occurrence.move(moved_occurrence.start+datetime.timedelta(hours=2),
                              moved_occurrence.end+datetime.timedelta(hours=2))
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertTrue(occurrences[1].moved)
        # check has_occurrence on both periods (the result should be reversed)
        period_pre = Period([self.recurring_reservation], span_pre[0], span_pre[1])
        period_post = Period([self.recurring_reservation], span_post[0], span_post[1])
        self.assertFalse(period_pre.has_occurrences())
        self.assertTrue(period_post.has_occurrences())
        # trigger bug discovered by gautamadude - modifying recurring reservation
        # breaks link between a persistent occurrence and the occurrence chain
        # leaving an "orphaned" occurrence
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertTrue(len(occurrences) == 3)
        self.recurring_reservation.start = datetime.datetime(2008, 1, 5, 8, 15)
        self.recurring_reservation.save()
        occurrences_later = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertEquals(len(occurrences_later), len(occurrences))

    def test_cancelled_occurrences(self):
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        cancelled_occurrence = occurrences[2]
        cancelled_occurrence.cancel()
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertTrue(occurrences[2].cancelled)
        cancelled_occurrence.uncancel()
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertFalse(occurrences[2].cancelled)
        # trigger bug discovered by gautamadude
        occurrences = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertTrue(len(occurrences) == 3)
        self.recurring_reservation.start = datetime.datetime(2008, 1, 5, 8, 15)
        self.recurring_reservation.save()
        occurrences_later = self.recurring_reservation.get_occurrences(start=self.start,
                                    end=self.end)
        self.assertEquals(len(occurrences_later), len(occurrences))
Пример #16
0
 def test_recurring_reservation_get_occurrences(self):
     recurring_reservation = Reservation(**self.recurring_data)
     occurrences = recurring_reservation.get_occurrences(start=datetime.datetime(2008, 1, 12, 0, 0),
                                 end=datetime.datetime(2008, 1, 20, 0, 0))
     self.assertEquals(["%s to %s" %(o.start, o.end) for o in occurrences],
         ['2008-01-12 08:00:00 to 2008-01-12 09:00:00', '2008-01-19 08:00:00 to 2008-01-19 09:00:00'])