def test_mini_calendar(self): user = User.objects.create_user( 'foo', '*****@*****.**', 'secret' ) event = Event.objects.create( start_date=timezone.now(), end_date=timezone.now(), all_day=True, created_by=user, title="The big event", description="Amazing event", repeat="NEVER", ) event2 = Event.objects.create( start_date=timezone.now() + timezone.timedelta(days=1), end_date=timezone.now() + timezone.timedelta(days=1), all_day=True, created_by=user, title="The other event", description="Incredible event", repeat="NEVER", ) event.save() event2.save() req = self.factory.get(self.url, {'cal_category': 'birthday'}) cal = show_calendar({}, req, mini=True) self.assertIn(str(now.month), cal) self.assertIn(str(now.year), cal) self.assertNotIn(event.title, cal) self.assertNotIn(event2.title, cal)
def test_show_calendar_no_events(self): req = self.factory.get(self.url) cal = show_calendar({}, req) self.assertIn(str(now.month), cal) self.assertIn(str(now.year), cal) self.assertNotIn('calendar-event', cal)