fix: state machine integ tests fail when workflow names already exist (#148)

Concurrent test runs in the same account would fail because `test_state_machine_definition.py` uses the same hardcoded state machine name in the test cases. This fixes it by generating a unique name using [`unique_name_from_base`](https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/utils.py#L79) from the sagemaker SDK which we are using the the sagemaker integ tests.

Most of the lines changed in this PR are trailing whitespace removals.

Author: wong-a

tests/integ/test_state_machine_definition.py

@@ -6,17 +6,17 @@
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
-# or in the "license" file accompanying this file. This file is distributed 
-# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
-# express or implied. See the License for the specific language governing 
+# or in the "license" file accompanying this file. This file is distributed
+# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+# express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 from __future__ import absolute_import
 
 import pytest
 import json
 
 from sagemaker.utils import unique_name_from_base
-from sagemaker.image_uris import retrieve 
+from sagemaker.image_uris import retrieve
 from stepfunctions import steps
 from stepfunctions.workflow import Workflow
 from stepfunctions.steps.utils import get_aws_partition
@@ -25,27 +25,27 @@
 
 @pytest.fixture(scope="module")
 def training_job_parameters(sagemaker_session, sagemaker_role_arn, record_set_fixture):
-    parameters = { 
-        "AlgorithmSpecification": { 
+    parameters = {
+        "AlgorithmSpecification": {
             "TrainingImage": retrieve(region=sagemaker_session.boto_session.region_name, framework='pca'),
             "TrainingInputMode": "File"
         },
-        "OutputDataConfig": { 
+        "OutputDataConfig": {
             "S3OutputPath": "s3://{}/".format(sagemaker_session.default_bucket())
         },
-        "StoppingCondition": { 
+        "StoppingCondition": {
             "MaxRuntimeInSeconds": 86400
         },
-        "ResourceConfig": { 
+        "ResourceConfig": {
             "InstanceCount": 1,
             "InstanceType": "ml.m5.large",
             "VolumeSizeInGB": 30
         },
         "RoleArn": sagemaker_role_arn,
-        "InputDataConfig":[ 
-        { 
-            "DataSource": { 
-                "S3DataSource": { 
+        "InputDataConfig":[
+        {
+            "DataSource": {
+                "S3DataSource": {
                     "S3DataDistributionType": "ShardedByS3Key",
                     "S3DataType": "ManifestFile",
                     "S3Uri": record_set_fixture.s3_data
@@ -54,7 +54,7 @@ def training_job_parameters(sagemaker_session, sagemaker_role_arn, record_set_fi
             "ChannelName": "train"
         }
         ],
-        "HyperParameters": { 
+        "HyperParameters": {
             "num_components": "48",
             "feature_dim": "784",
             "mini_batch_size": "200"
@@ -93,7 +93,7 @@ def test_pass_state_machine_creation(sfn_client, sfn_role_arn):
 
     definition = steps.Pass(pass_state_name, result=pass_state_result)
     workflow = Workflow(
-        'Test_Pass_Workflow',
+        unique_name_from_base('Test_Pass_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -164,7 +164,7 @@ def test_wait_state_machine_creation(sfn_client, sfn_role_arn):
     ])
 
     workflow = Workflow(
-        'Test_Wait_Workflow',
+        unique_name_from_base('Test_Wait_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -223,7 +223,7 @@ def test_parallel_state_machine_creation(sfn_client, sfn_role_arn):
     ])
 
     workflow = Workflow(
-        'Test_Parallel_Workflow',
+        unique_name_from_base('Test_Parallel_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -269,9 +269,9 @@ def test_map_state_machine_creation(sfn_client, sfn_role_arn):
     }
 
     map_state = steps.Map(
-        map_state_name, 
+        map_state_name,
         items_path=items_path,
-        iterator=steps.Pass(iterated_state_name), 
+        iterator=steps.Pass(iterated_state_name),
         max_concurrency=max_concurrency)
 
     definition = steps.Chain([
@@ -280,7 +280,7 @@ def test_map_state_machine_creation(sfn_client, sfn_role_arn):
     ])
 
     workflow = Workflow(
-        'Test_Map_Workflow',
+        unique_name_from_base('Test_Map_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -345,8 +345,8 @@ def test_choice_state_machine_creation(sfn_client, sfn_role_arn):
 
     definition.default_choice(
         steps.Fail(
-            default_state_name, 
-            error=default_error, 
+            default_state_name,
+            error=default_error,
             cause=default_cause
         )
     )
@@ -356,23 +356,23 @@ def test_choice_state_machine_creation(sfn_client, sfn_role_arn):
             value=first_choice_value
         ),
         steps.Pass(
-            first_match_name, 
+            first_match_name,
             result=first_choice_state_result
         )
     )
     definition.add_choice(
         steps.ChoiceRule.NumericEquals(
-            variable=variable, 
+            variable=variable,
             value=second_choice_value
-        ), 
+        ),
         steps.Pass(
-            second_match_name, 
+            second_match_name,
             result=second_choice_state_result
         )
     )
 
     workflow = Workflow(
-        'Test_Choice_Workflow',
+        unique_name_from_base('Test_Choice_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -385,10 +385,10 @@ def test_task_state_machine_creation(sfn_client, sfn_role_arn, training_job_para
     final_state_name = "FinalState"
     resource = f"arn:{get_aws_partition()}:states:::sagemaker:createTrainingJob.sync"
     task_state_result = "Task State Result"
-    asl_state_machine_definition = { 
+    asl_state_machine_definition = {
         "StartAt": task_state_name,
-        "States": { 
-            task_state_name: { 
+        "States": {
+            task_state_name: {
                 "Resource": resource,
                 "Parameters": training_job_parameters,
                 "Type": "Task",
@@ -410,9 +410,9 @@ def test_task_state_machine_creation(sfn_client, sfn_role_arn, training_job_para
         ),
         steps.Pass(final_state_name, result=task_state_result)
     ])
-    
+
     workflow = Workflow(
-        'Test_Task_Workflow',
+        unique_name_from_base('Test_Task_Workflow'),
         definition=definition,
         role=sfn_role_arn
     )
@@ -465,13 +465,13 @@ def test_catch_state_machine_creation(sfn_client, sfn_role_arn, training_job_par
     )
     task.add_catch(
         steps.Catch(
-            error_equals=[all_fail_error], 
+            error_equals=[all_fail_error],
             next_step=steps.Pass(all_error_state_name, result=catch_state_result)
         )
     )
 
     workflow = Workflow(
-        'Test_Catch_Workflow',
+        unique_name_from_base('Test_Catch_Workflow'),
         definition=task,
         role=sfn_role_arn
     )
@@ -518,15 +518,15 @@ def test_retry_state_machine_creation(sfn_client, sfn_role_arn, training_job_par
 
     task.add_retry(
         steps.Retry(
-            error_equals=[all_fail_error], 
-            interval_seconds=interval_seconds, 
-            max_attempts=max_attempts, 
+            error_equals=[all_fail_error],
+            interval_seconds=interval_seconds,
+            max_attempts=max_attempts,
             backoff_rate=backoff_rate
         )
     )
 
     workflow = Workflow(
-        'Test_Retry_Workflow',
+        unique_name_from_base('Test_Retry_Workflow'),
         definition=task,
         role=sfn_role_arn
     )