def do_test_error(self, expected_message, request=None, **kwargs):
x = WebApi()
if request is None:
request = self.new_request(**kwargs)
with self.assertRaises(x.Error) as cm:
x._verify_content_length(request)
message = "{}".format(cm.exception)
self.assertEqual(message, expected_message)
def test_HandleRequestSuccess(self):
x = WebApi()
x._handle_request = mock.MagicMock(return_value={"one": "two"})
response = x.post(object())
self.assertEqual(response.status_code, 200)
self.assertEqual(response["Content-Type"], "application/json")
response_dict = json.loads(response.content.decode())
self.assertDictEqual(response_dict, {
"one": "two",
"success": 1,
})
def test_HandleRequestError(self):
x = WebApi()
exception = x.Error("forced exception")
x._handle_request = mock.MagicMock(side_effect=exception)
response = x.post(object())
self.assertEqual(response.status_code, 200)
self.assertEqual(response["Content-Type"], "application/json")
response_dict = json.loads(response.content.decode())
self.assertDictEqual(response_dict, {
"success": 0,
"error message": "forced exception",
})
def test_Success(self):
x = WebApi()
request = self.new_request(content_type="application/json")
retval = x._verify_content_type(request)
self.assertIsNone(retval)
def test_RequestIsPassedInToHandleRequest(self):
x = WebApi()
x._handle_request = mock.MagicMock(return_value={})
request = object()
x.post(request)
x._handle_request.assert_called_once_with(request)
def test_ContentLengthValue_1(self):
x = WebApi()
request = self.new_request(CONTENT_LENGTH="1")
retval = x._verify_content_length(request)
self.assertEqual(retval, 1)
def test_Success(self):
x = WebApi()
request = self.new_request(CONTENT_LENGTH="1024")
retval = x._verify_content_length(request)
self.assertEqual(retval, 1024)
def do_test_raises_error(self, request, expected_message):
x = WebApi()
with self.assertRaises(x.Error) as cm:
x._verify_content_type(request)
actual_message = "{}".format(cm.exception)
self.assertEqual(actual_message, expected_message)
