def test_disable_python(self, mock_gen_aws_client, mock_get_serialized_lambda_function): mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.get_function = MagicMock( return_value={ "Configuration": { "FunctionName": "lambdaF", "Runtime": "python3.6", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaF", "Handler": "sentry_sdk.integrations.init_serverless_sdk.sentry_lambda_handler", "Layers": [ { "Arn": "arn:aws:lambda:us-east-2:1234:layer:something-else:2" }, { "Arn": "arn:aws:lambda:us-east-2:1234:layer:my-python-layer:34" }, ], "Environment": { "Variables": { "SENTRY_INITIAL_HANDLER": "lambda_handler.test_handler", "SENTRY_DSN": self.sentry_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", "OTHER": "hi", } }, }, }) mock_client.update_function_configuration = MagicMock() return_value = { "name": "lambdaF", "runtime": "python3.6", "version": -1, "outOfDate": False, "enabled": False, } mock_get_serialized_lambda_function.return_value = return_value assert self.get_response(action="disable", target="lambdaF").data == return_value mock_client.get_function.assert_called_with(FunctionName="lambdaF") mock_client.update_function_configuration.assert_called_with( FunctionName="lambdaF", Layers=["arn:aws:lambda:us-east-2:1234:layer:something-else:2"], Environment={"Variables": { "OTHER": "hi" }}, Handler="lambda_handler.test_handler", )
def test_update_version(self, mock_gen_aws_client, mock_get_serialized_lambda_function): mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.get_function = MagicMock( return_value={ "Configuration": { "FunctionName": "lambdaD", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaD", "Layers": [ { "Arn": "arn:aws:lambda:us-east-2:1234:layer:something-else:2" }, { "Arn": "arn:aws:lambda:us-east-2:1234:layer:my-layer:2" }, ], "Environment": { "Variables": { "NODE_OPTIONS": "-r @sentry/serverless/dist/auto", "SENTRY_DSN": self.sentry_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", "OTHER": "hi", } }, }, }) mock_client.update_function_configuration = MagicMock() return_value = { "name": "lambdaD", "runtime": "nodejs10.x", "version": 3, "outOfDate": False, "enabled": True, } mock_get_serialized_lambda_function.return_value = return_value assert self.get_response(action="updateVersion", target="lambdaD").data == return_value mock_client.get_function.assert_called_with(FunctionName="lambdaD") mock_client.update_function_configuration.assert_called_with( FunctionName="lambdaD", Layers=[ { "Arn": "arn:aws:lambda:us-east-2:1234:layer:something-else:2" }, "arn:aws:lambda:us-east-2:1234:layer:my-layer:3", ], )
def test_lambda_setup_layer_too_many_requests_exception( self, mock_react_view, mock_gen_aws_client, mock_get_supported_functions ): class MockException(Exception): pass too_many_requests_err = ( "An error occurred (TooManyRequestsException) when calling the " "UpdateFunctionConfiguration operation (reached max retries: 4): " "Rate exceeded" ) mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock( side_effect=Exception(too_many_requests_err) ) mock_client.describe_account = MagicMock(return_value={"Account": {"Name": "my_name"}}) mock_client.exceptions = MagicMock() mock_client.exceptions.ResourceConflictException = MockException mock_get_supported_functions.return_value = [ { "FunctionName": "lambdaB", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaB", }, ] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } resp = self.client.post( self.setup_path, {"lambdaB": True}, format="json", HTTP_ACCEPT="application/json", headers={"Content-Type": "application/json", "Accept": "application/json"}, ) assert resp.status_code == 200 assert not Integration.objects.filter(provider=self.provider.key).exists() failures = [ { "name": "lambdaB", "error": "Something went wrong! Please enable function manually after installation", } ] mock_react_view.assert_called_with( ANY, "awsLambdaFailureDetails", {"lambdaFunctionFailures": failures, "successCount": 0} )
def test_lambda_setup_layer_missing_role_error( self, mock_react_view, mock_gen_aws_client, mock_get_supported_functions ): class MockException(Exception): pass missing_role_err = ( "An error occurred (InvalidParameterValueException) when " "calling the UpdateFunctionConfiguration operation: " "The role defined for the function cannot be " "assumed by Lambda." ) mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock( side_effect=Exception(missing_role_err) ) mock_client.describe_account = MagicMock(return_value={"Account": {"Name": "my_name"}}) mock_client.exceptions = MagicMock() mock_client.exceptions.ResourceConflictException = MockException mock_get_supported_functions.return_value = [ { "FunctionName": "lambdaB", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaB", }, ] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } resp = self.client.post( self.setup_path, {"lambdaB": True}, format="json", HTTP_ACCEPT="application/json", headers={"Content-Type": "application/json", "Accept": "application/json"}, ) assert resp.status_code == 200 assert not Integration.objects.filter(provider=self.provider.key).exists() failures = [ {"name": "lambdaB", "error": "Invalid role associated with the lambda function"} ] mock_react_view.assert_called_with( ANY, "awsLambdaFailureDetails", {"lambdaFunctionFailures": failures, "successCount": 0} )
def test_lambda_setup_layer_error( self, mock_react_view, mock_gen_aws_client, mock_get_supported_functions ): class MockException(Exception): pass bad_layer = "arn:aws:lambda:us-east-2:546545:layer:another-layer:5" mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock( side_effect=Exception(f"Layer version {bad_layer} does not exist") ) mock_client.describe_account = MagicMock(return_value={"Account": {"Name": "my_name"}}) mock_client.exceptions = MagicMock() mock_client.exceptions.ResourceConflictException = MockException mock_get_supported_functions.return_value = [ { "FunctionName": "lambdaA", "Runtime": "nodejs12.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaA", }, { "FunctionName": "lambdaB", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaB", }, ] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } resp = self.client.post( self.setup_path, {"lambdaB": True}, format="json", HTTP_ACCEPT="application/json", headers={"Content-Type": "application/json", "Accept": "application/json"}, ) assert resp.status_code == 200 assert not Integration.objects.filter(provider=self.provider.key).exists() failures = [{"name": "lambdaB", "error": "Invalid existing layer another-layer"}] mock_react_view.assert_called_with( ANY, "awsLambdaFailureDetails", {"lambdaFunctionFailures": failures, "successCount": 0} )
def test_enable_python_layer_on_already_enabled( self, mock_gen_aws_client, mock_get_serialized_lambda_function): """ Test that ensures that if sentry-sdk is already enabled, then re-enabling it should not override the env variables since it could be problematic since the SENTRY_INITIAL_HANDLER env variable could be overriden the second time with "sentry_sdk.integrations.init_serverless_sdk. sentry_lambda_handler" and then disabling the sentry-sdk, would break the function because the Handler will be updated with an incorrect SENTRY_INITIAL_HANDLER value """ mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.get_function = MagicMock( return_value={ "Configuration": { "FunctionName": "lambdaZ", "Runtime": "python3.8", "Handler": "sentry_sdk.integrations.init_serverless_sdk.sentry_lambda_handler", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaZ", "Layers": [ "arn:aws:lambda:us-east-2:1234:layer:something-else:2", "arn:aws:lambda:us-east-2:1234:layer:my-python-layer:34", ], "Environment": { "Variables": { "SENTRY_INITIAL_HANDLER": "lambda_handler.test_handler", "SENTRY_DSN": self.sentry_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", } }, }, }) mock_client.update_function_configuration = MagicMock() return_value = { "name": "lambdaZ", "runtime": "python3.8", "version": 3, "outOfDate": False, "enabled": True, } mock_get_serialized_lambda_function.return_value = return_value assert self.get_response(action="enable", target="lambdaZ").data == return_value mock_client.get_function.assert_called_with(FunctionName="lambdaZ") mock_client.update_function_configuration.assert_called_with( FunctionName="lambdaZ", Layers=[ "arn:aws:lambda:us-east-2:1234:layer:something-else:2", "arn:aws:lambda:us-east-2:1234:layer:my-python-layer:34", ], Environment={ "Variables": { "SENTRY_INITIAL_HANDLER": "lambda_handler.test_handler", "SENTRY_DSN": self.sentry_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", } }, Handler= "sentry_sdk.integrations.init_serverless_sdk.sentry_lambda_handler", )
def test_lambda_setup_layer_success(self, mock_gen_aws_client, mock_get_supported_functions): mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock() mock_client.describe_account = MagicMock( return_value={"Account": { "Name": "my_name" }}) mock_get_supported_functions.return_value = [ { "FunctionName": "lambdaA", "Runtime": "nodejs12.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaA", }, { "FunctionName": "lambdaB", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaB", }, ] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "arn": arn, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } sentry_project_dsn = ProjectKey.get_default( project=self.projectA).get_dsn(public=True) resp = self.client.post( self.setup_path, {"lambdaB": True}, format="json", HTTP_ACCEPT="application/json", headers={ "Content-Type": "application/json", "Accept": "application/json" }, ) assert resp.status_code == 200 mock_client.update_function_configuration.assert_called_with( FunctionName="lambdaB", Layers=["arn:aws:lambda:us-east-2:1234:layer:my-layer:3"], Environment={ "Variables": { "NODE_OPTIONS": "-r @sentry/serverless/dist/auto", "SENTRY_DSN": sentry_project_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", } }, ) integration = Integration.objects.get(provider=self.provider.key) assert integration.name == "my_name us-east-2" assert integration.external_id == "599817902985-us-east-2" assert integration.metadata == { "arn": arn, "aws_external_id": aws_external_id } assert OrganizationIntegration.objects.filter( integration=integration, organization=self.organization)
def test_lambda_setup_layer_env_vars_limit_exceeded_exception( self, mock_react_view, mock_gen_aws_client, mock_get_supported_functions): class MockException(Exception): pass env_vars_size_limit_err = ( "An error occurred (InvalidParameterValueException) when calling the " "UpdateFunctionConfiguration operation: Lambda was unable to configure " "your environment variables because the environment variables you have " "provided exceeded the 4KB limit. String measured: {'MESSAGE':'This is production " "environment','TARGET_ENV' :'pre-production','IS_SERVERLESS':'true','STAGE':'pre-prod'" ) mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock( side_effect=Exception(env_vars_size_limit_err)) mock_client.describe_account = MagicMock( return_value={"Account": { "Name": "my_name" }}) mock_client.exceptions = MagicMock() mock_client.exceptions.ResourceConflictException = MockException mock_get_supported_functions.return_value = [ { "FunctionName": "lambdaB", "Runtime": "nodejs10.x", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaB", }, ] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } resp = self.client.post( self.setup_path, {"lambdaB": True}, format="json", HTTP_ACCEPT="application/json", headers={ "Content-Type": "application/json", "Accept": "application/json" }, ) assert resp.status_code == 200 assert not Integration.objects.filter( provider=self.provider.key).exists() failures = [{ "name": "lambdaB", "error": "Environment variables size limit of 4KB was exceeded", }] mock_react_view.assert_called_with(ANY, "awsLambdaFailureDetails", { "lambdaFunctionFailures": failures, "successCount": 0 })
def test_python_lambda_setup_layer_success(self, mock_gen_aws_client, mock_get_supported_functions): mock_client = Mock() mock_gen_aws_client.return_value = mock_client mock_client.update_function_configuration = MagicMock() mock_client.describe_account = MagicMock( return_value={"Account": { "Name": "my_name" }}) mock_get_supported_functions.return_value = [{ "FunctionName": "lambdaA", "Handler": "lambda_handler.test_handler", "Runtime": "python3.6", "FunctionArn": "arn:aws:lambda:us-east-2:599817902985:function:lambdaA", }] aws_external_id = "12-323" self.pipeline.state.step_index = 2 self.pipeline.state.data = { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, "project_id": self.projectA.id, } sentry_project_dsn = ProjectKey.get_default( project=self.projectA).get_dsn(public=True) resp = self.client.post( self.setup_path, data={"lambdaA": True}, format="json", HTTP_ACCEPT="application/json", headers={ "Content-Type": "application/json", "Accept": "application/json" }, ) assert resp.status_code == 200 mock_client.update_function_configuration.assert_called_once_with( FunctionName="lambdaA", Layers=["arn:aws:lambda:us-east-2:1234:layer:my-python-layer:34"], Environment={ "Variables": { "SENTRY_INITIAL_HANDLER": "lambda_handler.test_handler", "SENTRY_DSN": sentry_project_dsn, "SENTRY_TRACES_SAMPLE_RATE": "1.0", } }, Handler= "sentry_sdk.integrations.init_serverless_sdk.sentry_lambda_handler", ) integration = Integration.objects.get(provider=self.provider.key) assert integration.name == "my_name us-east-2" assert integration.external_id == "599817902985-us-east-2" assert integration.metadata == { "region": region, "account_number": account_number, "aws_external_id": aws_external_id, } assert OrganizationIntegration.objects.filter( integration=integration, organization=self.organization)