def test_queryset(self): "It should return 10 recent events by default." CinemaEventFactory.create_batch(6, date=make_date('2017-02-10')) GigEventFactory.create_batch(6, date=make_date('2017-02-12')) qs = recent_events() self.assertEqual(len(qs), 10) self.assertEqual(qs[5].kind, 'gig') self.assertEqual(qs[6].kind, 'cinema')
def setUp(self): super().setUp() CinemaEventFactory.create_batch(1, date=make_date('2015-01-01')) GigEventFactory.create_batch(2, date=make_date('2015-06-01')) TheatreEventFactory.create_batch(1, date=make_date('2017-01-01')) GigEventFactory.create_batch(1, date=make_date('2018-01-01')) TheatreEventFactory.create_batch(1, date=make_date('2018-01-01'))
def test_data(self): "General format of data is OK" GigEventFactory.create_batch(3, date=make_date('2018-01-01')) result = EventsGenerator('gig').get_per_year() self.assertIn('data', result) self.assertEqual(len(result['data']), 1) self.assertEqual(result['data'][0]['label'], '2018') self.assertEqual(result['data'][0]['value'], 3)
def test_result(self): GigEventFactory.create_batch(6, date=make_date('2017-02-12')) result = recent_events_card(5) self.assertEqual(result['card_title'], 'Recent events') self.assertEqual(len(result['event_list']), 5)
def test_queryset_num(self): "It should return the number of events requested." GigEventFactory.create_batch(6, date=make_date('2017-02-12')) qs = recent_events(5) self.assertEqual(len(qs), 5)