change unit tests

aws · Dec 30, 2023 · 45b4a1b · 45b4a1b
1 parent f3840cc
commit 45b4a1b
Showing 1 changed file with 67 additions and 1 deletion.
diff --git a/packages/aws-cdk-lib/aws-stepfunctions-tasks/test/emr/emr-create-cluster.test.ts b/packages/aws-cdk-lib/aws-stepfunctions-tasks/test/emr/emr-create-cluster.test.ts
@@ -1331,7 +1331,7 @@ test('Throws if timeoutDurationMinutes for Spot instances is greater than 1440 m
   }).toThrow(/timeoutDurationMinutes must be between 5 and 1440 minutes, got 1441/);
 });
 
-test('Throws if both timeout and timeoutDurationMinutes are not specified', () => {
+test('Throws if neither timeout nor timeoutDurationMinutes is specified', () => {
   // GIVEN
   const task = new EmrCreateCluster(stack, 'Task', {
     instances: {
@@ -1358,6 +1358,72 @@ test('Throws if both timeout and timeoutDurationMinutes are not specified', () =
   }).toThrow(/timeout must be specified/);
 });
 
+test('timeout takes precedence if both timeout and timeoutDurationMinutes are specified', () => {
+  // WHEN
+  const task = new EmrCreateCluster(stack, 'Task', {
+    instances: {
+      instanceFleets: [{
+        instanceFleetType: EmrCreateCluster.InstanceRoleType.MASTER,
+        launchSpecifications: {
+          spotSpecification: {
+            timeoutAction: EmrCreateCluster.SpotTimeoutAction.TERMINATE_CLUSTER,
+            timeout: cdk.Duration.minutes(5),
+            timeoutDurationMinutes: 10,
+          },
+        },
+        name: 'Main',
+        targetSpotCapacity: 1,
+      }],
+    },
+    clusterRole,
+    name: 'Cluster',
+    serviceRole,
+    integrationPattern: sfn.IntegrationPattern.REQUEST_RESPONSE,
+  });
+
+  // THEN
+  expect(stack.resolve(task.toStateJson())).toEqual({
+    Type: 'Task',
+    Resource: {
+      'Fn::Join': [
+        '',
+        [
+          'arn:',
+          {
+            Ref: 'AWS::Partition',
+          },
+          ':states:::elasticmapreduce:createCluster',
+        ],
+      ],
+    },
+    End: true,
+    Parameters: {
+      Name: 'Cluster',
+      Instances: {
+        KeepJobFlowAliveWhenNoSteps: true,
+        InstanceFleets: [{
+          InstanceFleetType: 'MASTER',
+          LaunchSpecifications: {
+            SpotSpecification: {
+              TimeoutAction: 'TERMINATE_CLUSTER',
+              TimeoutDurationMinutes: 5,
+            },
+          },
+          Name: 'Main',
+          TargetSpotCapacity: 1,
+        }],
+      },
+      VisibleToAllUsers: true,
+      JobFlowRole: {
+        Ref: 'ClusterRoleD9CA7471',
+      },
+      ServiceRole: {
+        Ref: 'ServiceRole4288B192',
+      },
+    },
+  });
+});
+
 test('Throws if both bidPrice and bidPriceAsPercentageOfOnDemandPrice are specified', () => {
   // GIVEN
   const task = new EmrCreateCluster(stack, 'Task', {