def test_retrieve_show(self):
"""Test retrieving a show"""
show = sample_show(name="Show 1")
sample_show(name="Show 2")
res = self.client.get(detail_url(show.id))
serializer = ShowSerializer(show)
self.assertEqual(res.status_code, status.HTTP_200_OK)
self.assertEqual(res.data, serializer.data)
def test_create_season_successful(self):
"""Test creating a new season"""
show = sample_show()
payload = {
'name': 'season 1',
'num_eps': 10,
'last_update': date.today(),
'show': show.id
}
res = self.client.post(SEASONS_URL, payload)
exists = Season.objects.filter(name=payload['name']).exists()
self.assertEqual(res.status_code, status.HTTP_201_CREATED)
self.assertTrue(exists)
def test_create_ep_successful(self):
"""Test creating a new ep"""
show = sample_show()
season = sample_season(show)
payload = {
'name': 'ep 1',
'idx': 10,
'last_update': date.today(),
'show': show.id,
'season': season.id
}
res = self.client.post(EPS_URL, payload)
exists = Ep.objects.filter(
name=payload['name']
).exists()
self.assertEqual(res.status_code, status.HTTP_201_CREATED)
self.assertTrue(exists)
def test_retrieve_shows(self):
"""Test retrieving shows"""
sample_show(name='Show 1')
sample_show(name='Show 2')
sample_show(name='Show 3')
res = self.client.get(SHOWS_URL)
shows = Show.objects.all().order_by('name')
serializer = ShowSerializer(shows, many=True)
self.assertEqual(res.status_code, status.HTTP_200_OK)
self.assertEqual(res.data, serializer.data)
def setUp(self):
self.client = APIClient()
self.show = sample_show()