def testGetEventDifferentCategories(self): """Test retrieving from different types""" events = [Event('Single event 1', 'type1', datetime.date(2012, 2, 20)), Event('Single event 2', 'type2', datetime.date(2012, 2, 20))] filter = SimpleEventFilter() self.assertEquals(1, len(filter.getEvents(events, 'type1', datetime.date(2012, 2, 20)))) self.assertEquals(1, len(filter.getEvents(events, 'type2', datetime.date(2012, 2, 20)))) self.assertEquals(0, len(filter.getEvents(events, 'type3', datetime.date(2012, 2, 20))))
def testGetEvent(self): """Test returning one day events""" type = 'type1' events = [Event('Single event 1', type, datetime.date(2012, 2, 20)), Event('Single event 2', type, datetime.date(2012, 2, 20)), Event('Single event 3', type, datetime.date(2012, 2, 21))] filter = SimpleEventFilter() self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 2, 19)))) self.assertEquals(2, len(filter.getEvents(events, type, datetime.date(2012, 2, 20)))) events3 = filter.getEvents(events, type, datetime.date(2012, 2, 21)) self.assertEquals(1, len(events3)) self.assertEquals('Single event 3', events3[0].content) self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 2, 22))))
def testGetEvent2(self): """Test retrieving multi day events""" type = 'type1' events = [Event('Single event 1', type, datetime.date(2012, 2, 10), datetime.date(2012, 2, 20)), Event('Single event 2', type, datetime.date(2012, 2, 15), datetime.date(2012, 2, 25)), Event('Single event 3', type, datetime.date(2012, 2, 20), datetime.date(2012, 2, 26))] filter = SimpleEventFilter() #print filter.getEvents(events, type, datetime.date(2012, 2, 10))['type1'] self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 2, 9)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2012, 2, 10)))) self.assertEquals(2, len(filter.getEvents(events, type, datetime.date(2012, 2, 15)))) self.assertEquals(3, len(filter.getEvents(events, type, datetime.date(2012, 2, 20)))) self.assertEquals(2, len(filter.getEvents(events, type, datetime.date(2012, 2, 25)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2012, 2, 26)))) self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 2, 27))))
def testGetEvent4(self): """Test that events come in a right order""" type = 'type1' events = [Event('Single event 3', type, datetime.date(2012, 1, 1)), Event('Single event 2', type, datetime.date(2011, 1, 1)), Event('Single event 1', type, datetime.date(2010, 1, 1))] filter = SimpleEventFilter() events = filter.getEvents(events, type, datetime.date(2013, 1, 1)) self.assertEquals(3, len(events)) self.assertEquals('Single event 1', events[0].content) self.assertEquals('Single event 2', events[1].content) self.assertEquals('Single event 3', events[2].content)
def testGetEvent3(self): """Test retrieving a multi-day event when it has different start and end dates""" type = 'type1' events = [Event('Single event 1', type, datetime.date(2011, 12, 1), datetime.date(2012, 1, 30))] filter = SimpleEventFilter() self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2011, 11, 30)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2011, 12, 1)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2011, 12, 31)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2012, 1, 1)))) self.assertEquals(1, len(filter.getEvents(events, type, datetime.date(2012, 1, 30)))) self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 2, 1)))) self.assertEquals(0, len(filter.getEvents(events, type, datetime.date(2012, 12, 1))))