def test_check_request_without_necessary_json_fields(self):
self.request.headers = {'Content-Type': "application/json"}
self.request.json = {'payload2': "fsdf"}
ok, data = check_request(self.request, 5)
self.assertFalse(ok)
self.assertDictEqual(data,
{'error': 'Wrong json structure: {payload: xxx}'})
def test_check_request_without_necessary_headers(self):
self.request.headers = {'Content-Type': "application/json2"}
ok, data = check_request(self.request, 5)
self.assertFalse(ok)
self.assertDictEqual(
data, {
'error':
'Request should contains header: Content-Type: application/json'
})
def save_data(id):
is_ok, data = check_request(request, id)
if not is_ok:
return HTTPResponse(data, 400)
side = request.url.split('/')[-1]
try:
save_file(id, side, data)
except OSError as ex:
return HTTPResponse({'error': str(ex)}, 400)
return HTTPResponse(status=200)
def test_check_request_wrong_ip(self):
self.request.headers = {'Content-Type': "application/json"}
self.request.json = {'payload': "fsdf"}
ok, data = check_request(self.request, 'd2222')
self.assertFalse(ok)
self.assertDictEqual(data, {'error': 'Id should be a integer'})
def test_check_request(self):
self.request.headers = {'Content-Type': "application/json"}
self.request.json = {'payload': "fsdf"}
ok, data = check_request(self.request, 5)
self.assertTrue(ok)
self.assertEqual(data, "fsdf")