def setUp(self):
     self.home = Page.objects.get(slug='home')
     self.user = User.objects.create_user('i', '*****@*****.**', 's3(r3t')
     self.calendar = CalendarPage(owner=self.user,
                                  slug="events",
                                  title="Events")
     self.home.add_child(instance=self.calendar)
     self.calendar.save_revision().publish()
     self.event = RecurringEventPage(slug="test-meeting",
                                     title="Test Meeting",
                                     repeat=Recurrence(
                                         dtstart=dt.date(1989, 1, 1),
                                         freq=WEEKLY,
                                         byweekday=[MO, WE, FR]),
                                     time_from=dt.time(13),
                                     time_to=dt.time(15, 30),
                                     holidays=self.calendar.holidays)
     self.calendar.add_child(instance=self.event)
     self.closedHols = ClosedForHolidaysPage(
         owner=self.user,
         overrides=self.event,
         all_holidays=True,
         holidays=self.calendar.holidays)
     self.event.add_child(instance=self.closedHols)
     self.closedHols.save_revision().publish()
示例#2
0
 def testFromEventPageClosedHolidays(self):
     chess = GroupPage(slug="chess-club", title="Chess Club")
     self.home.add_child(instance=chess)
     page = RecurringEventPage(owner=self.user,
                               slug="chess",
                               title="Chess",
                               repeat=Recurrence(dtstart=dt.date(
                                   2020, 1, 1),
                                                 freq=WEEKLY,
                                                 byweekday=[MO, WE, FR]),
                               time_from=dt.time(12),
                               time_to=dt.time(13),
                               holidays=self.calendar.holidays)
     chess.add_child(instance=page)
     page.save_revision().publish()
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        slug="closed-for-holidays",
                                        title="Closed for holidays",
                                        all_holidays=True,
                                        overrides=page,
                                        holidays=self.calendar.holidays)
     page.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     vcal = VCalendar.fromPage(page, self._getRequest("/events/chess/"))
     export = vcal.to_ical()
     props = [
         b"SUMMARY:Chess",
         b"DTSTART;TZID=Asia/Tokyo:20200101T12000",
         b"DTEND;TZID=Asia/Tokyo:20200101T13000",
         b"DTSTAMP:20200121T040000Z",
         b"UID:",
         b"SEQUENCE:1",
         b"RRULE:FREQ=WEEKLY;BYDAY=MO,WE,FR;WKST=SU",
         b"EXDATE;TZID=Asia/Tokyo:20200101T120000,20200120T120000,20200127T120000,",
         b",20251024T120000,20251027T120000,20251103T120000,20251114T120000,",
         b",20281117T120000,20281127T120000,20281204T120000,20281225T",
         b",20341030T120000,20341117T120000,20341127T120000,20341204T120000,",
         b",20361226T120000,20370102T120000,20370119T120000,20370126T120000,",
         b",20381101T120000,20381112T120000,20381129T120000,20381227T120000",
         b"CREATED:20200121T040000Z",
         b"DESCRIPTION:",
         b"LAST-MODIFIED:20200121T040000Z",
         b"LOCATION:",
         b"URL:http://joy.test/chess-club/chess/",
     ]
     for prop in props:
         with self.subTest(prop=prop):
             self.assertIn(prop, export)
 def testGetUpcomingEvents(self):
     event = RecurringEventPage(slug="RST",
                                title="Ruritania secret taxidermy",
                                repeat=Recurrence(dtstart=dt.date(
                                    1980, 1, 1),
                                                  freq=MONTHLY,
                                                  byweekday=[MO(1)]),
                                time_from=dt.time(20))
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(
         owner=self.user,
         overrides=event,
         all_holidays=False,
         cancellation_title="Closed for the holiday")
     closedHols.closed_for = [
         ClosedFor(name="Wellington Anniversary Day"),
         ClosedFor(name="Auckland Anniversary Day"),
         ClosedFor(name="Nelson Anniversary Day"),
         ClosedFor(name="Taranaki Anniversary Day"),
         ClosedFor(name="Otago Anniversary Day"),
         ClosedFor(name="Southland Anniversary Day"),
         ClosedFor(name="South Canterbury Anniversary Day"),
         ClosedFor(name="Hawke's Bay Anniversary Day"),
         ClosedFor(name="Marlborough Anniversary Day"),
         ClosedFor(name="Canterbury Anniversary Day"),
         ClosedFor(name="Chatham Islands Anniversary Day"),
         ClosedFor(name="Westland Anniversary Day")
     ]
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     events = ClosedForHolidaysPage.events.hols(self.calendar.holidays)   \
                                   .exclude(cancellation_title="")        \
                                   .upcoming().this().descendant_of(event)
     self.assertEqual(len(events), 1)
     title, page, url = events[0]
     self.assertEqual(title, "Closed for the holiday")
     self.assertEqual(page._future_datetime_from,
                      datetimetz(1990, 12, 3, 20, 0))
     self.assertEqual(url, "/events/RST/closed-for-holidays/")
 def testWhenEver(self):
     event = RecurringEventPage(slug="XYZ",
                                title="Xylophone yacht zombies",
                                repeat=Recurrence(dtstart=dt.date(
                                    1989, 1, 1),
                                                  freq=WEEKLY,
                                                  byweekday=[FR]),
                                time_from=dt.time(19),
                                holidays=self.calendar.holidays)
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        overrides=event,
                                        all_holidays=False,
                                        cancellation_title="XYZ Cancelled",
                                        holidays=self.calendar.holidays)
     closedHols.closed_for = [
         ClosedFor(name="Good Friday"),
         ClosedFor(name="Easter Monday")
     ]
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     self.assertEqual(closedHols.when,
                      "Closed on Good Friday and Easter Monday")
     self.assertIs(event._occursOn(dt.date(1989, 3, 24)), False)
 def testWombatGetEventsByDay(self):
     event = RecurringEventPage(slug="UVW",
                                title="Underwater viking wombats",
                                repeat=Recurrence(dtstart=dt.date(
                                    1989, 1, 1),
                                                  freq=MONTHLY,
                                                  byweekday=[MO(1)]),
                                time_from=dt.time(19))
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        overrides=event,
                                        all_holidays=False,
                                        cancellation_title="UVW Cancelled")
     closedHols.closed_for = [
         ClosedFor(name="New Year's Day"),
         ClosedFor(name="Day after New Year's Day"),
         ClosedFor(name="Good Friday"),
         ClosedFor(name="Easter Monday"),
         ClosedFor(name="Christmas Day"),
         ClosedFor(name="Boxing Day")
     ]
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     events = RecurringEventPage.events.hols(self.calendar.holidays)      \
                                .byDay(dt.date(1989,1,1), dt.date(1989,1,31))
     self.assertEqual(len(events), 31)
     evod = events[1]
     self.assertEqual(evod.date, dt.date(1989, 1, 2))
     self.assertEqual(evod.holiday, "Day after New Year's Day")
     self.assertEqual(len(evod.days_events), 1)
     self.assertEqual(len(evod.continuing_events), 0)
     title, page, url = evod.all_events[0]
     self.assertEqual(title, "UVW Cancelled")
     self.assertEqual(page.title, "Closed for holidays")
     self.assertEqual(page.at, "7pm")
     self.assertEqual(url, "/events/UVW/closed-for-holidays/")
 def testGetMyDates(self):
     event = RecurringEventPage(slug="UVW",
                                title="Underwater viking wombats",
                                repeat=Recurrence(dtstart=dt.date(
                                    1989, 1, 1),
                                                  freq=MONTHLY,
                                                  byweekday=[MO(1)]),
                                time_from=dt.time(19))
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        overrides=event,
                                        all_holidays=False)
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     dates = list(islice(closedHols._getMyDates(), 10))
     self.assertEqual(dates, [])
     closedHols.holidays = self.calendar.holidays
     dates = list(islice(closedHols._getMyDates(), 10))
     self.assertEqual(dates, [])
     closedHols.closed_for = [ClosedFor(name="Good Friday")]
     # Good Friday is never going to fall on a Monday
     dates = list(islice(closedHols._getMyDates(), 10))
     self.assertEqual(dates, [])
 def testClosedOn(self):
     event = RecurringEventPage(slug="XYZ",
                                title="Xylophone yacht zombies",
                                repeat=Recurrence(dtstart=dt.date(
                                    1989, 1, 1),
                                                  freq=WEEKLY,
                                                  byweekday=[FR]),
                                time_from=dt.time(19),
                                holidays=self.calendar.holidays)
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        overrides=event,
                                        all_holidays=False,
                                        cancellation_title="XYZ Cancelled")
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     self.assertEqual(closedHols._closedOn(dt.date(1989, 3, 24)), False)
     closedHols.holidays = self.calendar.holidays
     self.assertEqual(closedHols._closedOn(dt.date(1989, 3, 24)), False)
 def testFarFutureDt(self):
     event = RecurringEventPage(slug="UVW",
                                title="Underwater viking wombats",
                                repeat=Recurrence(dtstart=dt.date(
                                    1989, 1, 1),
                                                  freq=MONTHLY,
                                                  bymonthday=23),
                                time_from=dt.time(19))
     self.calendar.add_child(instance=event)
     closedHols = ClosedForHolidaysPage(owner=self.user,
                                        overrides=event,
                                        all_holidays=False)
     event.add_child(instance=closedHols)
     closedHols.save_revision().publish()
     dates = list(islice(closedHols._getMyDates(), 10))
     self.assertEqual(dates, [])
     closedHols.holidays = self.calendar.holidays
     dates = list(islice(closedHols._getMyDates(), 10))
     self.assertEqual(dates, [])
     closedHols.closed_for = [ClosedFor(name="Good Friday")]
     # Good Friday falls on 2038-04-23 but that is too far to calculate
     self.assertEqual(closedHols._future_datetime_from,
                      datetimetz(9999, 12, 29, 19, 0))
 def testDefaultHolidayChoices(self):
     page = ClosedForHolidaysPage(owner=self.user)
     form = self.Form({'overrides': self.event},
                      instance=page,
                      parent_page=self.event)
     self.assertCountEqual(
         form.fields['closed_for'].choices,
         [(name, name) for name in
          ("New Year's Day", "Day after New Year's Day",
           "New Year's Day (Observed)",
           "Day after New Year's Day (Observed)",
           'Wellington Anniversary Day', 'Auckland Anniversary Day',
           'Nelson Anniversary Day', 'Waitangi Day',
           'Waitangi Day (Observed)', 'Taranaki Anniversary Day',
           'Otago Anniversary Day', 'Good Friday', 'Easter Monday',
           'Southland Anniversary Day', 'Anzac Day', 'Anzac Day (Observed)',
           "Queen's Birthday", 'South Canterbury Anniversary Day',
           "Hawke's Bay Anniversary Day", 'Labour Day',
           'Marlborough Anniversary Day', 'Canterbury Anniversary Day',
           'Chatham Islands Anniversary Day', 'Westland Anniversary Day',
           'Christmas Day', 'Boxing Day', 'Christmas Day (Observed)',
           'Boxing Day (Observed)')])
 def testValid(self):
     page = ClosedForHolidaysPage(owner=self.user,
                                  holidays=self.event.holidays)
     form = self.Form(
         {
             'overrides': self.event,
             'all_holidays': False,
             'closed_for': ["St Swithin's Day", "Sleepy Head Day"],
             'cancellation_title': "Holiday",
             'cancellation_details': rich_text("No meeting today!"),
         },
         instance=page,
         parent_page=self.event)
     self.assertCountEqual(form.fields['closed_for'].choices,
                           [("Groundhog Day", "Groundhog Day"),
                            ("St Swithin's Day", "St Swithin's Day"),
                            ("Sleepy Head Day", "Sleepy Head Day")])
     self.assertEqual(form.initial['closed_for'], [])
     self.assertTrue(form.is_valid())  # is_valid() calls full_clean()
     self.assertDictEqual(form.errors, {})
     saved = form.save(commit=False)
     self.assertCountEqual(saved.closed,
                           ["St Swithin's Day", "Sleepy Head Day"])
 def testCanCreateOnlyOne(self):
     self.assertFalse(ClosedForHolidaysPage.can_create_at(self.event))
