update 9

Author: Dan Xie

src/lambda_codebase/initial_commit/bootstrap_repository/adf-build/china-support/china-forward-function/tests/test_handler.py

@@ -134,4 +134,4 @@ def test_sfn_name_extraction(aws_settings):
     sfn_arn = f"arn:{partition}:states:{region}:123456789012:stateMachine:{sfn_name}"
     with patch.dict(os.environ, {"SFN_ARN": sfn_arn}):
         importlib.reload(handler)
-        assert handler.sfn_name == sfn_name
+        assert handler.sfn_name == sfn_name

src/lambda_codebase/initial_commit/bootstrap_repository/adf-build/china-support/china-forward-function/tests/test_stepfunction_helper.py

@@ -1,3 +1,6 @@
+"""
+test for stepfunction
+"""
 import json
 from decimal import Decimal
 from unittest.mock import MagicMock, patch
@@ -70,8 +73,12 @@ def test_invoke_sfn_execution_with_default_name(self, stepfunction, mock_session
         region = aws_settings["region"]
         partition = aws_settings["partition"]
         _, mock_client = mock_session
+        execution_arn = (
+            f"arn:{partition}:states:{region}:"
+            f"123456789012:execution:test-sfn:test-execution"
+        )
         mock_response = {
-            "executionArn": f"arn:{partition}:states:{region}:123456789012:execution:test-sfn:test-execution"
+            "executionArn": execution_arn
         }
         mock_client.start_execution.return_value = mock_response
 
@@ -89,7 +96,12 @@ def test_invoke_sfn_execution_with_custom_name(self, stepfunction, mock_session,
         region = aws_settings["region"]
         partition = aws_settings["partition"]
         _, mock_client = mock_session
-        mock_response = {"executionArn": f"arn:{partition}:states:{region}:123456789012:execution:test-sfn:custom-name"}
+        mock_response = {
+            "executionArn": (
+                f"arn:{partition}:states:{region}:"
+                f"123456789012:execution:test-sfn:custom-name"
+            )
+        }
         mock_client.start_execution.return_value = mock_response
 
         response, name = stepfunction.invoke_sfn_execution(
@@ -107,7 +119,10 @@ def test_invoke_sfn_execution_with_decimal_data(self, stepfunction, mock_session
         partition = aws_settings["partition"]
         _, mock_client = mock_session
         mock_response = {
-            "executionArn": f"arn:{partition}:states:{region}:123456789012:execution:test-sfn:test-execution"
+            "executionArn": (
+                f"arn:{partition}:states:{region}:"
+                f"123456789012:execution:test-sfn:test-execution"
+            )
         }
         mock_client.start_execution.return_value = mock_response
 
@@ -139,4 +154,4 @@ def test_invoke_sfn_execution_exception(self, stepfunction, mock_session, aws_se
             )
 
         assert "Test exception" in str(excinfo.value)
-        stepfunction.logger.error.assert_called_once()
+        stepfunction.logger.error.assert_called_once()