def test_run_exception(self, mock_sys):
class TestException(Exception):
pass
expected_app_root = "/tmp/test/app_root"
expected_handler = "app.my_test_handler"
expected_lambda_runtime_api_addr = "test_addr"
mock_sys.exit.side_effect = TestException("Boom!")
with self.assertRaises(TestException):
bootstrap.run(expected_app_root, expected_handler,
expected_lambda_runtime_api_addr)
mock_sys.exit.assert_called_once_with(1)
def test_run(self, mock_runtime_client, mock_handle_event_request):
expected_app_root = "/tmp/test/app_root"
expected_handler = "app.my_test_handler"
expected_lambda_runtime_api_addr = "test_addr"
mock_event_request = MagicMock()
mock_event_request.x_amzn_trace_id = "123"
mock_runtime_client.return_value.wait_next_invocation.side_effect = [
mock_event_request,
MagicMock(),
]
with self.assertRaises(TypeError):
bootstrap.run(expected_app_root, expected_handler,
expected_lambda_runtime_api_addr)
mock_handle_event_request.assert_called_once()