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

Add email tests

Browse Source
This commit is contained in:
littlewolf 2019-09-20 14:42:00 +03:00
parent 49d2070544
commit 00d8f09881
1 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
apps/establishment/tests.py
View File

@ -22,22 +22,23 @@ class BaseTestCase(APITestCase):
self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'), self.client.cookies = SimpleCookie({'access_token': tokkens.get('access_token'),
'refresh_token': tokkens.get('refresh_token')}) 'refresh_token': tokkens.get('refresh_token')})
self.establishment_type = EstablishmentType.objects.create(name="Test establishment type")
class EstablishmentTests(BaseTestCase): class EstablishmentTests(BaseTestCase):
def test_establishment_CRD(self): def test_establishment_CRD(self):
response = self.client.get('/api/back/establishments/', format='json') response = self.client.get('/api/back/establishments/', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(response.status_code, status.HTTP_200_OK)
establishment_type = EstablishmentType.objects.create(name="Test establishment type")
data = { data = {
'name': 'Test establishment', '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') response = self.client.post('/api/back/establishments/', data=data, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(response.status_code, status.HTTP_201_CREATED)
print(response.json())
response = self.client.get('/api/back/establishments/1/', format='json') response = self.client.get('/api/back/establishments/1/', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK) 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') response = self.client.delete('/api/back/establishments/employees/1/', format='json')
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) 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)