如何使用CDK模板测试Step Function是否包含特定属性的任务？

一、验证状态机中存在目标SqsSendMessage任务 #

你可以借助CDK的aws-cdk-lib/assertions模块，通过解析状态机的DefinitionString（状态机逻辑的JSON序列化字符串）来验证任务存在及属性正确性。以下是两种常用实现方式：

方式1：解析定义JSON后断言 #

这种方式更灵活，适合复杂属性的校验：

it("should contain the 'Send to Queue' SqsSendMessage task with correct properties", () => {
  // 找到状态机资源的属性
  const stateMachineResources = template.findResources("AWS::StepFunctions::StateMachine");
  // 取第一个状态机资源（若堆栈中只有一个状态机）
  const stateMachineProps = Object.values(stateMachineResources)[0].Properties;
  // 解析状态机定义JSON
  const stateDefinition = JSON.parse(stateMachineProps.DefinitionString);

  // 遍历所有状态，筛选目标SQS发送任务
  const sendToQueueTask = Object.values(stateDefinition.States).find(state => 
    state.Type === "Task" &&
    // 对应WAIT_FOR_TASK_TOKEN集成模式的SQS任务ARN格式
    state.Resource === "arn:aws:states:::sqs:sendMessage.waitForTaskToken" &&
    // 校验队列URL
    state.Parameters["QueueUrl"] === sqsStack.queueName &&
    // 校验消息分组ID的JSON路径
    state.Parameters["MessageGroupId"] === "$.messageGroupId" &&
    // 校验超时时间（15分钟=900秒）
    state.TimeoutSeconds === 900 &&
    // 校验结果路径
    state.ResultPath === "$.result"
  );

  // 断言任务存在且核心属性正确
  expect(sendToQueueTask).toBeDefined();
  // 可根据实际需求细化MessageBody的校验
  expect(sendToQueueTask.Parameters["MessageBody"]).toBeDefined();
});

方式2：用hasResourceProperties匹配正则 #

如果属性较简单，可以直接通过正则匹配DefinitionString：

it("should include the SqsSendMessage task with required configurations", () => {
  template.hasResourceProperties("AWS::StepFunctions::StateMachine", {
    DefinitionString: Match.stringLikeRegexp(`"Resource": "arn:aws:states:::sqs:sendMessage.waitForTaskToken"`),
    DefinitionString: Match.stringLikeRegexp(`"QueueUrl": "${sqsStack.queueName}"`),
    DefinitionString: Match.stringLikeRegexp(`"MessageGroupId": "$.messageGroupId"`),
    DefinitionString: Match.stringLikeRegexp(`"TimeoutSeconds": 900`),
    DefinitionString: Match.stringLikeRegexp(`"ResultPath": "$.result"`),
  });
});

二、检查任务是否属于状态机Definition的一部分 #

上面的两种方式本质都是在验证任务是否存在于状态机的DefinitionString中——这个字符串是状态机整个执行逻辑的JSON序列化结果，所有状态（包括你的SqsSendMessage任务）都嵌套在States对象下。只要能在解析后的stateDefinition.States中找到目标任务，或者通过正则匹配到任务的特征内容，就说明它属于Definition的一部分。

三、这种细粒度测试是否必要？ #

分场景判断：

• 必要场景：如果这个SQS发送任务是业务核心流程的关键节点（比如订单消息投递、异步任务触发），细粒度测试能有效防止后续代码变更时误删任务、修改核心属性（如超时时间、队列地址）导致的生产故障；团队协作时，这类测试也能作为活文档，明确状态机的预期结构。
• 非必要场景：如果任务只是辅助性逻辑（比如日志上报、次要数据同步），保持“状态机存在”的粗粒度测试即可，避免增加不必要的维护成本——如果任务属性频繁变动，测试代码也需要同步更新，反而会成为负担。

内容的提问来源于stack exchange，提问作者Luke