class TestPageForm(TestCase):
    Form = ClosedForHolidaysPage.get_edit_handler().get_form_class()

    def setUp(self):
        holidays = Holidays(holidaySetting=None)
        holidays.register(WeatherDays())
        self.home = Page.objects.get(slug='home')
        self.user = User.objects.create_user('i', '*****@*****.**', 's3(r3t')
        self.calendar = CalendarPage(owner=self.user,
                                     slug="events",
                                     title="Events")
        self.calendar.holidays = holidays
        self.home.add_child(instance=self.calendar)
        self.calendar.save_revision().publish()
        self.event = RecurringEventPage(slug="committee-meeting",
                                        title="Committee Meeting",
                                        repeat=Recurrence(
                                            dtstart=dt.date(2017, 1, 1),
                                            freq=MONTHLY,
                                            byweekday=[MO(1), MO(3)]),
                                        time_from=dt.time(13),
                                        time_to=dt.time(15, 30),
                                        holidays=holidays)
        self.calendar.add_child(instance=self.event)

    def testValid(self):
        page = ClosedForHolidaysPage(owner=self.user,
                                     holidays=self.event.holidays)
        form = self.Form(
            {
                'overrides': self.event,
                'all_holidays': False,
                'closed_for': ["St Swithin's Day", "Sleepy Head Day"],
                'cancellation_title': "Holiday",
                'cancellation_details': rich_text("No meeting today!"),
            },
            instance=page,
            parent_page=self.event)
        self.assertCountEqual(form.fields['closed_for'].choices,
                              [("Groundhog Day", "Groundhog Day"),
                               ("St Swithin's Day", "St Swithin's Day"),
                               ("Sleepy Head Day", "Sleepy Head Day")])
        self.assertEqual(form.initial['closed_for'], [])
        self.assertTrue(form.is_valid())  # is_valid() calls full_clean()
        self.assertDictEqual(form.errors, {})
        saved = form.save(commit=False)
        self.assertCountEqual(saved.closed,
                              ["St Swithin's Day", "Sleepy Head Day"])

    def testDefaultHolidayChoices(self):
        page = ClosedForHolidaysPage(owner=self.user)
        form = self.Form({'overrides': self.event},
                         instance=page,
                         parent_page=self.event)
        self.assertCountEqual(
            form.fields['closed_for'].choices,
            [(name, name) for name in
             ("New Year's Day", "Day after New Year's Day",
              "New Year's Day (Observed)",
              "Day after New Year's Day (Observed)",
              'Wellington Anniversary Day', 'Auckland Anniversary Day',
              'Nelson Anniversary Day', 'Waitangi Day',
              'Waitangi Day (Observed)', 'Taranaki Anniversary Day',
              'Otago Anniversary Day', 'Good Friday', 'Easter Monday',
              'Southland Anniversary Day', 'Anzac Day', 'Anzac Day (Observed)',
              "Queen's Birthday", 'South Canterbury Anniversary Day',
              "Hawke's Bay Anniversary Day", 'Labour Day',
              'Marlborough Anniversary Day', 'Canterbury Anniversary Day',
              'Chatham Islands Anniversary Day', 'Westland Anniversary Day',
              'Christmas Day', 'Boxing Day', 'Christmas Day (Observed)',
              'Boxing Day (Observed)')])
