def test_post_invocation_result(self, mock_runtime_client):
runtime_client = LambdaRuntimeClient("localhost:1234")
response_data = "data"
invoke_id = "1234"
runtime_client.post_invocation_result(invoke_id, response_data)
mock_runtime_client.post_invocation_result.assert_called_once_with(
invoke_id, response_data.encode("utf-8"), "application/json")
def test_post_invocation_result_failure(self, mock_runtime_client):
runtime_client = LambdaRuntimeClient("localhost:1234")
response_data = "data"
invoke_id = "1234"
mock_runtime_client.post_invocation_result.side_effect = RuntimeError(
"Failed to post invocation response")
with self.assertRaisesRegex(RuntimeError,
"Failed to post invocation response"):
runtime_client.post_invocation_result(invoke_id, response_data)
def test_post_invocation_result_binary_data(self, mock_runtime_client):
runtime_client = LambdaRuntimeClient("localhost:1234")
response_data = b"binary_data"
invoke_id = "1234"
content_type = "application/octet-stream"
runtime_client.post_invocation_result(invoke_id, response_data,
content_type)
mock_runtime_client.post_invocation_result.assert_called_once_with(
invoke_id, response_data, content_type)