test: added more failing tests for new feature

Author: theburningmonk

lib/deploy/stepFunctions/compileIamRole.test.js

@@ -1259,4 +1259,66 @@ describe('#compileIamRole', () => {
     ];
     expect(lambdaPermissions[0].Resource).to.deep.eq(lambdaArns);
   });
+
+  it('should support local function names', () => {
+    const getStateMachine = (name, functionName) => ({
+      name,
+      definition: {
+        StartAt: 'A',
+        States: {
+          A: {
+            Type: 'Task',
+            Resource: 'arn:aws:states:::lambda:invoke',
+            Parameters: {
+              FunctionName: functionName,
+              Payload: {
+                'ExecutionName.$': '$$.Execution.Name',
+              },
+            },
+            End: true,
+          },
+        },
+      },
+    });
+
+    const lambda1 = { Ref: 'hello-world' };
+    const lambda2 = { 'Fn::GetAtt': ['hello-world', 'Arn'] };
+
+    serverless.functions = {
+      'hello-world': {
+        handler: 'hello-world.handler',
+      },
+    };
+
+    serverless.service.stepFunctions = {
+      stateMachines: {
+        myStateMachine1: getStateMachine('sm1', lambda1),
+        myStateMachine2: getStateMachine('sm2', lambda2),
+      },
+    };
+
+    serverlessStepFunctions.compileIamRole();
+    const statements = serverlessStepFunctions.serverless.service
+      .provider.compiledCloudFormationTemplate.Resources.IamRoleStateMachineExecution
+      .Properties.Policies[0].PolicyDocument.Statement;
+
+    const lambdaPermissions = statements.filter(s => _.isEqual(s.Action, ['lambda:InvokeFunction']));
+    expect(lambdaPermissions).to.have.lengthOf(1);
+
+    const lambdaArns = [
+      {
+        'Fn::Sub': [
+          'arn:aws:lambda:${AWS::Region}:${AWS::AccountId}:function:${FunctionName}',
+          { FunctionName: { Ref: 'HelloDashworldLambdaFunction' } },
+        ],
+      },
+      {
+        'Fn::GetAtt': [
+          'HelloDashworldLambdaFunction',
+          'Arn',
+        ],
+      },
+    ];
+    expect(lambdaPermissions[0].Resource).to.deep.eq(lambdaArns);
+  });
 });

lib/deploy/stepFunctions/compileStateMachines.test.js

@@ -775,6 +775,6 @@ describe('#compileStateMachines', () => {
     expect(params).to.haveOwnProperty(functionParam);
 
     const refParam = params[functionParam];
-    expect(refParam).to.eql({ 'Fn::GetAtt': ['HelloDashWorldLambdaFunction', 'Arn'] });
+    expect(refParam).to.eql({ 'Fn::GetAtt': ['HelloDashworldLambdaFunction', 'Arn'] });
   });
 });