update coverall to 100%

Author: horike37

lib/index.js

@@ -160,6 +160,7 @@ class ServerlessStepFunctions {
     }
     message += '\n';
     this.serverless.cli.consoleLog(message);
+    return message;
   }
 }
 module.exports = ServerlessStepFunctions;

lib/index.test.js

@@ -5,6 +5,8 @@ const sinon = require('sinon');
 const BbPromise = require('bluebird');
 const Serverless = require('serverless/lib/Serverless');
 const AwsProvider = require('serverless/lib/plugins/aws/provider/awsProvider');
+const CLI = require('serverless/lib/classes/CLI');
+const chalk = require('chalk');
 const ServerlessStepFunctions = require('./index');
 
 describe('#index', () => {
@@ -18,8 +20,8 @@ describe('#index', () => {
       stage: 'dev',
       region: 'us-east-1',
     };
-    serverless.init();
     serverless.setProvider('aws', new AwsProvider(serverless));
+    serverless.cli = new CLI(serverless);
     serverlessStepFunctions = new ServerlessStepFunctions(serverless, options);
   });
 
@@ -211,4 +213,140 @@ describe('#index', () => {
         });
     });
   });
+
+  describe('#deploy()', () => {
+    let consoleLogStub;
+
+    beforeEach(() => {
+      consoleLogStub = sinon.stub(serverless.cli, 'consoleLog').returns();
+      serverlessStepFunctions.endpointInfo = 'https://example.com';
+    });
+
+    afterEach(() => {
+      serverlessStepFunctions.serverless.cli.consoleLog.restore();
+    });
+
+    it('should not display endpoints if http event not given', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {};
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+
+    it('should display endpoints if http event given as object', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {
+        stateMachines: {
+          myStateMachine: {
+            define: 'my-define',
+            events: [
+              {
+                http: {
+                  path: 'foo/bar',
+                  method: 'post',
+                },
+              },
+            ],
+          },
+        },
+      };
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n  POST - https://example.com/foo/bar';
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+
+    it('should display endpoints if http event given as string', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {
+        stateMachines: {
+          myStateMachine: {
+            define: 'my-define',
+            events: [
+              {
+                http: 'post foo/bar',
+              },
+            ],
+          },
+        },
+      };
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n  POST - https://example.com/foo/bar';
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+
+    it('should display endpoints if http event given with path as slash', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {
+        stateMachines: {
+          myStateMachine: {
+            define: 'my-define',
+            events: [
+              {
+                http: 'post /',
+              },
+            ],
+          },
+        },
+      };
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n  POST - https://example.com';
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+
+    it('should display endpoints if events given as object', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {
+        stateMachines: {
+          myStateMachine: {
+            define: 'my-define',
+            events: {},
+          },
+        },
+      };
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+
+    it('should display endpoints if http event does not exists', () => {
+      serverlessStepFunctions.serverless.service.stepFunctions = {
+        stateMachines: {
+          myStateMachine: {
+            define: 'my-define',
+            events: [
+              {
+                foo: 'bar',
+              },
+            ],
+          },
+        },
+      };
+      let expectedMessage = '';
+      expectedMessage += `${chalk.yellow.underline('Serverless StepFunctions OutPuts')}\n`;
+      expectedMessage += `${chalk.yellow('endpoints:')}`;
+      expectedMessage += '\n';
+      const message = serverlessStepFunctions.display();
+      expect(consoleLogStub.calledOnce).to.equal(true);
+      expect(message).to.equal(expectedMessage);
+    });
+  });
 });

lib/invoke/invoke.test.js

@@ -8,6 +8,7 @@ const sinon = require('sinon');
 const path = require('path');
 const Serverless = require('serverless/lib/Serverless');
 const AwsProvider = require('serverless/lib/plugins/aws/provider/awsProvider');
+const CLI = require('serverless/lib/classes/CLI');
 const ServerlessStepFunctions = require('./../index');
 
 describe('invoke', () => {
@@ -16,8 +17,8 @@ describe('invoke', () => {
 
   beforeEach(() => {
     serverless = new Serverless();
-    serverless.init();
     serverless.setProvider('aws', new AwsProvider(serverless));
+    serverless.cli = new CLI(serverless);
     serverless.service.service = 'new-service';
     const options = {
       stage: 'dev',