fix: regression

packages/@aws-cdk-testing/framework-integ/test/aws-stepfunctions/test/integ.custom-state.js.snapshot/aws-stepfunctions-custom-state-integ.template.json

@@ -20,7 +20,7 @@
   "StateMachine2E01A3A5": {
    "Type": "AWS::StepFunctions::StateMachine",
    "Properties": {
-    "DefinitionString": "{\"StartAt\":\"my custom task\",\"States\":{\"my custom task\":{\"Next\":\"final step\",\"Type\":\"Task\",\"Resource\":\"arn:aws:states:::dynamodb:putItem\",\"Parameters\":{\"TableName\":\"my-cool-table\",\"Item\":{\"id\":{\"S\":\"my-entry\"}}},\"ResultPath\":null,\"Retry\":[{\"ErrorEquals\":[\"States.ALL\"],\"IntervalSeconds\":10,\"MaxAttempts\":5}],\"Catch\":[{\"ErrorEquals\":[\"States.ALL\"],\"Next\":\"failed\"}]},\"final step\":{\"Type\":\"Pass\",\"End\":true},\"failed\":{\"Type\":\"Fail\",\"Error\":\"DidNotWork\",\"Cause\":\"We got stuck\"}},\"TimeoutSeconds\":30}",
+    "DefinitionString": "{\"StartAt\":\"my custom task\",\"States\":{\"my custom task\":{\"Next\":\"final step\",\"Type\":\"Task\",\"Resource\":\"arn:aws:states:::dynamodb:putItem\",\"Parameters\":{\"TableName\":\"my-cool-table\",\"Item\":{\"id\":{\"S\":\"my-entry\"}}},\"ResultPath\":null,\"Retry\":[{\"ErrorEquals\":[\"States.Timeout\"],\"IntervalSeconds\":10,\"MaxAttempts\":5},{\"ErrorEquals\":[\"States.Permissions\"],\"IntervalSeconds\":20,\"MaxAttempts\":2}],\"Catch\":[{\"ErrorEquals\":[\"States.ALL\"],\"Next\":\"failed\"}]},\"final step\":{\"Type\":\"Pass\",\"End\":true},\"failed\":{\"Type\":\"Fail\",\"Error\":\"DidNotWork\",\"Cause\":\"We got stuck\"}},\"TimeoutSeconds\":30}",
     "RoleArn": {
      "Fn::GetAtt": [
       "StateMachineRoleB840431D",

packages/@aws-cdk-testing/framework-integ/test/aws-stepfunctions/test/integ.custom-state.ts

@@ -23,6 +23,11 @@ const stateJson = {
     },
   },
   ResultPath: null,
+  Retry: [{
+    ErrorEquals: [sfn.Errors.PERMISSIONS],
+    IntervalSeconds: 20,
+    MaxAttempts: 2,
+  }],
 };
 
 const failure = new sfn.Fail(stack, 'failed', {
@@ -36,7 +41,7 @@ const custom = new sfn.CustomState(stack, 'my custom task', {
 
 custom.addCatch(failure);
 custom.addRetry({
-  errors: [sfn.Errors.ALL],
+  errors: [sfn.Errors.TIMEOUT],
   interval: cdk.Duration.seconds(10),
   maxAttempts: 5,
 });

packages/aws-cdk-lib/aws-stepfunctions/README.md

@@ -562,6 +562,10 @@ Custom states can be chained together with any of the other states to create you
 definition. You will also need to provide any permissions that are required to the `role` that
 the State Machine uses.
 
+The Retry and Catch fields are available for error handling.
+You can configure the Retry field by defining it in the JSON object or by adding it using the `addRetry` method.
+However, the Catch field cannot be configured by defining it in the JSON object, so it must be added using the `addCatch` method.
+
 The following example uses the `DynamoDB` service integration to insert data into a DynamoDB table.
 
 ```ts

packages/aws-cdk-lib/aws-stepfunctions/lib/states/custom-state.ts

@@ -68,10 +68,17 @@ export class CustomState extends State implements IChainable, INextable {
    * Returns the Amazon States Language object for this state
    */
   public toStateJson(): object {
-    return {
+    const state = {
       ...this.renderNextEnd(),
       ...this.stateJson,
       ...this.renderRetryCatch(),
     };
+
+    // merge the Retry filed defined in the stateJson into the state
+    if (Array.isArray(this.stateJson.Retry)) {
+      state.Retry = [...state.Retry, ...this.stateJson.Retry];
+    }
+
+    return state;
   }
 }

packages/aws-cdk-lib/aws-stepfunctions/test/custom-state.test.ts

@@ -163,4 +163,83 @@ describe('Custom State', () => {
       },
     );
   });
+
+  test('respect the Retry field in the stateJson', () => {
+    // GIVEN
+    const custom = new sfn.CustomState(stack, 'Custom', {
+      stateJson: {
+        Type: 'Task',
+        Resource: 'arn:aws:states:::dynamodb:putItem',
+        Parameters: {
+          TableName: 'MyTable',
+          Item: {
+            id: {
+              S: 'MyEntry',
+            },
+          },
+        },
+        ResultPath: null,
+        Retry: [
+          {
+            ErrorEquals: [sfn.Errors.TIMEOUT],
+            IntervalSeconds: 20,
+            MaxAttempts: 2,
+          },
+          {
+            ErrorEquals: [sfn.Errors.RESULT_PATH_MATCH_FAILURE],
+            IntervalSeconds: 20,
+            MaxAttempts: 2,
+          },
+        ],
+      },
+    });
+    const chain = sfn.Chain.start(custom);
+
+    // WHEN
+    custom.addRetry({
+      errors: [sfn.Errors.PERMISSIONS],
+      interval: cdk.Duration.seconds(10),
+      maxAttempts: 5,
+    });
+
+    // THEN
+    expect(render(stack, chain)).toStrictEqual(
+      {
+        StartAt: 'Custom',
+        States: {
+          Custom: {
+            Type: 'Task',
+            Resource: 'arn:aws:states:::dynamodb:putItem',
+            Parameters: {
+              TableName: 'MyTable',
+              Item: {
+                id: {
+                  S: 'MyEntry',
+                },
+              },
+            },
+            ResultPath: null,
+            Retry: [
+              {
+                ErrorEquals: ['States.Permissions'],
+                IntervalSeconds: 10,
+                MaxAttempts: 5,
+              },
+              {
+                ErrorEquals: ['States.Timeout'],
+                IntervalSeconds: 20,
+                MaxAttempts: 2,
+              },
+              {
+                ErrorEquals: ['States.ResultPathMatchFailure'],
+                IntervalSeconds: 20,
+                MaxAttempts: 2,
+              },
+            ],
+            End: true,
+          },
+        },
+      },
+    );
+  });
 });