class Test(TestCase):
    def setUp(self):
        self.home = Page.objects.get(slug='home')
        self.user = User.objects.create_user('i', '*****@*****.**', 's3(r3t')
        self.calendar = CalendarPage(owner=self.user,
                                     slug="events",
                                     title="Events")
        self.home.add_child(instance=self.calendar)
        self.calendar.save_revision().publish()
        self.event = RecurringEventPage(slug="test-meeting",
                                        title="Test Meeting",
                                        repeat=Recurrence(
                                            dtstart=dt.date(1989, 1, 1),
                                            freq=WEEKLY,
                                            byweekday=[MO, WE, FR]),
                                        time_from=dt.time(13),
                                        time_to=dt.time(15, 30),
                                        holidays=self.calendar.holidays)
        self.calendar.add_child(instance=self.event)
        self.closedHols = ClosedForHolidaysPage(
            owner=self.user,
            overrides=self.event,
            all_holidays=True,
            holidays=self.calendar.holidays)
        self.event.add_child(instance=self.closedHols)
        self.closedHols.save_revision().publish()

    def testCanCreateOnlyOne(self):
        self.assertFalse(ClosedForHolidaysPage.can_create_at(self.event))

    def testInit(self):
        self.assertEqual(self.closedHols.all_holidays, True)
        self.assertEqual(self.closedHols.title, "Closed for holidays")
        self.assertEqual(self.closedHols.local_title, "Closed for holidays")
        self.assertEqual(self.closedHols.slug, "closed-for-holidays")

    def testGetEventsByDay(self):
        events = RecurringEventPage.events.hols(self.calendar.holidays)      \
                                   .byDay(dt.date(1989,1,1), dt.date(1989,1,31))
        self.assertEqual(len(events), 31)
        evod = events[0]
        self.assertEqual(evod.date, dt.date(1989, 1, 1))
        self.assertEqual(evod.holiday, "New Year's Day")
        self.assertEqual(len(evod.days_events), 0)
        self.assertEqual(len(evod.continuing_events), 0)

    @freeze_timetz("1990-11-11 16:29:00")
    def testEventFutureExceptions(self):
        request = RequestFactory().get("/test")
        request.user = self.user
        request.session = {}
        exceptions = self.event._futureExceptions(request)
        self.assertEqual(len(exceptions), 1)
        c4h = exceptions[0]
        self.assertEqual(c4h.title, "Closed for holidays")
        self.assertEqual(c4h._future_datetime_from,
                         datetimetz(1990, 11, 16, 13, 0))

    def testEventOccursOn(self):
        self.assertIs(self.event._occursOn(dt.date(1989, 1, 1)), False)
        self.assertIs(self.event._occursOn(dt.date(1989, 1, 2)), False)
        self.assertIs(self.event._occursOn(dt.date(1989, 1, 6)), True)
        self.assertIs(self.event._occursOn(dt.date(1989, 1, 16)), False)

    @freeze_timetz("2022-12-24")
    def testEventNextDate(self):
        self.assertEqual(self.event.next_date, dt.date(2022, 12, 28))

    @freeze_timetz("2023-01-31")
    def testEventPastDatetime(self):
        self.assertEqual(self.event._past_datetime_from,
                         datetimetz(2023, 1, 27, 13, 0))

    @freeze_timetz("1990-10-11 16:29:00")
    def testGetUpcomingEvents(self):
        event = RecurringEventPage(slug="RST",
                                   title="Ruritania secret taxidermy",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1980, 1, 1),
                                                     freq=MONTHLY,
                                                     byweekday=[MO(1)]),
                                   time_from=dt.time(20))
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(
            owner=self.user,
            overrides=event,
            all_holidays=False,
            cancellation_title="Closed for the holiday")
        closedHols.closed_for = [
            ClosedFor(name="Wellington Anniversary Day"),
            ClosedFor(name="Auckland Anniversary Day"),
            ClosedFor(name="Nelson Anniversary Day"),
            ClosedFor(name="Taranaki Anniversary Day"),
            ClosedFor(name="Otago Anniversary Day"),
            ClosedFor(name="Southland Anniversary Day"),
            ClosedFor(name="South Canterbury Anniversary Day"),
            ClosedFor(name="Hawke's Bay Anniversary Day"),
            ClosedFor(name="Marlborough Anniversary Day"),
            ClosedFor(name="Canterbury Anniversary Day"),
            ClosedFor(name="Chatham Islands Anniversary Day"),
            ClosedFor(name="Westland Anniversary Day")
        ]
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        events = ClosedForHolidaysPage.events.hols(self.calendar.holidays)   \
                                      .exclude(cancellation_title="")        \
                                      .upcoming().this().descendant_of(event)
        self.assertEqual(len(events), 1)
        title, page, url = events[0]
        self.assertEqual(title, "Closed for the holiday")
        self.assertEqual(page._future_datetime_from,
                         datetimetz(1990, 12, 3, 20, 0))
        self.assertEqual(url, "/events/RST/closed-for-holidays/")

    @freeze_timetz("1990-10-11 16:29:00")
    def testGetPastEvents(self):
        events = ClosedForHolidaysPage.events.hols(self.calendar.holidays)   \
                                      .past().this()
        self.assertEqual(len(events), 1)
        title, page, url = events[0]
        self.assertEqual(title, "")
        self.assertEqual(page._past_datetime_from,
                         datetimetz(1990, 9, 24, 13, 0))
        self.assertEqual(url, "/events/test-meeting/closed-for-holidays/")

    @freeze_timetz("1990-10-11 16:29:00")
    def testWombatGetEventsByDay(self):
        event = RecurringEventPage(slug="UVW",
                                   title="Underwater viking wombats",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1989, 1, 1),
                                                     freq=MONTHLY,
                                                     byweekday=[MO(1)]),
                                   time_from=dt.time(19))
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(owner=self.user,
                                           overrides=event,
                                           all_holidays=False,
                                           cancellation_title="UVW Cancelled")
        closedHols.closed_for = [
            ClosedFor(name="New Year's Day"),
            ClosedFor(name="Day after New Year's Day"),
            ClosedFor(name="Good Friday"),
            ClosedFor(name="Easter Monday"),
            ClosedFor(name="Christmas Day"),
            ClosedFor(name="Boxing Day")
        ]
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        events = RecurringEventPage.events.hols(self.calendar.holidays)      \
                                   .byDay(dt.date(1989,1,1), dt.date(1989,1,31))
        self.assertEqual(len(events), 31)
        evod = events[1]
        self.assertEqual(evod.date, dt.date(1989, 1, 2))
        self.assertEqual(evod.holiday, "Day after New Year's Day")
        self.assertEqual(len(evod.days_events), 1)
        self.assertEqual(len(evod.continuing_events), 0)
        title, page, url = evod.all_events[0]
        self.assertEqual(title, "UVW Cancelled")
        self.assertEqual(page.title, "Closed for holidays")
        self.assertEqual(page.at, "7pm")
        self.assertEqual(url, "/events/UVW/closed-for-holidays/")

    def testClosedForDates(self):
        dates = list(islice(self.closedHols._closed_for_dates, 10))
        self.assertEqual(dates, [
            dt.date(1989, 1, 2),
            dt.date(1989, 1, 16),
            dt.date(1989, 1, 23),
            dt.date(1989, 1, 30),
            dt.date(1989, 2, 6),
            dt.date(1989, 3, 13),
            dt.date(1989, 3, 20),
            dt.date(1989, 3, 24),
            dt.date(1989, 3, 27),
            dt.date(1989, 6, 5)
        ])

    def testGetMyDates(self):
        event = RecurringEventPage(slug="UVW",
                                   title="Underwater viking wombats",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1989, 1, 1),
                                                     freq=MONTHLY,
                                                     byweekday=[MO(1)]),
                                   time_from=dt.time(19))
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(owner=self.user,
                                           overrides=event,
                                           all_holidays=False)
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        dates = list(islice(closedHols._getMyDates(), 10))
        self.assertEqual(dates, [])
        closedHols.holidays = self.calendar.holidays
        dates = list(islice(closedHols._getMyDates(), 10))
        self.assertEqual(dates, [])
        closedHols.closed_for = [ClosedFor(name="Good Friday")]
        # Good Friday is never going to fall on a Monday
        dates = list(islice(closedHols._getMyDates(), 10))
        self.assertEqual(dates, [])

    def testClosedOn(self):
        event = RecurringEventPage(slug="XYZ",
                                   title="Xylophone yacht zombies",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1989, 1, 1),
                                                     freq=WEEKLY,
                                                     byweekday=[FR]),
                                   time_from=dt.time(19),
                                   holidays=self.calendar.holidays)
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(owner=self.user,
                                           overrides=event,
                                           all_holidays=False,
                                           cancellation_title="XYZ Cancelled")
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        self.assertEqual(closedHols._closedOn(dt.date(1989, 3, 24)), False)
        closedHols.holidays = self.calendar.holidays
        self.assertEqual(closedHols._closedOn(dt.date(1989, 3, 24)), False)

    def testStatus(self):
        self.assertEqual(self.closedHols.status, "cancelled")
        self.assertEqual(self.closedHols.status_text, "Closed for holidays.")

    def testWhen(self):
        self.assertEqual(self.closedHols.when, "Closed on all holidays")

    def testWhenEver(self):
        event = RecurringEventPage(slug="XYZ",
                                   title="Xylophone yacht zombies",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1989, 1, 1),
                                                     freq=WEEKLY,
                                                     byweekday=[FR]),
                                   time_from=dt.time(19),
                                   holidays=self.calendar.holidays)
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(owner=self.user,
                                           overrides=event,
                                           all_holidays=False,
                                           cancellation_title="XYZ Cancelled",
                                           holidays=self.calendar.holidays)
        closedHols.closed_for = [
            ClosedFor(name="Good Friday"),
            ClosedFor(name="Easter Monday")
        ]
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        self.assertEqual(closedHols.when,
                         "Closed on Good Friday and Easter Monday")
        self.assertIs(event._occursOn(dt.date(1989, 3, 24)), False)

    # def testActualHolidaysClosed(self):
    #     event = RecurringEventPage(slug      = "Z1",
    #                                title     = "ZetaOne",
    #                                repeat    = Recurrence(dtstart=dt.date(2020,1,1),
    #                                                       freq=DAILY),
    #                                time_from = dt.time(19),
    #                                holidays = self.calendar.holidays)
    #     self.calendar.add_child(instance=event)
    #     closedHols = ClosedForHolidaysPage(owner = self.user,
    #                                        overrides = event,
    #                                        all_holidays = False,
    #                                        cancellation_title = "Z1 Cancelled",
    #                                        holidays = self.calendar.holidays)
    #     closedHols.closed_for = [ ClosedFor(name="New Year's Day"),
    #                               ClosedFor(name="New Year's Day (Observed)"),
    #                               ClosedFor(name="Labour Day"),
    #                               ClosedFor(name="Christmas Day"),
    #                               ClosedFor(name="Christmas Day (Observed)") ]
    #     event.add_child(instance=closedHols)
    #     closedHols.save_revision().publish()
    #     self.assertCountEqual(closedHols._actual_closed_for_names,
    #                           ["New Year's Day",
    #                            "New Year's Day (Observed)",
    #                            "Labour Day",
    #                            "Christmas Day",
    #                            "Christmas Day (Observed)"])

    def testAt(self):
        self.assertEqual(self.closedHols.at.strip(), "1pm")

    @freeze_timetz("1989-02-15")
    def testCurrentDt(self):
        # Taranaki Anniversary Day
        self.assertEqual(self.closedHols._current_datetime_from,
                         datetimetz(1989, 3, 13, 13, 0))

    @freeze_timetz("1989-02-15")
    def testFutureDt(self):
        # Taranaki Anniversary Day
        self.assertEqual(self.closedHols._future_datetime_from,
                         datetimetz(1989, 3, 13, 13, 0))

    @freeze_timetz("1989-02-15")
    def testPastDt(self):
        # Waitangi Day
        self.assertEqual(self.closedHols._past_datetime_from,
                         datetimetz(1989, 2, 6, 13, 0))

    @freeze_timetz("1989-02-15")
    def testFutureDtNoHolidays(self):
        self.closedHols.holidays = None
        self.assertEqual(self.closedHols._future_datetime_from,
                         ClosedForHolidaysPage.MAX_DATETIME)

    @freeze_timetz("1989-02-15")
    def testPastDtNoHolidays(self):
        self.closedHols.holidays = None
        self.assertEqual(self.closedHols._past_datetime_from,
                         ClosedForHolidaysPage.MIN_DATETIME)

    @freeze_timetz("1989-02-15")
    def testFutureDtCancelled(self):
        # a cancellation page trumps the holiday
        cancellation = CancellationPage(owner=self.user,
                                        overrides=self.event,
                                        except_date=dt.date(1989, 3, 13))
        self.event.add_child(instance=cancellation)
        self.assertEqual(self.closedHols._future_datetime_from,
                         datetimetz(1989, 3, 20, 13, 0))

    @freeze_timetz("1989-02-15")
    def testPastDtCancelled(self):
        # a cancellation page trumps the holiday
        cancellation = CancellationPage(owner=self.user,
                                        overrides=self.event,
                                        except_date=dt.date(1989, 2, 6))
        self.event.add_child(instance=cancellation)
        self.assertEqual(self.closedHols._past_datetime_from,
                         datetimetz(1989, 1, 30, 13, 0))

    @freeze_timetz("1989-02-15")
    def testPastDtShut(self):
        # a shutdown page trumps the holiday
        shutdown = ExtCancellationPage(owner=self.user,
                                       overrides=self.event,
                                       cancelled_from_date=dt.date(
                                           1989, 1, 29),
                                       cancelled_to_date=dt.date(1989, 2, 10))
        self.event.add_child(instance=shutdown)
        self.assertEqual(self.closedHols._past_datetime_from,
                         datetimetz(1989, 1, 23, 13, 0))

    @freeze_timetz("1989-02-15")
    def testFutureDtShut(self):
        # a shutdown page trumps the holiday
        shutdown = ExtCancellationPage(owner=self.user,
                                       overrides=self.event,
                                       cancelled_from_date=dt.date(
                                           1989, 3, 12),
                                       cancelled_to_date=dt.date(1989, 3, 20))
        self.event.add_child(instance=shutdown)
        self.assertEqual(self.closedHols._future_datetime_from,
                         datetimetz(1989, 3, 24, 13, 0))

    @freeze_timetz("1989-02-15")
    def testFutureDtShutUntilFurtherNotice(self):
        # a shutdown page trumps the holiday
        shutdown = ExtCancellationPage(owner=self.user,
                                       overrides=self.event,
                                       cancelled_from_date=dt.date(
                                           1989, 3, 12))
        self.event.add_child(instance=shutdown)
        self.assertIsNone(self.closedHols._future_datetime_from)

    @freeze_timetz("1989-02-15")
    def testFarFutureDt(self):
        event = RecurringEventPage(slug="UVW",
                                   title="Underwater viking wombats",
                                   repeat=Recurrence(dtstart=dt.date(
                                       1989, 1, 1),
                                                     freq=MONTHLY,
                                                     bymonthday=23),
                                   time_from=dt.time(19))
        self.calendar.add_child(instance=event)
        closedHols = ClosedForHolidaysPage(owner=self.user,
                                           overrides=event,
                                           all_holidays=False)
        event.add_child(instance=closedHols)
        closedHols.save_revision().publish()
        dates = list(islice(closedHols._getMyDates(), 10))
        self.assertEqual(dates, [])
        closedHols.holidays = self.calendar.holidays
        dates = list(islice(closedHols._getMyDates(), 10))
        self.assertEqual(dates, [])
        closedHols.closed_for = [ClosedFor(name="Good Friday")]
        # Good Friday falls on 2038-04-23 but that is too far to calculate
        self.assertEqual(closedHols._future_datetime_from,
                         datetimetz(9999, 12, 29, 19, 0))

    def testGroup(self):
        self.assertIsNone(self.closedHols.group)

    def testOverridesRepeat(self):
        self.assertEqual(self.closedHols.overrides_repeat, self.event.repeat)

    def testGetContext(self):
        request = RequestFactory().get("/test")
        context = self.closedHols.get_context(request)
        self.assertIn('overrides', context)

    def testClosedForStr(self):
        xmas = ClosedFor(name="☧mas")
        self.assertEqual(str(xmas), "☧mas")
示例#14
0
 def testClosedForHolidaysVerboseName(self):
     self.assertEqual(ClosedForHolidaysPage.get_verbose_name(),
                      "Closed for holidays")