def test_guest_register(self): # Create guest response = self.client.get(reverse('home')) user = response.context['user'] token = response.wsgi_request.token self.assertTrue(user.is_guest) self.client.credentials( HTTP_AUTHORIZATION='Token {}'.format(token,) ) data = { 'password': '******' } expected_response_keys = {'token', 'is_guest', 'number', 'certificate'} response = self.client.post(self.register_url, data) filename = "{}.pub".format(response.data['number'],) storage = get_storage() self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(expected_response_keys, set(response.data.keys())) self.assertTrue(storage.exists(filename)) self.assertEqual(storage.url(filename), response.data['certificate'])
def test_login(self): data = self.data.copy() data['number'] = self.number filename = '{}.pub'.format(self.number,) certificate_file = get_storage().path(filename) with open(certificate_file, 'rb') as file_data: data['certificate'] = file_data response = self.client.post(self.login_url, data) expected_response_keys = {'token', 'is_guest', 'number'} self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(expected_response_keys, set(response.data.keys()))
def test_no_guest_register(self): data = { 'password': '******' } expected_response_keys = {'token', 'is_guest', 'number', 'certificate'} response = self.client.post(self.register_url, data) filename = "{}.pub".format(response.data['number'],) storage = get_storage() self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(expected_response_keys, set(response.data.keys())) self.assertTrue(storage.exists(filename)) self.assertEqual(storage.url(filename), response.data['certificate'])
def test_register(self): data = { "password": "******", "first_name": "Test", "last_name": "Testoff", "middle_name": "J", "phone": "0977678889", "email": "*****@*****.**", } expected_response_keys = {'token', 'is_guest', 'number', 'certificate'} response = self.client.post(self.register_url, data) filename = "{}.pub".format(response.data['number'],) storage = get_storage() self.assertEqual(response.status_code, status.HTTP_201_CREATED) self.assertEqual(expected_response_keys, set(response.data.keys())) self.assertTrue(storage.exists(filename)) self.assertEqual(storage.url(filename), response.data['certificate'])