test: add test for jsonata

Author: zirkelc

lib/deploy/stepFunctions/compileStateMachines.test.js

@@ -16,7 +16,7 @@ describe('#compileStateMachines', () => {
     serverless.cli = new CLI(serverless);
     serverless.configSchemaHandler = {
       // eslint-disable-next-line no-unused-vars
-      defineTopLevelProperty: (propertyName, propertySchema) => {},
+      defineTopLevelProperty: (propertyName, propertySchema) => { },
     };
     serverless.servicePath = true;
     serverless.service.service = 'step-functions';
@@ -1728,4 +1728,59 @@ describe('#compileStateMachines', () => {
     const stateMachineObj = JSON.parse(stateMachine.Properties.DefinitionString);
     expect(stateMachineObj.States).to.haveOwnProperty('One');
   });
+
+  it('should compile with the new JSONata fields', () => {
+    serverless.service.stepFunctions = {
+      stateMachines: {
+        myStateMachine1: {
+          id: 'Test',
+          name: 'Test',
+          definition: {
+            Comment: 'Test JSONata features',
+            QueryLanguage: 'JSONata',
+            StartAt: 'Store inputs',
+            States: {
+              'Store inputs': {
+                Type: 'Pass',
+                Next: 'Output transformation',
+                Assign: {
+                  desiredPrice: '{% $states.input.desired_price %}',
+                  maximumWait: '{% $states.input.max_days %}',
+                },
+              },
+              'Output transformation': {
+                Type: 'Succeed',
+                Output: {
+                  lastName:
+                    "{% 'Last=>' & $states.input.customer.lastName %}",
+                  orderValue: '{% $states.input.order.total %}',
+                },
+              },
+            },
+          },
+        },
+      },
+    };
+
+    serverlessStepFunctions.compileStateMachines();
+
+    const stateMachine = serverlessStepFunctions.serverless.service
+      .provider.compiledCloudFormationTemplate.Resources
+      .Test;
+
+    expect(stateMachine.Properties.DefinitionString).not.to.haveOwnProperty('Fn::Sub');
+    const stateMachineObj = JSON.parse(stateMachine.Properties.DefinitionString);
+    expect(stateMachineObj.States).to.haveOwnProperty('Store inputs');
+    expect(stateMachineObj.States).to.haveOwnProperty('Output transformation');
+
+    expect(stateMachineObj.QueryLanguage).to.equal('JSONata');
+    expect(stateMachineObj.States['Store inputs'].Assign).to.deep.equal({
+      desiredPrice: '{% $states.input.desired_price %}',
+      maximumWait: '{% $states.input.max_days %}',
+    });
+    expect(stateMachineObj.States['Output transformation'].Output).to.deep.equal({
+      lastName: "{% 'Last=>' & $states.input.customer.lastName %}",
+      orderValue: '{% $states.input.order.total %}',
+    });
+  });
 });