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)
