chore: black formatted

Author: Sashwatdas123

tests/cli/test_layers.py

@@ -193,7 +193,10 @@ def test_layers_install_validation_no_keys(aws_credentials, cli_runner):
     )
 
     assert result.exit_code != 0
-    assert "Please provide either the --nr-api-key or the --nr-ingest-key flag" in result.stderr
+    assert (
+        "Please provide either the --nr-api-key or the --nr-ingest-key flag"
+        in result.stderr
+    )
 
 
 @mock_aws
@@ -229,7 +232,10 @@ def test_layers_install_validation_both_keys(aws_credentials, cli_runner):
     )
 
     assert result.exit_code != 0
-    assert "Please provide either the --nr-api-key or the --nr-ingest-key flag, but not both" in result.stderr
+    assert (
+        "Please provide either the --nr-api-key or the --nr-ingest-key flag, but not both"
+        in result.stderr
+    )
 
 
 @mock_aws
@@ -240,12 +246,13 @@ def test_layers_install_with_verbose_output(aws_credentials, cli_runner):
     register_groups(cli)
 
     # Mock a successful install to trigger verbose output
-    with patch("newrelic_lambda_cli.layers.install") as mock_install, \
-         patch("newrelic_lambda_cli.functions.get_aliased_functions") as mock_get_functions:
-        
+    with patch("newrelic_lambda_cli.layers.install") as mock_install, patch(
+        "newrelic_lambda_cli.functions.get_aliased_functions"
+    ) as mock_get_functions:
+
         mock_get_functions.return_value = ["test-function"]
         mock_install.return_value = True
-        
+
         result = cli_runner.invoke(
             cli,
             [

tests/test_layers.py

@@ -1183,12 +1183,20 @@ def test_add_new_relic_with_ingest_key(aws_credentials, mock_function_config):
     )
 
     # Verify ingest key is used for NEW_RELIC_LICENSE_KEY
-    assert update_kwargs["Environment"]["Variables"]["NEW_RELIC_LICENSE_KEY"] == "test-ingest-key-12345"
-    assert update_kwargs["Environment"]["Variables"]["NEW_RELIC_LAMBDA_EXTENSION_ENABLED"] == "true"
+    assert (
+        update_kwargs["Environment"]["Variables"]["NEW_RELIC_LICENSE_KEY"]
+        == "test-ingest-key-12345"
+    )
+    assert (
+        update_kwargs["Environment"]["Variables"]["NEW_RELIC_LAMBDA_EXTENSION_ENABLED"]
+        == "true"
+    )
 
 
 @mock_aws
-def test_add_new_relic_without_license_key_or_ingest_key(aws_credentials, mock_function_config):
+def test_add_new_relic_without_license_key_or_ingest_key(
+    aws_credentials, mock_function_config
+):
     """Test _add_new_relic function when neither license key nor ingest key is provided"""
     session = boto3.Session(region_name="us-east-1")
 
@@ -1206,11 +1214,14 @@ def test_add_new_relic_without_license_key_or_ingest_key(aws_credentials, mock_f
     )
 
     # When no license key or ingest key is provided, extension should be disabled
-    assert update_kwargs["Environment"]["Variables"]["NEW_RELIC_LAMBDA_EXTENSION_ENABLED"] == "false"
+    assert (
+        update_kwargs["Environment"]["Variables"]["NEW_RELIC_LAMBDA_EXTENSION_ENABLED"]
+        == "false"
+    )
     assert "NEW_RELIC_LICENSE_KEY" not in update_kwargs["Environment"]["Variables"]
 
 
-@mock_aws 
+@mock_aws
 def test_add_new_relic_staging_region(aws_credentials, mock_function_config):
     """Test _add_new_relic function with staging region"""
     session = boto3.Session(region_name="us-east-1")
@@ -1231,24 +1242,35 @@ def test_add_new_relic_staging_region(aws_credentials, mock_function_config):
     )
 
     # Verify staging endpoints are set
