From ea3059dd66d2095a516986a7847ef8e01608af2f Mon Sep 17 00:00:00 2001
From: yamatatsu <the.kinnikuman@gmail.com>
Date: Fri, 4 Mar 2022 12:38:27 +0900
Subject: [PATCH] test behavior of action

---
 .../aws-iotevents/test/detector-model.test.ts | 95 +++++++++++++++++++
 1 file changed, 95 insertions(+)

diff --git a/packages/@aws-cdk/aws-iotevents/test/detector-model.test.ts b/packages/@aws-cdk/aws-iotevents/test/detector-model.test.ts
index ce8389c92e50d..ec754f25c1aad 100644
--- a/packages/@aws-cdk/aws-iotevents/test/detector-model.test.ts
+++ b/packages/@aws-cdk/aws-iotevents/test/detector-model.test.ts
@@ -176,6 +176,101 @@ test('can set actions to events', () => {
   });
 });
 
+
+test('can set an action to multiple detector models', () => {
+  // GIVEN an action
+  const action: iotevents.IAction = {
+    bind: (_, { role }) => {
+      role.addToPrincipalPolicy(new iam.PolicyStatement({
+        actions: ['lambda:InvokeFunction'],
+        resources: ['arn:aws:lambda:us-east-1:123456789012:function:MyFn'],
+      }));
+      return {
+        configuration: {
+          lambda: { functionArn: 'arn:aws:lambda:us-east-1:123456789012:function:MyFn' },
+        },
+      };
+    },
+  };
+
+  // WHEN the action is set to two detector models
+  new iotevents.DetectorModel(stack, 'MyDetectorModel1', {
+    detectorModelName: 'MyDetectorModel1',
+    initialState: new iotevents.State({
+      stateName: 'test-state',
+      onEnter: [{
+        eventName: 'test-eventName1',
+        condition: iotevents.Expression.currentInput(input),
+        actions: [action],
+      }],
+    }),
+  });
+  new iotevents.DetectorModel(stack, 'MyDetectorModel2', {
+    detectorModelName: 'MyDetectorModel2',
+    initialState: new iotevents.State({
+      stateName: 'test-state',
+      onEnter: [{
+        eventName: 'test-eventName1',
+        condition: iotevents.Expression.currentInput(input),
+        actions: [action],
+      }],
+    }),
+  });
+
+  // THEN creates two detector model resouces and two iam policy resources
+  Template.fromStack(stack).resourceCountIs('AWS::IoTEvents::DetectorModel', 2);
+  Template.fromStack(stack).resourceCountIs('AWS::IAM::Policy', 2);
+
+  Template.fromStack(stack).hasResourceProperties('AWS::IoTEvents::DetectorModel', {
+    DetectorModelName: 'MyDetectorModel1',
+    DetectorModelDefinition: {
+      States: [
+        Match.objectLike({
+          OnEnter: {
+            Events: [{
+              Actions: [{ Lambda: { FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:MyFn' } }],
+            }],
+          },
+        }),
+      ],
+    },
+  });
+  Template.fromStack(stack).hasResourceProperties('AWS::IoTEvents::DetectorModel', {
+    DetectorModelName: 'MyDetectorModel2',
+    DetectorModelDefinition: {
+      States: [
+        Match.objectLike({
+          OnEnter: {
+            Events: [{
+              Actions: [{ Lambda: { FunctionArn: 'arn:aws:lambda:us-east-1:123456789012:function:MyFn' } }],
+            }],
+          },
+        }),
+      ],
+    },
+  });
+  Template.fromStack(stack).hasResourceProperties('AWS::IAM::Policy', {
+    Roles: [{ Ref: 'MyDetectorModel1DetectorModelRoleB36845CD' }],
+    PolicyDocument: {
+      Statement: [{
+        Action: 'lambda:InvokeFunction',
+        Effect: 'Allow',
+        Resource: 'arn:aws:lambda:us-east-1:123456789012:function:MyFn',
+      }],
+    },
+  });
+  Template.fromStack(stack).hasResourceProperties('AWS::IAM::Policy', {
+    Roles: [{ Ref: 'MyDetectorModel2DetectorModelRole3C437E90' }],
+    PolicyDocument: {
+      Statement: [{
+        Action: 'lambda:InvokeFunction',
+        Effect: 'Allow',
+        Resource: 'arn:aws:lambda:us-east-1:123456789012:function:MyFn',
+      }],
+    },
+  });
+});
+
 test('can set states with transitions', () => {
   // GIVEN
   const firstState = new iotevents.State({