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)