-    assert update_kwargs["Environment"]["Variables"]["NEW_RELIC_TELEMETRY_ENDPOINT"] == "https://staging-cloud-collector.newrelic.com/aws/lambda/v1"
-    assert update_kwargs["Environment"]["Variables"]["NEW_RELIC_LOG_ENDPOINT"] == "https://staging-log-api.newrelic.com/log/v1"
+    assert (
+        update_kwargs["Environment"]["Variables"]["NEW_RELIC_TELEMETRY_ENDPOINT"]
+        == "https://staging-cloud-collector.newrelic.com/aws/lambda/v1"
+    )
+    assert (
+        update_kwargs["Environment"]["Variables"]["NEW_RELIC_LOG_ENDPOINT"]
+        == "https://staging-log-api.newrelic.com/log/v1"
+    )
 
 
-def test_install_with_both_api_key_and_ingest_key(aws_credentials, mock_function_config):
+def test_install_with_both_api_key_and_ingest_key(
+    aws_credentials, mock_function_config
+):
     """Test install function fails when both API key and ingest key are provided"""
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
-    
-    with pytest.raises(UsageError, match="Please provide either the --nr-api-key or the --nr-ingest-key flag, but not both"):
+
+    with pytest.raises(
+        UsageError,
+        match="Please provide either the --nr-api-key or the --nr-ingest-key flag, but not both",
+    ):
         install(
             layer_install(
                 session=mock_session,
                 nr_account_id=12345,
                 nr_api_key="test-api-key",
-                nr_ingest_key="test-ingest-key"
+                nr_ingest_key="test-ingest-key",
             ),
-            "test-function"
+            "test-function",
         )
 
 
@@ -1257,36 +1279,40 @@ def test_install_with_ingest_key_success(aws_credentials, mock_function_config):
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
     mock_client = mock_session.client.return_value
-    
-    with patch("newrelic_lambda_cli.layers._get_license_key_outputs") as mock_get_license_key_outputs, \
-         patch("newrelic_lambda_cli.layers.get_function") as mock_get_function, \
-         patch("newrelic_lambda_cli.layers._add_new_relic") as mock_add_new_relic:
-        
+
+    with patch(
+        "newrelic_lambda_cli.layers._get_license_key_outputs"
+    ) as mock_get_license_key_outputs, patch(
+        "newrelic_lambda_cli.layers.get_function"
+    ) as mock_get_function, patch(
+        "newrelic_lambda_cli.layers._add_new_relic"
+    ) as mock_add_new_relic:
+
         # Set up policy_arn so GraphQL validation is skipped
         mock_get_license_key_outputs.return_value = (None, "12345", "test-policy-arn")
         mock_get_function.return_value = mock_function_config("python3.12")
         mock_add_new_relic.return_value = {
             "FunctionName": "test-function",
             "Environment": {"Variables": {"NEW_RELIC_LICENSE_KEY": "test-ingest-key"}},
-            "Layers": ["test-layer"]
+            "Layers": ["test-layer"],
         }
         mock_client.update_function_configuration.return_value = {
             "Configuration": {
                 "Layers": [{"Arn": "existing-layer"}],
-                "FunctionArn": "test-function"
+                "FunctionArn": "test-function",
             }
         }
-        
+
         result = install(
             layer_install(
                 session=mock_session,
                 nr_account_id=12345,
                 nr_ingest_key="test-ingest-key",
                 enable_extension=False,
             ),
-            "test-function"
+            "test-function",
         )
-        
+
         assert result is True
         mock_client.update_function_configuration.assert_called_once()
 
@@ -1297,26 +1323,27 @@ def test_install_with_apm_and_verbose(aws_credentials, mock_function_config):
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
     mock_client = mock_session.client.return_value
