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_RequestIsPassedInToHandleRequest(self):
x = WebApi()
x._handle_request = mock.MagicMock(return_value={})
request = object()
x.post(request)
x._handle_request.assert_called_once_with(request)
