phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add class to test childs

Browse Source
This commit is contained in:
littlewolf 2019-09-20 14:55:34 +03:00
parent ee93905556
commit e616d3970c
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
apps/establishment/tests.py
View File

@ -65,19 +65,25 @@ class EmployeeTests(BaseTestCase):
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
class EmailTests(BaseTestCase): # Class to test childs
def test_email_CRD(self): class ChildTestCase(BaseTestCase):
response = self.client.get('/api/back/establishments/emails/', format='json') def setUp(self):
self.assertEqual(response.status_code, status.HTTP_200_OK) super().setUp()
self.establishment = Establishment.objects.create(
establishment = Establishment.objects.create(
name="Test establishment", name="Test establishment",
establishment_type_id=self.establishment_type.id establishment_type_id=self.establishment_type.id
) )
# Test childs
class EmailTests(ChildTestCase):
def test_email_CRD(self):
response = self.client.get('/api/back/establishments/emails/', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
data = { data = {
'email': "test@test.com", 'email': "test@test.com",
'establishment': establishment.id 'establishment': self.establishment.id
} }
response = self.client.post('/api/back/establishments/emails/', data=data) response = self.client.post('/api/back/establishments/emails/', data=data)
@ -90,19 +96,14 @@ class EmailTests(BaseTestCase):
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
class PhoneTests(BaseTestCase): class PhoneTests(ChildTestCase):
def test_phone_CRD(self): def test_phone_CRD(self):
response = self.client.get('/api/back/establishments/phones/', format='json') response = self.client.get('/api/back/establishments/phones/', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)
establishment = Establishment.objects.create(
name="Test establishment",
establishment_type_id=self.establishment_type.id
)
data = { data = {
'phone': "+79999999999", 'phone': "+79999999999",
'establishment': establishment.id 'establishment': self.establishment.id
} }
response = self.client.post('/api/back/establishments/phones/', data=data) response = self.client.post('/api/back/establishments/phones/', data=data)