def test_post_invocation_error(self, mock_runtime_client):
runtime_client = LambdaRuntimeClient("localhost:1234")
error_data = "data"
invoke_id = "1234"
xray_fault = "xray_fault"
runtime_client.post_invocation_error(invoke_id, error_data, xray_fault)
mock_runtime_client.post_error.assert_called_once_with(
invoke_id, error_data, xray_fault)
def test_post_invocation_error_with_too_large_xray_cause(
self, mock_runtime_client):
runtime_client = LambdaRuntimeClient("localhost:1234")
error_data = "data"
invoke_id = "1234"
too_large_xray_fault = "a" * int(1024 * 1024)
runtime_client.post_invocation_error(invoke_id, error_data,
too_large_xray_fault)
mock_runtime_client.post_error.assert_called_once_with(
invoke_id, error_data, "")