def test_continues_uploading_messages_when_they_downloaded_ok_but_one_of_them_has__missing__mesh_header(
):
successful_message_1 = mock_mesh_message()
successful_message_2 = mock_mesh_message()
unsuccessful_message = mock_mesh_message(
validation_error=_a_missing_header_exception())
mock_uploader = MagicMock()
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
forwarder = build_forwarder(
list_message_ids=[
successful_message_1.id,
unsuccessful_message.id,
successful_message_2.id,
],
retrieve_message=[
successful_message_1, unsuccessful_message, successful_message_2
],
uploader=mock_uploader,
probe=probe,
)
forwarder.forward_messages()
mock_uploader.upload.assert_has_calls([
call(successful_message_1, forward_message_event),
call(successful_message_2, forward_message_event),
])
def test_continues_uploading_messages_when_one_of_them_has_invalid_mesh_header(
):
successful_message_1 = mock_mesh_message()
successful_message_2 = mock_mesh_message()
unsuccessful_message = mock_mesh_message(
validation_error=_an_invalid_header_exception())
mock_uploader = MagicMock()
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
forwarder = build_forwarder(
incoming_messages=[
successful_message_1, unsuccessful_message, successful_message_2
],
uploader=mock_uploader,
probe=probe,
)
forwarder.forward_messages()
mock_uploader.upload.assert_has_calls([
call(successful_message_1, forward_message_event),
call(successful_message_2, forward_message_event),
])
def test_acknowledges_multiple_message():
mock_messages = [mock_mesh_message(), mock_mesh_message()]
forwarder = build_forwarder(incoming_messages=mock_messages)
forwarder.forward_messages()
for mock_message in mock_messages:
mock_message.acknowledge.assert_called_once()
def test_acknowledges_multiple_message():
mock_messages = [mock_mesh_message(), mock_mesh_message()]
forwarder = build_forwarder(
list_message_ids=[mock_messages[0].id, mock_messages[1].id],
retrieve_message=mock_messages)
forwarder.forward_messages()
for mock_message in mock_messages:
mock_message.acknowledge.assert_called_once()
def test_does_not_catch_generic_exception():
invalid_message = mock_mesh_message(validation_error=Exception)
forwarder = build_forwarder(list_message_ids=[invalid_message],
retrieve_message=[invalid_message])
with pytest.raises(Exception):
forwarder.forward_messages()
def test_raises_retryable_exception_when_mesh_message_ack_raises_mesh_network_exception(
):
mock_message = mock_mesh_message(acknowledge_error=mesh_client_error())
forwarder = build_forwarder(incoming_messages=[mock_message])
with pytest.raises(RetryableException):
forwarder.forward_messages()
def test_that_when_a_network_error_occurs_on_retrieving_multiple_messages_it_logs_them(
):
mock_uploader = MagicMock()
logger = logging.getLogger("awsmesh.forwarder")
good_message = mock_mesh_message()
forwarder = build_forwarder(
list_message_ids=[
"bad_message_id", good_message.id, "another_bad_message_id"
],
retrieve_message=[
mesh_client_network_error(), good_message,
mesh_client_network_error()
],
uploader=mock_uploader,
)
try:
with patch.object(logger, "info") as mock_info:
forwarder.forward_messages()
except Exception:
pass
mock_uploader.upload.assert_called_once_with(good_message, mock.ANY)
mock_info.assert_called_once_with(
"Raising single retryable exception, actually caught 2 message exception(s)"
)
def test_acknowledges_message():
mock_message = mock_mesh_message()
forwarder = build_forwarder(incoming_messages=[mock_message], )
forwarder.forward_messages()
mock_message.acknowledge.assert_called_once()
def test_catches_invalid_header_error():
forwarder = build_forwarder(incoming_messages=[
mock_mesh_message(validation_error=_an_invalid_header_exception())
])
try:
forwarder.forward_messages()
except InvalidMeshHeader:
pytest.fail("InvalidMeshHeader was raised when it shouldn't have been")
def test_catches_missing_header_error():
forwarder = build_forwarder(retrieve_message=[
mock_mesh_message(validation_error=_a_missing_header_exception())
])
try:
forwarder.forward_messages()
except MissingMeshHeader:
pytest.fail("MissingMeshHeader was raised when it shouldn't have been")
def test_validates_message():
mock_message = mock_mesh_message()
forwarder = build_forwarder(list_message_ids=[mock_message.id],
retrieve_message=[mock_message])
forwarder.forward_messages()
mock_message.validate.assert_called_once()
def test_acknowledges_message():
mock_message = mock_mesh_message()
forwarder = build_forwarder(
list_message_ids=[mock_message.id],
retrieve_message=[mock_message],
)
forwarder.forward_messages()
mock_message.acknowledge.assert_called_once()
def test_forwards_multiple_messages():
mock_message_one = mock_mesh_message()
mock_message_two = mock_mesh_message()
mock_uploader = MagicMock()
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
forwarder = build_forwarder(
incoming_messages=[mock_message_one, mock_message_two],
uploader=mock_uploader,
probe=probe,
)
forwarder.forward_messages()
mock_uploader.upload.assert_has_calls([
call(mock_message_one, forward_message_event),
call(mock_message_two, forward_message_event),
])
assert mock_uploader.upload.call_count == 2
def test_forwards_message():
mock_message = mock_mesh_message()
mock_uploader = MagicMock()
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
forwarder = build_forwarder(incoming_messages=[mock_message],
uploader=mock_uploader,
probe=probe)
forwarder.forward_messages()
mock_uploader.upload.assert_called_once_with(mock_message,
forward_message_event)
def test_record_message_metadata():
mock_output = MagicMock()
message = mock_mesh_message()
forward_message_event = ForwardMessageEvent(mock_output)
forward_message_event.record_message_metadata(message)
forward_message_event.finish()
mock_output.log_event.assert_called_with(
FORWARD_MESSAGE_EVENT,
{
"messageId": message.id,
"sender": message.sender,
"recipient": message.recipient,
"fileName": message.file_name,
},
)
def test_that_when_a_network_error_occurs_on_retrieving_a_message_it_continues_to_download_other_messages(
):
mock_uploader = MagicMock()
good_message = mock_mesh_message()
forwarder = build_forwarder(
list_message_ids=["bad_message_id", good_message.id],
retrieve_message=[mesh_client_network_error(), good_message],
uploader=mock_uploader,
)
try:
forwarder.forward_messages()
except Exception:
pass
mock_uploader.upload.assert_called_once_with(good_message, mock.ANY)
def test_uploads_message_with_unexpected_mesh_header_if_validation_is_disabled(
):
message_with_unexpected_header = mock_mesh_message(
validation_error=_an_invalid_header_exception())
mock_uploader = MagicMock()
forwarder = build_forwarder(
incoming_messages=[message_with_unexpected_header],
uploader=mock_uploader,
disable_message_header_validation=True,
)
forwarder.forward_messages()
mock_uploader.upload.assert_has_calls([
call(message_with_unexpected_header, mock.ANY),
])
def test_that_when_a_non_network_error_occurs_on_retrieving_a_message_it_is_not_caught_and_processing_stops(
):
mock_uploader = MagicMock()
good_message = mock_mesh_message()
non_network_exception = Exception(
"some funky unexpected occurrence on retrieving message")
forwarder = build_forwarder(
list_message_ids=["bad_message_id", good_message.id],
retrieve_message=[non_network_exception, good_message],
uploader=mock_uploader,
)
with pytest.raises(Exception) as raised_e_info:
forwarder.forward_messages()
assert raised_e_info.value == non_network_exception
assert mock_uploader.upload.call_count == 0
def test_records_message_progress():
probe = MagicMock()
mesh_message = mock_mesh_message()
forwarder = build_forwarder(
incoming_messages=[mesh_message],
probe=probe,
)
forwarder.forward_messages()
probe.assert_has_calls([
call.new_poll_inbox_event(),
call.new_poll_inbox_event().record_message_batch_count(1),
call.new_poll_inbox_event().finish(),
call.new_forward_message_event(),
call.new_forward_message_event().record_message_metadata(mesh_message),
call.new_forward_message_event().finish(),
])
def test_records_error_when_mesh_message_ack_raises_mesh_network_exception():
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
network_error = mesh_client_error("Network error")
mock_message = mock_mesh_message(acknowledge_error=network_error, )
forwarder = build_forwarder(incoming_messages=[mock_message], probe=probe)
with pytest.raises(RetryableException):
forwarder.forward_messages()
forward_message_event.assert_has_calls(
[
call.record_message_metadata(mock_message),
call.record_mesh_client_network_error(network_error),
call.finish(),
],
any_order=False,
)
def test_records_error_when_message_has_invalid_header():
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
header_error = _an_invalid_header_exception(header_name="fruit_header", )
message = mock_mesh_message(validation_error=header_error, )
forwarder = build_forwarder(
incoming_messages=[message],
probe=probe,
)
forwarder.forward_messages()
forward_message_event.assert_has_calls(
[
call.record_message_metadata(message),
call.record_invalid_mesh_header(header_error),
call.finish(),
],
any_order=False,
)
def test_records_error_when_message_uploader_raises_uploader_error():
probe = MagicMock()
forward_message_event = MagicMock()
probe.new_forward_message_event.return_value = forward_message_event
message = mock_mesh_message()
exception = UploaderError("error_message")
forwarder = build_forwarder(
uploader_error=exception,
incoming_messages=[message],
probe=probe,
)
forwarder.forward_messages()
forward_message_event.assert_has_calls(
[
call.record_message_metadata(message),
call.record_uploader_error(exception),
call.finish(),
],
any_order=False,
)
def test_does_not_catch_generic_exception():
forwarder = build_forwarder(
incoming_messages=[mock_mesh_message(validation_error=Exception)])
with pytest.raises(Exception):
forwarder.forward_messages()