-    
-    with patch("newrelic_lambda_cli.layers._get_license_key_outputs") as mock_get_license_key_outputs, \
-         patch("newrelic_lambda_cli.layers.get_function") as mock_get_function, \
-         patch("newrelic_lambda_cli.layers._add_new_relic") as mock_add_new_relic:
-        
+
+    with patch(
+        "newrelic_lambda_cli.layers._get_license_key_outputs"
+    ) as mock_get_license_key_outputs, patch(
+        "newrelic_lambda_cli.layers.get_function"
+    ) as mock_get_function, patch(
+        "newrelic_lambda_cli.layers._add_new_relic"
+    ) as mock_add_new_relic:
+
         mock_get_license_key_outputs.return_value = (None, "12345", "test-policy-arn")
         mock_get_function.return_value = mock_function_config("python3.12")
         mock_add_new_relic.return_value = {
             "FunctionName": "test-function",
             "Environment": {"Variables": {"NEW_RELIC_LICENSE_KEY": "test-key"}},
-            "Layers": ["test-layer"]
+            "Layers": ["test-layer"],
         }
         mock_client.update_function_configuration.return_value = {
-            "Configuration": {
-                "Layers": [],
-                "FunctionArn": "test-function"
-            }
+            "Configuration": {"Layers": [], "FunctionArn": "test-function"}
         }
         mock_client.tag_resource.return_value = {}
-        
+
         result = install(
             layer_install(
                 session=mock_session,
@@ -1327,9 +1354,9 @@ def test_install_with_apm_and_verbose(aws_credentials, mock_function_config):
                 verbose=True,
                 enable_extension=False,
             ),
-            "test-function"
+            "test-function",
         )
-        
+
         assert result is True
         mock_client.tag_resource.assert_called_once()
 
@@ -1339,56 +1366,62 @@ def test_install_function_not_found(aws_credentials):
     """Test install function when function is not found"""
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
-    
+
     with patch("newrelic_lambda_cli.layers.get_function") as mock_get_function:
         mock_get_function.return_value = None
-        
+
         result = install(
             layer_install(
                 session=mock_session,
                 nr_account_id=12345,
                 nr_api_key="test-api-key",
             ),
-            "nonexistent-function"
+            "nonexistent-function",
         )
-        
+
         assert result is False
 
 
 def test_install_secret_account_mismatch(aws_credentials):
     """Test install with managed secret account ID mismatch"""
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
-    
-    with patch("newrelic_lambda_cli.layers._get_license_key_outputs") as mock_get_license_key_outputs, \
-         patch("newrelic_lambda_cli.layers.get_function") as mock_get_function:
-        
+
+    with patch(
+        "newrelic_lambda_cli.layers._get_license_key_outputs"
+    ) as mock_get_license_key_outputs, patch(
+        "newrelic_lambda_cli.layers.get_function"
+    ) as mock_get_function:
+
         # Different account ID in existing secret
         mock_get_license_key_outputs.return_value = (None, "99999", "test-policy-arn")
         mock_get_function.return_value = {"Configuration": {"FunctionArn": "test"}}
-        
+
         with pytest.raises(UsageError, match="A managed secret already exists"):
             install(
                 layer_install(
                     session=mock_session,
                     nr_account_id=12345,  # Different from secret account ID
                     nr_ingest_key="test-ingest-key",
                 ),
-                "test-function"
+                "test-function",
             )
 
 
 def test_install_extension_without_secret_or_api_key(aws_credentials):
     """Test install with extension enabled but no secret or API key"""
     mock_session = MagicMock()
     mock_session.region_name = "us-east-1"
-    
-    with patch("newrelic_lambda_cli.layers._get_license_key_outputs") as mock_get_license_key_outputs, \
-         patch("newrelic_lambda_cli.layers.get_function") as mock_get_function:
-        
+
+    with patch(
+        "newrelic_lambda_cli.layers._get_license_key_outputs"
+    ) as mock_get_license_key_outputs, patch(
+        "newrelic_lambda_cli.layers.get_function"
+    ) as mock_get_function:
+
         mock_get_license_key_outputs.return_value = (None, None, None)  # No secret
         mock_get_function.return_value = {"Configuration": {"FunctionArn": "test"}}
-        
+
         with pytest.raises(UsageError, match="In order to use `--enable-extension`"):
             install(
                 layer_install(
@@ -1397,5 +1430,5 @@ def test_install_extension_without_secret_or_api_key(aws_credentials):
                     enable_extension=True,
                     nr_ingest_key="test-ingest-key",
                 ),
-                "test-function"
+                "test-function",
             )