test: added failing test for new feature

Author: theburningmonk

lib/deploy/stepFunctions/compileStateMachines.test.js

@@ -731,4 +731,50 @@ describe('#compileStateMachines', () => {
     expect(stateMachineObj.States).to.haveOwnProperty('RefreshLead');
     expect(stateMachineObj.States).to.haveOwnProperty('EndState');
   });
+
+  it('should support local function names', () => {
+    serverless.service.functions = {
+      'hello-world': {
+        handler: 'hello-world.handler',
+      },
+    };
+    serverless.service.stepFunctions = {
+      stateMachines: {
+        myStateMachine1: {
+          id: 'Test',
+          definition: {
+            StartAt: 'Lambda',
+            States: {
+              Lambda: {
+                Type: 'Task',
+                Resource: {
+                  'Fn::GetAtt': ['hello-world', 'Arn'],
+                },
+                End: true,
+              },
+            },
+          },
+        },
+      },
+    };
+
+    serverlessStepFunctions.compileStateMachines();
+    const stateMachine = serverlessStepFunctions.serverless.service
+      .provider.compiledCloudFormationTemplate.Resources
+      .Test;
+
+    expect(stateMachine.Properties.DefinitionString).to.haveOwnProperty('Fn::Sub');
+    expect(stateMachine.Properties.DefinitionString['Fn::Sub']).to.have.lengthOf(2);
+
+    const [json, params] = stateMachine.Properties.DefinitionString['Fn::Sub'];
+    const modifiedDefinition = JSON.parse(json);
+
+    const lambda = modifiedDefinition.States.Lambda;
+    expect(lambda.Resource.startsWith('${')).to.eq(true);
+    const functionParam = lambda.Resource.replace(/[${}]/g, '');
+    expect(params).to.haveOwnProperty(functionParam);
+
+    const refParam = params[functionParam];
+    expect(refParam).to.eql({ 'Fn::GetAtt': ['HelloDashWorldLambdaFunction', 'Arn'] });
+  });
 });