From 00d8f09881093fddcb83148869e7cf70ee22bf1b Mon Sep 17 00:00:00 2001 From: littlewolf Date: Fri, 20 Sep 2019 14:42:00 +0300 Subject: [PATCH] Add email tests --- apps/establishment/tests.py | 31 ++++++++++++++++++++++++++++--- 1 file changed, 28 insertions(+), 3 deletions(-) diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py index 08e89f10..6d84cbe1 100644 --- a/apps/establishment/tests.py +++ b/apps/establishment/tests.py @@ -22,22 +22,23 @@ class BaseTestCase(APITestCase): self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'), 'refresh_token': tokkens.get('refresh_token')}) + self.establishment_type = EstablishmentType.objects.create(name="Test establishment type") + class EstablishmentTests(BaseTestCase): def test_establishment_CRD(self): response = self.client.get('/api/back/establishments/', format='json') self.assertEqual(response.status_code, status.HTTP_200_OK) - establishment_type = EstablishmentType.objects.create(name="Test establishment type") + data = { 'name': 'Test establishment', - 'type_id': establishment_type.id, + 'type_id': self.establishment_type.id, } response = self.client.post('/api/back/establishments/', data=data, format='json') self.assertEqual(response.status_code, status.HTTP_201_CREATED) - print(response.json()) response = self.client.get('/api/back/establishments/1/', format='json') self.assertEqual(response.status_code, status.HTTP_200_OK) @@ -65,3 +66,27 @@ class EmployeeTests(BaseTestCase): response = self.client.delete('/api/back/establishments/employees/1/', format='json') self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + +class EmailTests(BaseTestCase): + def test_email_CRD(self): + response = self.client.get('/api/back/establishments/emails/', format='json') + self.assertEqual(response.status_code, status.HTTP_200_OK) + + establishment = Establishment.objects.create( + name="Test establishment", + establishment_type_id=self.establishment_type.id + ) + + data = { + 'email': "test@test.com", + 'establishment': establishment.id + } + + response = self.client.post('/api/back/establishments/emails/', data=data) + self.assertEqual(response.status_code, status.HTTP_201_CREATED) + + response = self.client.get('/api/back/establishments/emails/1/', format='json') + self.assertEqual(response.status_code, status.HTTP_200_OK) + + response = self.client.delete('/api/back/establishments/emails/1/') + self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)