Beispiel #1
0
 def _data(self):
     """Return birthdays sorted by day an fullname."""
     if HAS_PERSON:
         birthdays = self.get_birthdays_from_persons()
     else:
         birthdays = self.get_birthdays_from_users()
     return sort_birthdays(birthdays)
Beispiel #2
0
 def test_sort_birthdays(self):
     birthdays = [
         dict(birthday='19/07', fullname='FFF'),
         dict(birthday='19/07', fullname='EEE'),
         dict(birthday='01/01', fullname='DDD'),
         dict(birthday='28/01', fullname='CCC'),
         dict(birthday='01/01', fullname='BBB'),
         dict(birthday='28/01', fullname='AAA'),
     ]
     sorted_birthdays = sort_birthdays(birthdays)
     days = [day for day in sorted_birthdays]
     # we have 3 different dates: 01/01, 28/01 and 19/07
     self.assertEqual(len(days), 3)
     # we test each date do see if it is sorted
     self.assertEqual(
         sorted_birthdays[days[0]],
         [
             {'fullname': 'BBB', 'birthday': '01/01'},
             {'fullname': 'DDD', 'birthday': '01/01'}
         ]
     )
     self.assertEqual(
         sorted_birthdays[days[1]],
         [
             {'fullname': 'AAA', 'birthday': '28/01'},
             {'fullname': 'CCC', 'birthday': '28/01'},
         ]
     )
     self.assertEqual(
         sorted_birthdays[days[2]],
         [
             {'fullname': 'EEE', 'birthday': '19/07'},
             {'fullname': 'FFF', 'birthday': '19/07'},
         ]
     )
 def test_long_period(self):
     render = self.renderer(assignment=birthdayportlet.Assignment(365))
     self.assertFalse(render.available)
     birthday1 = datetime.date(datetime.now())
     birthday2 = datetime.date(datetime.now() + timedelta(days=364))
     self.portal.invokeFactory('Person', 'name1', birthday=birthday1)
     self.portal.invokeFactory('Person', 'name2', birthday=birthday2)
     self.pw.doActionFor(self.portal['name1'], 'publish')
     self.pw.doActionFor(self.portal['name2'], 'publish')
     birthdays = render.get_birthdays_from_persons()
     birthdays = sort_birthdays(birthdays)
     self.assertEqual(2, len(birthdays))