From 85fef10eb301371c62c1ee5ce3aa25139882a5f9 Mon Sep 17 00:00:00 2001 From: littlewolf Date: Fri, 20 Sep 2019 16:28:23 +0300 Subject: [PATCH] Fix establishment tests --- apps/establishment/tests.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py index c112e067..12afdce2 100644 --- a/apps/establishment/tests.py +++ b/apps/establishment/tests.py @@ -39,13 +39,14 @@ class EstablishmentTests(BaseTestCase): } 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) - 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) - 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)