def test_client_handles_error_response_and_fire_response_handler(self):
self.client._fire_response = Mock(return_value=None)
# Create and process Request
msg = ErrorResponse(request=None, error_code=666, error_message="test message")
payload = msg._to_bytes()
# Handle error response message
self.client._handle_message(self.test_channel, payload)
# Check that message response was properly delivered to handler
self.assertEqual(self.client._fire_response.call_count, 1)
def test_error_response(self):
error_code = 99
error_message = "This is an error"
response = ErrorResponse(request=None,
error_code=error_code,
error_message=error_message)
PP.pprint(vars(response))
message = response._to_bytes()
PP.pprint(message)
result = Message._from_bytes(message)
PP.pprint(vars(result))
self.assertEqual(error_code, result.error_code)
self.assertEqual(error_message, result.error_message)
self.assertEqual(Message.MESSAGE_TYPE_ERROR, result.message_type)
def test_error_response(self):
error_code = 99
error_message = "This is an error"
response = ErrorResponse(request=None,
error_code=error_code,
error_message=error_message)
pp.pprint(vars(response))
message = response._to_bytes()
pp.pprint(message)
result = Message._from_bytes(message)
pp.pprint(vars(result))
assert result.error_code == error_code
assert result.error_message == error_message
assert result.message_type == Message.MESSAGE_TYPE_ERROR
def on_request(self, request):
if self._return_error:
response = ErrorResponse(request,
error_code=self._error_code,
error_message=self._error_message)
else:
response = Response(request)
def run_task():
try:
self.m_client.send_response(response)
except Exception, e:
logging.info(e.message)
raise e
def on_request(self, request):
if self._return_error:
response = ErrorResponse(request, error_code=self._error_code, error_message=self._error_message)
else:
response = Response(request)
def run_task():
try:
self.m_client.send_response(response)
except Exception as ex:
logging.info(ex)
raise ex
self.m_executor.submit(run_task)