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)