Fix establishment tests

This commit is contained in:
littlewolf 2019-09-20 16:28:23 +03:00
parent aaff7fd3ba
commit 85fef10eb3

View File

@ -39,13 +39,14 @@ class EstablishmentTests(BaseTestCase):
} }
response = self.client.post('/api/back/establishments/', data=data, format='json') response = self.client.post('/api/back/establishments/', data=data, format='json')
print(f"ESTABLISHMENT: {response.json()}")
self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(response.status_code, status.HTTP_201_CREATED)
response = self.client.get('/api/back/establishments/2/', format='json') establishment = response.json()
response = self.client.get(f'/api/back/establishments/{establishment["id"]}/', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)
response = self.client.delete('/api/back/establishments/2/', format='json') response = self.client.delete(f'/api/back/establishments/{establishment["id"]}/', format='json')
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)