newrelic-lambda-cli 0.7.6__tar.gz → 0.9.0__tar.gz

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: newrelic-lambda-cli
-Version: 0.7.6
+Version: 0.9.0
 Summary: A CLI to install the New Relic AWS Lambda integration and layers.
 Home-page: https://github.com/newrelic/newrelic-lambda-cli
 Author: New Relic
@@ -55,9 +55,9 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * java8.al2
 * java11
 * java17
-* nodejs12.x
-* nodejs14.x
 * nodejs16.x
+* nodejs18.x
+* nodejs20.x
 * provided
 * provided.al2
 * python3.7
@@ -65,13 +65,14 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * python3.9
 * python3.10
 * python3.11
+* python3.12
 
 **Note:** Automatic handler wrapping is only supported for Node.js, Python and Java. For other runtimes,
 manual function wrapping is required using the runtime specific New Relic agent.
 
 ## Requirements
 
-* Python >= 3.3 <= 3.11
+* Python >= 3.3 <= 3.12
 * Retrieve your [New relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) and [User API Key](https://docs.newrelic.com/docs/apis/get-started/intro-apis/types-new-relic-api-keys#user-api-key)
 
 ## Recommendations
@@ -131,6 +132,7 @@ newrelic-lambda integrations install \
 | `--aws-role-policy` | No | Specify an alternative IAM role policy ARN for this integration. |
 | `--disable-license-key-secret` | No | Don't create a managed secret for your account's New Relic License Key |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Uninstall Integration
 
@@ -144,6 +146,7 @@ newrelic-lambda integrations uninstall
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--force` or `-f` | No | Forces uninstall non-interactively |
 | `--nr-account-id` or `-a` | No | The [New Relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) for the integration. Only required if also uninstalling the New Relic AWS Lambda integration. Can also use the `NEW_RELIC_ACCOUNT_ID` environment variable. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Update Integration
 
@@ -171,6 +174,7 @@ newrelic-lambda integrations update \
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--disable-license-key-secret` | No | Disable automatic creation of the license key secret on update. The secret is not created if it exists. |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 ### AWS Lambda Layers
 
@@ -243,6 +247,7 @@ newrelic-lambda subscriptions install --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to add a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while installing subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--filter-pattern` | No | Specify a custom log subscription filter pattern. To collect all logs use `--filter-pattern ""`. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
@@ -257,6 +262,7 @@ newrelic-lambda subscriptions uninstall --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to remove a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while uninstalling subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
 | `--aws-region` or `-r` | No | The AWS region this function is located. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/README.md

@@ -37,9 +37,9 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * java8.al2
 * java11
 * java17
-* nodejs12.x
-* nodejs14.x
 * nodejs16.x
+* nodejs18.x
+* nodejs20.x
 * provided
 * provided.al2
 * python3.7
@@ -47,13 +47,14 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * python3.9
 * python3.10
 * python3.11
+* python3.12
 
 **Note:** Automatic handler wrapping is only supported for Node.js, Python and Java. For other runtimes,
 manual function wrapping is required using the runtime specific New Relic agent.
 
 ## Requirements
 
-* Python >= 3.3 <= 3.11
+* Python >= 3.3 <= 3.12
 * Retrieve your [New relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) and [User API Key](https://docs.newrelic.com/docs/apis/get-started/intro-apis/types-new-relic-api-keys#user-api-key)
 
 ## Recommendations
@@ -113,6 +114,7 @@ newrelic-lambda integrations install \
 | `--aws-role-policy` | No | Specify an alternative IAM role policy ARN for this integration. |
 | `--disable-license-key-secret` | No | Don't create a managed secret for your account's New Relic License Key |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Uninstall Integration
 
@@ -126,6 +128,7 @@ newrelic-lambda integrations uninstall
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--force` or `-f` | No | Forces uninstall non-interactively |
 | `--nr-account-id` or `-a` | No | The [New Relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) for the integration. Only required if also uninstalling the New Relic AWS Lambda integration. Can also use the `NEW_RELIC_ACCOUNT_ID` environment variable. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Update Integration
 
@@ -153,6 +156,7 @@ newrelic-lambda integrations update \
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--disable-license-key-secret` | No | Disable automatic creation of the license key secret on update. The secret is not created if it exists. |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 ### AWS Lambda Layers
 
@@ -225,6 +229,7 @@ newrelic-lambda subscriptions install --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to add a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while installing subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--filter-pattern` | No | Specify a custom log subscription filter pattern. To collect all logs use `--filter-pattern ""`. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
@@ -239,6 +244,7 @@ newrelic-lambda subscriptions uninstall --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to remove a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while uninstalling subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
 | `--aws-region` or `-r` | No | The AWS region this function is located. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/cli/__init__.py

@@ -6,6 +6,7 @@ from newrelic_lambda_cli.cli import functions, integrations, layers, subscriptio
 
 
 @click.group()
+@click.version_option()
 @click.option("--verbose", "-v", help="Increase verbosity", is_flag=True)
 @click.pass_context
 def cli(ctx, verbose):

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/cli/integrations.py

@@ -39,6 +39,14 @@ def register(group):
     help="Determines if logs are forwarded to New Relic Logging",
     is_flag=True,
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option(
     "--memory-size",
     "-m",
@@ -195,6 +203,14 @@ def install(ctx, **kwargs):
     required=False,
     type=click.INT,
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option("--force", "-f", help="Force uninstall non-interactively", is_flag=True)
 def uninstall(**kwargs):
     """Uninstall New Relic AWS Lambda Integration"""
@@ -250,6 +266,14 @@ def uninstall(**kwargs):
     default=None,
     help="Determines if logs are forwarded to New Relic Logging",
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option(
     "--memory-size",
     "-m",
@@ -272,6 +296,14 @@ def uninstall(**kwargs):
     metavar="<role_name>",
     show_default=False,
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option(
     "--enable-license-key-secret/--disable-license-key-secret",
     default=True,

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/cli/subscriptions.py

@@ -37,6 +37,14 @@ def register(group):
     multiple=True,
     required=True,
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option(
     "excludes",
     "--exclude",
@@ -92,6 +100,14 @@ def install(**kwargs):
     multiple=True,
     required=True,
 )
+@click.option(
+    "--stackname",
+    default="NewRelicLogIngestion",
+    help="The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function",
+    metavar="<arn>",
+    show_default=False,
+    required=False,
+)
 @click.option(
     "excludes",
     "--exclude",

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/functions.py

@@ -70,8 +70,8 @@ def get_aliased_functions(input):
     functions = [
         function
         for function in input.functions
-        if function.lower()
-        not in ("all", "installed", "not-installed", "newrelic-log-ingestion")
+        if function.lower() not in ("all", "installed", "not-installed")
+        and "newrelic-log-ingestion" not in function.lower()
         and function not in input.excludes
     ]
 

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/integrations.py

@@ -43,8 +43,8 @@ def _get_role(session, role_name):
         raise click.UsageError(str(e))
 
 
-def _check_for_ingest_stack(session):
-    return _get_cf_stack_status(session, INGEST_STACK_NAME)
+def _check_for_ingest_stack(session, stack_name):
+    return _get_cf_stack_status(session, stack_name)
 
 
 def _get_cf_stack_status(session, stack_name, nr_account_id=None):
@@ -86,6 +86,62 @@ def _get_cf_stack_status(session, stack_name, nr_account_id=None):
         return res["Stacks"][0]["StackStatus"]
 
 
+def get_unique_newrelic_log_ingestion_name(session, stackname=None):
+    if not stackname:
+        stackname = INGEST_STACK_NAME
+    stack_id = _get_cf_stack_id(session, stack_name=stackname)
+    if stack_id:
+        return "newrelic-log-ingestion-%s" % (stack_id.split("/")[2].split("-")[4])
+
+
+def get_newrelic_log_ingestion_function(session, stackname=None):
+    unique_log_ingestion_name = get_unique_newrelic_log_ingestion_name(
+        session, stackname
+    )
+    if unique_log_ingestion_name:
+        function = get_function(session, unique_log_ingestion_name)
+        return function
+
+
+def _get_cf_stack_id(session, stack_name, nr_account_id=None):
+    """Returns the StackId of the CloudFormation stack if it exists"""
+    try:
+        res = session.client("cloudformation").describe_stacks(StackName=stack_name)
+        if nr_account_id is not None:
+            stack_output_account_id = _get_stack_output_value(
+                session, ["NrAccountId"]
+            ).get("NrAccountId")
+            # Checking outputs here to protect against installs done
+            # with older CLI versions. We don't want to constantly warn users
+            # who installed on previous versions with no outputs.
+            if stack_output_account_id and stack_output_account_id != str(
+                nr_account_id
+            ):
+                warning(
+                    "WARNING: Managed secret already exists in this region for "
+                    "New Relic account {0}.\n"
+                    "Current CLI behavior limits the setup of one managed "
+                    "secret per region.\n"
+                    "To set up an additional secret for New Relic account {1} "
+                    "see our docs:\n{2}.\n"
+                    "Or run this command with --disable-license-key-secret to "
+                    "avoid attempting to create a new managed secret.".format(
+                        stack_output_account_id, nr_account_id, NR_DOCS_ACT_LINKING_URL
+                    )
+                )
+    except botocore.exceptions.ClientError as e:
+        if (
+            e.response
+            and "ResponseMetadata" in e.response
+            and "HTTPStatusCode" in e.response["ResponseMetadata"]
+            and e.response["ResponseMetadata"]["HTTPStatusCode"] in (400, 404)
+        ):
+            return None
+        raise click.UsageError(str(e))
+    else:
+        return res["Stacks"][0]["StackId"]
+
+
 # TODO: Merge this with create_integration_role?
 def _create_role(input):
     assert isinstance(input, IntegrationInstall)
@@ -200,29 +256,35 @@ def _import_log_ingestion_function(input, nr_license_key):
     )
 
     with open(template_path) as template:
-        change_set_name = "%s-IMPORT-%d" % (INGEST_STACK_NAME, int(time.time()))
-        click.echo("Creating change set: %s" % change_set_name)
-
-        change_set = client.create_change_set(
-            StackName=INGEST_STACK_NAME,
-            TemplateBody=template.read(),
-            Parameters=parameters,
-            Capabilities=capabilities,
-            Tags=[{"Key": key, "Value": value} for key, value in input.tags]
-            if input.tags
-            else [],
-            ChangeSetType="IMPORT",
-            ChangeSetName=change_set_name,
-            ResourcesToImport=[
-                {
-                    "ResourceType": "AWS::Lambda::Function",
-                    "LogicalResourceId": "NewRelicLogIngestionFunctionNoCap",
-                    "ResourceIdentifier": {"FunctionName": "newrelic-log-ingestion"},
-                }
-            ],
+        unique_log_ingestion_name = get_unique_newrelic_log_ingestion_name(
+            input.session
         )
+        if unique_log_ingestion_name:
+            change_set_name = "%s-IMPORT-%d" % (INGEST_STACK_NAME, int(time.time()))
+            click.echo("Creating change set: %s" % change_set_name)
+
+            change_set = client.create_change_set(
+                StackName=INGEST_STACK_NAME,
+                TemplateBody=template.read(),
+                Parameters=parameters,
+                Capabilities=capabilities,
+                Tags=[{"Key": key, "Value": value} for key, value in input.tags]
+                if input.tags
+                else [],
+                ChangeSetType="IMPORT",
+                ChangeSetName=change_set_name,
+                ResourcesToImport=[
+                    {
+                        "ResourceType": "AWS::Lambda::Function",
+                        "LogicalResourceId": "NewRelicLogIngestionFunctionNoCap",
+                        "ResourceIdentifier": {
+                            "FunctionName": unique_log_ingestion_name
+                        },
+                    }
+                ],
+            )
 
-        _exec_change_set(client, change_set, "IMPORT")
+            _exec_change_set(client, change_set, "IMPORT")
 
 
 def _create_log_ingestion_function(
@@ -241,11 +303,11 @@ def _create_log_ingestion_function(
     click.echo("Fetching new CloudFormation template url")
     template_url = _get_sar_template_url(input.session)
 
-    change_set_name = "%s-%s-%d" % (INGEST_STACK_NAME, mode, int(time.time()))
+    change_set_name = "%s-%s-%d" % (input.stackname, mode, int(time.time()))
     click.echo("Creating change set: %s" % change_set_name)
 
     change_set = client.create_change_set(
-        StackName=INGEST_STACK_NAME,
+        StackName=input.stackname,
         TemplateURL=template_url,
         Parameters=parameters,
         Capabilities=capabilities,
@@ -256,10 +318,10 @@ def _create_log_ingestion_function(
         ChangeSetName=change_set_name,
     )
 
-    _exec_change_set(client, change_set, mode)
+    _exec_change_set(client, change_set, mode, input.stackname)
 
 
-def _exec_change_set(client, change_set, mode, stack_name=INGEST_STACK_NAME):
+def _exec_change_set(client, change_set, mode, stack_name):
     click.echo(
         "Waiting for change set creation to complete, this may take a minute... ",
         nl=False,
@@ -324,7 +386,6 @@ def update_log_ingestion_function(input):
         len(stack_resources) > 0
         and stack_resources[0]["ResourceType"] == "AWS::CloudFormation::Stack"
     ):
-
         click.echo("Unwrapping nested stack... ", nl=False)
 
         # Set the ingest function itself to disallow deletes
@@ -342,9 +403,24 @@ def update_log_ingestion_function(input):
 
         # We can't change props during import, so let's set them to their current values
         lambda_client = input.session.client("lambda")
-        old_props = lambda_client.get_function_configuration(
-            FunctionName="newrelic-log-ingestion"
+        unique_log_ingestion_name = get_unique_newrelic_log_ingestion_name(
+            input.session
         )
+        old_props = {}
+        try:
+            old_props = lambda_client.get_function_configuration(
+                FunctionName=unique_log_ingestion_name
+            )
+        except lambda_client.exceptions.ResourceNotFoundException:
+            old_props = lambda_client.get_function_configuration(
+                FunctionName="newrelic-log-ingestion"
+            )
+            warning(
+                "It looks like an old log ingestion function is present in this region. "
+                "Consider manually deleting this as it is no longer used and "
+                "has been replaced by a log ingestion function specific to the stack."
+            )
+
         old_role_name = old_props["Role"].split("/")[-1]
         old_nr_license_key = old_props["Environment"]["Variables"]["LICENSE_KEY"]
         old_enable_logs = False
@@ -424,19 +500,19 @@ def remove_log_ingestion_function(input):
     assert isinstance(input, IntegrationUninstall)
 
     client = input.session.client("cloudformation")
-    stack_status = _check_for_ingest_stack(input.session)
+    stack_status = _check_for_ingest_stack(input.session, input.stackname)
     if stack_status is None:
         click.echo(
             "No New Relic AWS Lambda log ingestion found in region %s, skipping"
             % input.session.region_name
         )
         return
-    click.echo("Deleting New Relic log ingestion stack '%s'" % INGEST_STACK_NAME)
-    client.delete_stack(StackName=INGEST_STACK_NAME)
+    click.echo("Deleting New Relic log ingestion stack '%s'" % input.stackname)
+    client.delete_stack(StackName=input.stackname)
     click.echo(
         "Waiting for stack deletion to complete, this may take a minute... ", nl=False
     )
-    client.get_waiter("stack_delete_complete").wait(StackName=INGEST_STACK_NAME)
+    client.get_waiter("stack_delete_complete").wait(StackName=input.stackname)
     success("Done")
 
 
@@ -518,22 +594,35 @@ def install_log_ingestion(
     """
     assert isinstance(input, IntegrationInstall)
     function = get_function(input.session, "newrelic-log-ingestion")
-    if function is None:
-        stack_status = _check_for_ingest_stack(input.session)
-        if stack_status is None:
-            click.echo(
-                "Setting up 'newrelic-log-ingestion' function in region: %s"
-                % input.session.region_name
+    if function:
+        warning(
+            "It looks like an old log ingestion function is present in this region. "
+            "Consider manually deleting this as it is no longer used and "
+            "has been replaced by a log ingestion function specific to the stack."
+        )
+    stack_status = _check_for_ingest_stack(input.session, input.stackname)
+    if stack_status is None:
+        click.echo(
+            "Setting up CloudFormation Stack %s in region: %s"
+            % (input.stackname, input.session.region_name)
+        )
+        try:
+            _create_log_ingestion_function(
+                input,
+                nr_license_key,
             )
-            try:
-                _create_log_ingestion_function(
-                    input,
-                    nr_license_key,
-                )
-            except Exception as e:
-                failure("Failed to create 'newrelic-log-ingestion' function: %s" % e)
-                return False
-        else:
+            return True
+        except Exception as e:
+            failure(
+                "CloudFormation Stack NewRelicLogIngestion exists (status: %s).\n"
+                "Please manually delete the stack and re-run this command."
+                % stack_status
+            )
+            return False
+    else:
+        function = get_newrelic_log_ingestion_function(input.session)
+
+        if function is None:
             failure(
                 "CloudFormation Stack NewRelicLogIngestion exists (status: %s), but "
                 "newrelic-log-ingestion Lambda function does not.\n"
@@ -541,12 +630,13 @@ def install_log_ingestion(
                 % stack_status
             )
             return False
-    else:
-        success(
-            "The 'newrelic-log-ingestion' function already exists in region %s, "
-            "skipping" % input.session.region_name
-        )
-    return True
+        else:
+            success(
+                "The CloudFormation Stack NewRelicLogIngestion and "
+                "newrelic-log-ingestion function already exists in region %s, "
+                "skipping" % input.session.region_name
+            )
+            return True
 
 
 @catch_boto_errors
@@ -558,16 +648,7 @@ def update_log_ingestion(input):
     """
     assert isinstance(input, IntegrationUpdate)
 
-    function = get_function(input.session, "newrelic-log-ingestion")
-    if function is None:
-        failure(
-            "No 'newrelic-log-ingestion' function in region '%s'. "
-            "Run 'newrelic-lambda integrations install' to install it."
-            % input.session.region_name
-        )
-        return False
-
-    stack_status = _check_for_ingest_stack(input.session)
+    stack_status = _check_for_ingest_stack(input.session, input.stackname)
     if stack_status is None:
         failure(
             "No 'NewRelicLogIngestion' stack in region '%s'. "
@@ -578,10 +659,19 @@ def update_log_ingestion(input):
         )
         return False
 
+    function = get_newrelic_log_ingestion_function(input.session, input.stackname)
+    if function is None:
+        failure(
+            "No newrelic-log-ingestion function in region '%s'. "
+            "Run 'newrelic-lambda integrations install' to install it."
+            % input.session.region_name
+        )
+        return False
+
     try:
         update_log_ingestion_function(input)
     except Exception as e:
-        failure("Failed to update 'newrelic-log-ingestion' function: %s" % e)
+        failure("Failed to update newrelic-log-ingestion function: %s" % e)
         return False
     else:
         return True
@@ -592,7 +682,7 @@ def get_log_ingestion_license_key(session):
     """
     Fetches the license key value from the log ingestion function
     """
-    function = get_function(session, "newrelic-log-ingestion")
+    function = get_newrelic_log_ingestion_function(session)
     if function:
         return function["Configuration"]["Environment"]["Variables"]["LICENSE_KEY"]
     return None

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/layers.py

@@ -89,7 +89,7 @@ def _add_new_relic(input, config, nr_license_key):
             "Unsupported Lambda runtime for '%s': %s"
             % (config["Configuration"]["FunctionArn"], runtime)
         )
-        return True
+        return False
 
     architectures = config["Configuration"].get("Architectures", ["x86_64"])
     architecture = architectures[0]

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/subscriptions.py

@@ -5,6 +5,8 @@ import click
 
 from newrelic_lambda_cli.cliutils import failure, success, warning
 from newrelic_lambda_cli.functions import get_function
+from newrelic_lambda_cli.integrations import get_unique_newrelic_log_ingestion_name
+from newrelic_lambda_cli.integrations import get_newrelic_log_ingestion_function
 from newrelic_lambda_cli.types import (
     LayerInstall,
     SubscriptionInstall,
@@ -83,10 +85,17 @@ def _remove_subscription_filter(session, function_name, filter_name):
 @catch_boto_errors
 def create_log_subscription(input, function_name):
     assert isinstance(input, SubscriptionInstall)
-    destination = get_function(input.session, "newrelic-log-ingestion")
+    function = get_function(input.session, "newrelic-log-ingestion")
+    if function:
+        warning(
+            "It looks like an old log ingestion function is present in this region. "
+            "Consider manually deleting this as it is no longer used and "
+            "has been replaced by a log ingestion function specific to the stack."
+        )
+    destination = get_newrelic_log_ingestion_function(input.session, input.stackname)
     if destination is None:
         failure(
-            "Could not find 'newrelic-log-ingestion' function. Is the New Relic AWS "
+            "Could not find newrelic-log-ingestion function. Is the New Relic AWS "
             "integration installed?"
         )
         return False

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/templates/import-template.yaml

@@ -62,7 +62,7 @@ Resources:
         Key: 466768951184/arn:aws:serverlessrepo:us-east-1:463657938898:applications-NewRelic-log-ingestion-versions-2.2.1/2dcb4087-186a-42c8-bc1f-cf93780786c0
       Description: Sends log data from CloudWatch Logs and S3 to New Relic Infrastructure (Cloud integrations) and New Relic Logging
       Handler: function.lambda_handler
-      FunctionName: newrelic-log-ingestion
+      FunctionName: !Join ['-', ['newrelic-log-ingestion', !Select [4, !Split ['-', !Select [2, !Split ['/', !Ref AWS::StackId]]]]]]
       MemorySize:
         Ref: MemorySize
       Runtime: python3.9
@@ -84,7 +84,7 @@ Resources:
         Key: 466768951184/arn:aws:serverlessrepo:us-east-1:463657938898:applications-NewRelic-log-ingestion-versions-2.2.1/2dcb4087-186a-42c8-bc1f-cf93780786c0
       Description: Sends log data from CloudWatch Logs and S3 to New Relic Infrastructure (Cloud integrations) and New Relic Logging
       Handler: function.lambda_handler
-      FunctionName: newrelic-log-ingestion
+      FunctionName: !Join ['-', ['newrelic-log-ingestion', !Select [4, !Split ['-', !Select [2, !Split ['/', !Ref AWS::StackId]]]]]]
       MemorySize:
         Ref: MemorySize
       Runtime: python3.9

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/types.py

@@ -8,6 +8,7 @@ INTEGRATION_INSTALL_KEYS = [
     "aws_permissions_check",
     "aws_role_policy",
     "enable_logs",
+    "stackname",
     "memory_size",
     "linked_account_name",
     "nr_account_id",
@@ -26,6 +27,7 @@ INTEGRATION_UNINSTALL_KEYS = [
     "aws_profile",
     "aws_region",
     "aws_permissions_check",
+    "stackname",
     "nr_account_id",
     "force",
 ]
@@ -36,6 +38,7 @@ INTEGRATION_UPDATE_KEYS = [
     "aws_region",
     "aws_permissions_check",
     "enable_logs",
+    "stackname",
     "memory_size",
     "nr_account_id",
     "nr_api_key",
@@ -80,6 +83,7 @@ SUBSCRIPTION_INSTALL_KEYS = [
     "aws_region",
     "aws_permissions_check",
     "functions",
+    "stackname",
     "excludes",
     "filter_pattern",
 ]
@@ -90,6 +94,7 @@ SUBSCRIPTION_UNINSTALL_KEYS = [
     "aws_region",
     "aws_permissions_check",
     "functions",
+    "stackname",
     "excludes",
 ]
 

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli/utils.py

@@ -22,28 +22,20 @@ RUNTIME_CONFIG = {
         "Handler": "com.newrelic.java.HandlerWrapper::",
         "LambdaExtension": True,
     },
-    "nodejs12.x": {
-        "Handler": "newrelic-lambda-wrapper.handler",
-        "LambdaExtension": True,
-    },
-    "nodejs14.x": {
+    "nodejs16.x": {
         "Handler": "newrelic-lambda-wrapper.handler",
         "LambdaExtension": True,
     },
-    "nodejs16.x": {
+    "nodejs18.x": {
         "Handler": "newrelic-lambda-wrapper.handler",
         "LambdaExtension": True,
     },
-    "nodejs18.x": {
+    "nodejs20.x": {
         "Handler": "newrelic-lambda-wrapper.handler",
         "LambdaExtension": True,
     },
     "provided": {"LambdaExtension": True},
     "provided.al2": {"LambdaExtension": True},
-    "python3.6": {
-        "Handler": "newrelic_lambda_wrapper.handler",
-        "LambdaExtension": False,
-    },
     "python3.7": {
         "Handler": "newrelic_lambda_wrapper.handler",
         "LambdaExtension": True,
@@ -64,6 +56,10 @@ RUNTIME_CONFIG = {
         "Handler": "newrelic_lambda_wrapper.handler",
         "LambdaExtension": True,
     },
+    "python3.12": {
+        "Handler": "newrelic_lambda_wrapper.handler",
+        "LambdaExtension": True,
+    },
 }
 
 

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/newrelic_lambda_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: newrelic-lambda-cli
-Version: 0.7.6
+Version: 0.9.0
 Summary: A CLI to install the New Relic AWS Lambda integration and layers.
 Home-page: https://github.com/newrelic/newrelic-lambda-cli
 Author: New Relic
@@ -55,9 +55,9 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * java8.al2
 * java11
 * java17
-* nodejs12.x
-* nodejs14.x
 * nodejs16.x
+* nodejs18.x
+* nodejs20.x
 * provided
 * provided.al2
 * python3.7
@@ -65,13 +65,14 @@ A CLI to install the New Relic AWS Lambda integration and layers.
 * python3.9
 * python3.10
 * python3.11
+* python3.12
 
 **Note:** Automatic handler wrapping is only supported for Node.js, Python and Java. For other runtimes,
 manual function wrapping is required using the runtime specific New Relic agent.
 
 ## Requirements
 
-* Python >= 3.3 <= 3.11
+* Python >= 3.3 <= 3.12
 * Retrieve your [New relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) and [User API Key](https://docs.newrelic.com/docs/apis/get-started/intro-apis/types-new-relic-api-keys#user-api-key)
 
 ## Recommendations
@@ -131,6 +132,7 @@ newrelic-lambda integrations install \
 | `--aws-role-policy` | No | Specify an alternative IAM role policy ARN for this integration. |
 | `--disable-license-key-secret` | No | Don't create a managed secret for your account's New Relic License Key |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Uninstall Integration
 
@@ -144,6 +146,7 @@ newrelic-lambda integrations uninstall
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--force` or `-f` | No | Forces uninstall non-interactively |
 | `--nr-account-id` or `-a` | No | The [New Relic Account ID](https://docs.newrelic.com/docs/accounts/install-new-relic/account-setup/account-id) for the integration. Only required if also uninstalling the New Relic AWS Lambda integration. Can also use the `NEW_RELIC_ACCOUNT_ID` environment variable. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 #### Update Integration
 
@@ -171,6 +174,7 @@ newrelic-lambda integrations update \
 | `--aws-region` or `-r` | No | The AWS region for the integration. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |
 | `--disable-license-key-secret` | No | Disable automatic creation of the license key secret on update. The secret is not created if it exists. |
 | `--tag <key> <value>` | No | Sets tags on the CloudFormation Stacks this CLI creates. Can be used multiple times, example: `--tag key1 value1 --tag key2 value2`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 
 ### AWS Lambda Layers
 
@@ -243,6 +247,7 @@ newrelic-lambda subscriptions install --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to add a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while installing subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--filter-pattern` | No | Specify a custom log subscription filter pattern. To collect all logs use `--filter-pattern ""`. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
@@ -257,6 +262,7 @@ newrelic-lambda subscriptions uninstall --function <name or arn>
 | Option | Required? | Description |
 |--------|-----------|-------------|
 | `--function` or `-f` | Yes | The AWS Lambda function name or ARN in which to remove a log subscription. Can provide multiple `--function` arguments. Will also accept `all`, `installed` and `not-installed` similar to `newrelic-lambda functions list`. |
+| `--stackname` | No | The AWS Cloudformation stack name which contains the newrelic-log-ingestion lambda function. If no value is provided, the command searches for the NewRelicLogIngestion stack |
 | `--exclude` or `-e` | No | A function name to exclude while uninstalling subscriptions. Can provide multiple `--exclude` arguments. Only checked when `all`, `installed` and `not-installed` are used. See `newrelic-lambda functions list` for function names. |
 | `--aws-profile` or `-p` | No | The AWS profile to use for this command. Can also use `AWS_PROFILE`. Will also check `AWS_ACCESS_KEY_ID` and `AWS_SECRET_ACCESS_KEY` environment variables if not using AWS CLI. |
 | `--aws-region` or `-r` | No | The AWS region this function is located. Can use `AWS_DEFAULT_REGION` environment variable. Defaults to AWS session region. |

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/setup.py

@@ -6,7 +6,7 @@ README = open(os.path.join(os.path.dirname(__file__), "README.md"), "r").read()
 
 setup(
     name="newrelic-lambda-cli",
-    version="0.7.6",
+    version="0.9.0",
     python_requires=">=3.3",
     description="A CLI to install the New Relic AWS Lambda integration and layers.",
     long_description=README,

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/tests/test_integrations.py

@@ -43,7 +43,7 @@ def test__check_for_ingest_stack_none_when_not_found():
         )
     }
     session = MagicMock(**describe_stack_mock)
-    assert _check_for_ingest_stack(session) is None
+    assert _check_for_ingest_stack(session, "test_stack_name") is None
 
 
 def test__check_for_ingest_stack_status_when_found():
@@ -56,7 +56,7 @@ def test__check_for_ingest_stack_status_when_found():
         }
     }
     session = MagicMock(**describe_stack_mock)
-    assert _check_for_ingest_stack(session) == "CREATE_COMPLETE"
+    assert _check_for_ingest_stack(session, "test_stack_name") == "CREATE_COMPLETE"
 
 
 @patch("newrelic_lambda_cli.integrations.success")
@@ -70,7 +70,11 @@ def test__create_log_ingestion_function__defaults(success_mock):
 
         _create_log_ingestion_function(
             integration_install(
-                session=session, enable_logs=False, memory_size=128, timeout=30
+                session=session,
+                enable_logs=False,
+                memory_size=128,
+                timeout=30,
+                stackname="NewRelicLogIngestion",
             ),
             "test_key",
         )
@@ -115,6 +119,7 @@ def test__create_log_ingestion_function__opts(success_mock):
                 memory_size=256,
                 role_name="CustomExecRole",
                 timeout=60,
+                stackname="NewRelicLogIngestion",
             ),
             "test_key",
         )
@@ -156,8 +161,8 @@ def test_remove_log_ingestion_function(success_mock):
     session.assert_has_calls(
         [
             call.client("cloudformation"),
-            call.client().describe_stacks(StackName="NewRelicLogIngestion"),
-            call.client().delete_stack(StackName="NewRelicLogIngestion"),
+            call.client().describe_stacks(StackName=None),
+            call.client().delete_stack(StackName=None),
         ],
         any_order=True,
     )
@@ -178,7 +183,7 @@ def test_remove_log_ingestion_function_not_present(success_mock):
     session.assert_has_calls(
         [
             call.client("cloudformation"),
-            call.client().describe_stacks(StackName="NewRelicLogIngestion"),
+            call.client().describe_stacks(StackName=None),
         ],
         any_order=True,
     )
@@ -406,7 +411,7 @@ def test_remove_integration_role(aws_credentials):
 def test_install_log_ingestion(aws_credentials, mock_function_config):
     mock_session = MagicMock()
     mock_client = mock_session.client.return_value
-    mock_client.get_function.return_value = mock_function_config("python3.6")
+    mock_client.get_function.return_value = mock_function_config("python3.7")
 
     assert (
         install_log_ingestion(
@@ -418,7 +423,15 @@ def test_install_log_ingestion(aws_credentials, mock_function_config):
 
     mock_client.get_function.reset_mock(return_value=True)
     mock_client.get_function.return_value = None
-    mock_client.describe_stacks.return_value = {"Stacks": [{"StackStatus": "peachy"}]}
+    mock_client.describe_stacks.return_value = {
+        "Stacks": [
+            {
+                "StackId": "arn:aws:cloudformation:us-east-2:363096433263:stack/NewRelicLogIngestion/ec0fa5a0-abdb-11ee-9bc5-0625ef1d38db",
+                "StackName": "NewRelicLogIngestion",
+                "StackStatus": "CREATE_COMPLETE",
+            }
+        ]
+    }
     assert (
         install_log_ingestion(
             integration_install(nr_account_id=123456789, session=mock_session),
@@ -428,7 +441,10 @@ def test_install_log_ingestion(aws_credentials, mock_function_config):
     )
 
     mock_client.describe_stacks.reset_mock(return_value=True)
-    mock_client.describe_stacks.return_value = {"Stacks": [{"StackStatus": None}]}
+    mock_client.describe_stacks.return_value = {
+        "Stacks": [{"StackId": None, "StackName": None, "StackStatus": None}]
+    }
+    mock_client._check_for_ingest_stack.return_value = None
     assert (
         install_log_ingestion(
             integration_install(nr_account_id=123456789, session=mock_session),
@@ -451,7 +467,7 @@ def test_update_log_ingestion(aws_credentials, mock_function_config):
     )
 
     mock_client.get_function.reset_mock(return_value=True)
-    mock_client.get_function.return_value = mock_function_config("python3.6")
+    mock_client.get_function.return_value = mock_function_config("python3.7")
     mock_client.describe_stacks.return_value = {"Stacks": [{"StackStatus": None}]}
 
     assert (
@@ -462,7 +478,15 @@ def test_update_log_ingestion(aws_credentials, mock_function_config):
     )
 
     mock_client.describe_stacks.reset_mock(return_value=True)
-    mock_client.describe_stacks.return_value = {"Stacks": [{"StackStatus": "peachy"}]}
+    mock_client.describe_stacks.return_value = {
+        "Stacks": [
+            {
+                "StackId": "arn:aws:cloudformation:us-east-2:363096433263:stack/NewRelicLogIngestion/ec0fa5a0-abdb-11ee-9bc5-0625ef1d38db",
+                "StackName": "NewRelicLogIngestion",
+                "StackStatus": "CREATE_COMPLETE",
+            }
+        ]
+    }
 
     assert (
         update_log_ingestion(

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/tests/test_layers.py

@@ -24,7 +24,7 @@ from .conftest import layer_install, layer_uninstall
 def test_add_new_relic(aws_credentials, mock_function_config):
     session = boto3.Session(region_name="us-east-1")
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
 
     assert config["Configuration"]["Handler"] == "original_handler"
 
@@ -71,10 +71,10 @@ def test_add_new_relic(aws_credentials, mock_function_config):
             config,
             nr_license_key=None,
         )
-        is True
+        is False
     )
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
     config["Configuration"]["Layers"] = [{"Arn": get_arn_prefix("us-east-1")}]
     assert (
         _add_new_relic(
@@ -93,7 +93,7 @@ def test_add_new_relic(aws_credentials, mock_function_config):
 
     with patch("newrelic_lambda_cli.layers.index") as mock_index:
         mock_index.return_value = []
-        config = mock_function_config("python3.11")
+        config = mock_function_config("python3.12")
         assert (
             _add_new_relic(
                 layer_install(
@@ -190,15 +190,15 @@ def test_add_new_relic(aws_credentials, mock_function_config):
     ) as mock_isatty:
         mock_isatty.return_value = True
         mock_prompt.return_value = 0
-        result = layer_selection(mock_layers, "python3.11", "x86_64")
+        result = layer_selection(mock_layers, "python3.12", "x86_64")
         assert result == "arn:aws:lambda:us-east-1:123456789:layer/javajava"
 
     with patch("sys.stdout.isatty") as mock_isatty:
         mock_isatty.return_value = False
         with pytest.raises(UsageError):
-            layer_selection(mock_layers, "python3.11", "x86_64")
+            layer_selection(mock_layers, "python3.12", "x86_64")
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
     _add_new_relic(
         layer_install(
             session=session,
@@ -216,7 +216,7 @@ def test_add_new_relic(aws_credentials, mock_function_config):
         in config["Configuration"]["Environment"]["Variables"]
     )
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
     config["Configuration"]["Environment"]["Variables"]["NEW_RELIC_FOO"] = "bar"
     config["Configuration"]["Layers"] = [{"Arn": get_arn_prefix("us-east-1")}]
     update_kwargs = _add_new_relic(
@@ -234,30 +234,12 @@ def test_add_new_relic(aws_credentials, mock_function_config):
     assert "NEW_RELIC_FOO" in update_kwargs["Environment"]["Variables"]
     assert update_kwargs["Layers"][0] != get_arn_prefix("us-east-1")
 
-    config = mock_function_config("python3.6")
-    update_kwargs = _add_new_relic(
-        layer_install(
-            session=session,
-            aws_region="us-east-1",
-            nr_account_id=12345,
-            enable_extension=True,
-            enable_extension_function_logs=True,
-            upgrade=True,
-        ),
-        config,
-        "foobarbaz",
-    )
-    assert (
-        update_kwargs["Environment"]["Variables"]["NEW_RELIC_LAMBDA_EXTENSION_ENABLED"]
-        == "false"
-    )
-
 
 @mock_lambda
 def test_remove_new_relic(aws_credentials, mock_function_config):
     session = boto3.Session(region_name="us-east-1")
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
     config["Configuration"]["Handler"] = "newrelic_lambda_wrapper.handler"
     config["Configuration"]["Environment"]["Variables"][
         "NEW_RELIC_LAMBDA_HANDLER"
@@ -281,7 +263,7 @@ def test_remove_new_relic(aws_credentials, mock_function_config):
         is True
     )
 
-    config = mock_function_config("python3.11")
+    config = mock_function_config("python3.12")
     config["Configuration"]["Handler"] = "what is this?"
     assert (
         _remove_new_relic(
@@ -352,7 +334,7 @@ def test_install_failure(aws_credentials, mock_function_config):
         "newrelic_lambda_cli.layers._get_license_key_outputs"
     ) as mock_get_license_key_outputs:
         mock_get_license_key_outputs.return_value = ("license_arn", "12345", "policy")
-        config = mock_function_config("python3.11")
+        config = mock_function_config("python3.12")
         mock_client.get_function.return_value = config
         with pytest.raises(UsageError):
             install(
@@ -370,7 +352,7 @@ def test_install(aws_credentials, mock_function_config):
         mock_client = mock_session.client.return_value
         mock_client.get_function.reset_mock(return_value=True)
         mock_get_license_key_outputs.return_value = ("license_arn", "12345", "policy")
-        config = mock_function_config("python3.11")
+        config = mock_function_config("python3.12")
         mock_client.get_function.return_value = config
         with pytest.raises(UsageError):
             install(
@@ -393,11 +375,11 @@ def test_install(aws_credentials, mock_function_config):
             install(
                 layer_install(nr_account_id=12345, session=mock_session), "foobarbaz"
             )
-            is True
+            is False
         )
 
         mock_client.get_function.reset_mock(return_value=True)
-        config = mock_function_config("python3.11")
+        config = mock_function_config("python3.12")
         mock_client.get_function.return_value = config
         assert (
             install(
@@ -443,7 +425,7 @@ def test_uninstall(aws_credentials, mock_function_config):
         assert uninstall(layer_uninstall(session=mock_session), "foobarbaz") is True
 
         mock_client.get_function.reset_mock(return_value=True)
-        config = mock_function_config("python3.11")
+        config = mock_function_config("python3.12")
         mock_client.get_function.return_value = config
         assert uninstall(layer_uninstall(session=mock_session), "foobarbaz") is False
 
@@ -476,6 +458,6 @@ def test_uninstall(aws_credentials, mock_function_config):
 
 
 def test_layers_index():
-    layers = index("ap-southeast-1", "nodejs14.x", "x86_64")
+    layers = index("ap-southeast-1", "nodejs20.x", "x86_64")
 
     assert len(layers) == 1

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/tests/test_subscriptions.py

@@ -21,13 +21,24 @@ def test__get_log_group_name():
 @patch("newrelic_lambda_cli.subscriptions._create_subscription_filter", autospec=True)
 @patch("newrelic_lambda_cli.subscriptions._get_subscription_filters", autospec=True)
 @patch("newrelic_lambda_cli.subscriptions._remove_subscription_filter", autospec=True)
+@patch("newrelic_lambda_cli.integrations.get_function", autospec=True)
 @patch("newrelic_lambda_cli.subscriptions.get_function", autospec=True)
+@patch(
+    "newrelic_lambda_cli.integrations.get_unique_newrelic_log_ingestion_name",
+    autospec=True,
+)
 def test_create_log_subscription(
+    mock_get_unique_newrelic_log_ingestion_name,
+    mock_subscriptions_get_function,
     mock_get_function,
     mock_remove_subscription_filter,
     mock_get_subscription_filters,
     mock_create_subscription_filter,
 ):
+    mock_get_unique_newrelic_log_ingestion_name.return_value = "newrelic-log-ingestion"
+
+    mock_subscriptions_get_function.side_effect = None
+
     mock_get_function.side_effect = (
         None,
         {"Configuration": {"FunctionArn": "FooBarBaz"}},

{newrelic-lambda-cli-0.7.6 → newrelic-lambda-cli-0.9.0}/tests/test_utils.py

@@ -80,10 +80,9 @@ def test_supports_lambda_extension():
             "java17",
             "java11",
             "java8.al2",
-            "nodejs12.x",
-            "nodejs14.x",
             "nodejs16.x",
             "nodejs18.x",
+            "nodejs20.x",
             "provided",
             "provided.al2",
             "python3.7",
@@ -91,6 +90,7 @@ def test_supports_lambda_extension():
             "python3.9",
             "python3.10",
             "python3.11",
+            "python3.12",
         )
     )
     assert not any(