def test_integrations_uninstall(cli_runner): """ Assert that 'newrelic-lambda integrations uninstall' uninstall the log ingestion function/role if present """ register_groups(cli) result = cli_runner.invoke( cli, [ "integrations", "uninstall", "--no-aws-permissions-check", "--nr-account-id", "12345678", ], env={"AWS_DEFAULT_REGION": "us-east-1"}, input="y\ny", ) assert result.exit_code == 0 assert result.stdout == ( "This will uninstall the New Relic AWS Lambda integration role. Are you sure you want to proceed? [y/N]: y\n" # noqa "No New Relic AWS Lambda Integration found, skipping\n" "This will uninstall the New Relic AWS Lambda log ingestion function and role. Are you sure you want to proceed? [y/N]: y\n" # noqa "No New Relic AWS Lambda log ingestion found in region us-east-1, skipping\n" "✨ Uninstall Complete ✨\n" ) session = boto3.Session(region_name="us-east-1") create_integration_role(session, None, 12345678) create_log_ingestion_function(session, "mock-nr-license-key") result2 = cli_runner.invoke( cli, [ "integrations", "uninstall", "--no-aws-permissions-check", "--nr-account-id", "12345678", ], env={"AWS_DEFAULT_REGION": "us-east-1"}, input="y\ny", ) assert result2.exit_code == 0 assert result2.stdout == ( "This will uninstall the New Relic AWS Lambda integration role. Are you sure you want to proceed? [y/N]: y\n" # noqa "Deleting New Relic AWS Lambda Integration stack 'NewRelicLambdaIntegrationRole-12345678'\n" # noqa "Waiting for stack deletion to complete, this may take a minute... ✔️ Done\n" "This will uninstall the New Relic AWS Lambda log ingestion function and role. Are you sure you want to proceed? [y/N]: y\n" # noqa "Deleting New Relic log ingestion stack 'NewRelicLogIngestion'\n" "Waiting for stack deletion to complete, this may take a minute... ✔️ Done\n" "✨ Uninstall Complete ✨\n" )
def install( aws_profile, aws_region, aws_permissions_check, aws_role_policy, linked_account_name, nr_account_id, nr_api_key, nr_region, ): """Install New Relic AWS Lambda Integration""" session = boto3.Session(profile_name=aws_profile, region_name=aws_region) if aws_permissions_check: permissions.ensure_integration_install_permissions(session) click.echo("Validating New Relic credentials") gql_client = api.validate_gql_credentials(nr_account_id, nr_api_key, nr_region) click.echo("Retrieving integration license key") nr_license_key = api.retrieve_license_key(gql_client) click.echo("Checking for a pre-existing link between New Relic and AWS") integrations.validate_linked_account(session, gql_client, linked_account_name) click.echo( "Creating the AWS role for the New Relic AWS Lambda Integration") role = integrations.create_integration_role(session, aws_role_policy, nr_account_id) install_success = False if role: click.echo("Linking New Relic account to AWS account") api.create_integration_account(gql_client, nr_account_id, linked_account_name, role) click.echo( "Enabling Lambda integration on the link between New Relic and AWS" ) install_success = api.enable_lambda_integration( gql_client, nr_account_id, linked_account_name) click.echo( "Creating newrelic-log-ingestion Lambda function in AWS account") install_success = install_success and integrations.install_log_ingestion( session, nr_license_key) if install_success: done("Install Complete") else: failure("Install Incomplete. See messages above for details.")
def install(ctx, **kwargs): """Install New Relic AWS Lambda Integration""" input = IntegrationInstall(session=None, verbose=ctx.obj["VERBOSE"], **kwargs) input = input._replace( session=boto3.Session( profile_name=input.aws_profile, region_name=input.aws_region ) ) if not input.linked_account_name: input = input._replace( linked_account_name=( "New Relic Lambda Integration - %s" % integrations.get_aws_account_id(input.session) ) ) if input.aws_permissions_check: permissions.ensure_integration_install_permissions(input) click.echo("Validating New Relic credentials") gql_client = api.validate_gql_credentials(input) click.echo("Retrieving integration license key") nr_license_key = api.retrieve_license_key(gql_client) click.echo("Checking for a pre-existing link between New Relic and AWS") integrations.validate_linked_account(gql_client, input) install_success = True click.echo("Creating the AWS role for the New Relic AWS Lambda Integration") role = integrations.create_integration_role(input) install_success = install_success and role if role: click.echo("Linking New Relic account to AWS account") res = api.create_integration_account(gql_client, input, role) install_success = res and install_success click.echo("Enabling Lambda integration on the link between New Relic and AWS") res = api.enable_lambda_integration(gql_client, input) install_success = res and install_success if input.enable_license_key_secret: click.echo("Creating the managed secret for the New Relic License Key") res = integrations.install_license_key(input, nr_license_key) install_success = install_success and res if input.enable_cw_ingest: click.echo("Creating newrelic-log-ingestion Lambda function in AWS account") res = integrations.install_log_ingestion(input, nr_license_key) install_success = res and install_success if install_success: done("Install Complete") if input.verbose: click.echo( "\nNext steps: Add the New Relic layers to your Lambda functions with " "the below command.\n" ) command = [ "$", "newrelic-lambda", "layers", "install", "--function", "all", "--nr-account-id", input.nr_account_id, ] if input.aws_profile: command.append("--aws-profile %s" % input.aws_profile) if input.aws_region: command.append("--aws-region %s" % input.aws_region) click.echo(" ".join(command)) else: failure("Install Incomplete. See messages above for details.", exit=True)