Add tests

Author: devpow112

.eslintrc

@@ -5,7 +5,8 @@
         "es6": true
     },
     "parserOptions": {
-        "sourceType": "module"
+        "sourceType": "module",
+        "ecmaVersion": 2019
     },
     "extends": "eslint:recommended",
     "rules": {
@@ -22,6 +23,18 @@
         "indent": [
             "error",
             2
+        ],
+        "semi": [
+            "error",
+            "always"
+        ],
+        "object-curly-spacing": [
+            "error",
+            "always"
+        ],
+        "array-bracket-spacing": [
+            "error",
+            "never"
         ]
     }
 }

.gitignore

@@ -1,3 +1,5 @@
+.nyc_output/
 .vscode/
+coverage/
 dist/
 node_modules/

.mocharc.json

@@ -0,0 +1,6 @@
+{
+  "spec": "test/**/*.spec.js",
+  "slow": 10,
+  "timeout": 50,
+  "require": "@babel/register"
+}

.nycrc

@@ -0,0 +1,15 @@
+{
+  "all": true,
+  "check-coverage": true,
+  "lines": 95,
+  "statements": 95,
+  "functions": 95,
+  "branches": 90,
+  "include": [
+    "src/**/*.js"
+  ],
+  "reporter": [
+    "lcov",
+    "text"
+  ]
+}

package.json

@@ -7,7 +7,7 @@
     "build": "babel --out-dir dist src",
     "lint": "eslint .",
     "format": "npm run lint -s -- --fix",
-    "test": "npm run lint -s"
+    "test": "npm run lint -s && nyc mocha --colors"
   },
   "repository": {
     "type": "git",
@@ -35,7 +35,14 @@
     "@babel/cli": "^7.10.5",
     "@babel/core": "^7.11.4",
     "@babel/preset-env": "^7.11.0",
+    "@babel/register": "^7.10.5",
     "babel-plugin-add-module-exports": "^1.0.2",
-    "eslint": "^7.7.0"
+    "chai": "^4.2.0",
+    "eslint": "^7.7.0",
+    "eslint-plugin-chai-friendly": "^0.6.0",
+    "eslint-plugin-mocha": "^8.0.0",
+    "mocha": "^8.1.1",
+    "nyc": "^15.1.0",
+    "sinon": "^9.0.3"
   }
 }

src/functionValuePlugin.js renamed to src/function-value-plugin.js

@@ -5,7 +5,7 @@ export class FunctionValuePlugin {
     this.variableResolvers = {
       'fn.arn': (value) => this.getFunctionARNStatement(value),
       'fn.name': (value) => this.getFunctionNameStatement(value)
-    }
+    };
   }
 
   getFunctionLogicalId(value) {

src/index.js

@@ -1,3 +1,3 @@
-import { FunctionValuePlugin } from './functionValuePlugin';
+import { FunctionValuePlugin } from './function-value-plugin';
 
 export default FunctionValuePlugin;

test/.eslintrc

@@ -0,0 +1,20 @@
+{
+  "env": {
+    "mocha": true
+  },
+  "plugins": [
+    "mocha",
+    "chai-friendly"
+  ],
+  "parserOptions": {
+    "sourceType": "module"
+  },
+  "rules": {
+    "mocha/no-identical-title": [
+      "error"
+    ],
+    "chai-friendly/no-unused-expressions": [
+      "error"
+    ]
+  }
+}

test/function-value-plugin.spec.js

@@ -0,0 +1,44 @@
+import Plugin from '../src/index';
+import sinon from 'sinon';
+import { expect } from 'chai';
+
+describe('plugin', () => {
+  const functionName = 'health';
+  const logicalId = 'HealthLambdaFunction';
+  let variableResolvers;
+
+  before(() => {
+    const service = { getAllFunctions: sinon.stub().returns([functionName]) };
+    const provider = {
+      naming: { getLambdaLogicalId: sinon.stub().returns(logicalId) }
+    };
+    const serverless = {
+      service,
+      getProvider: sinon.stub().returns(provider)
+    };
+
+    variableResolvers = new Plugin(serverless).variableResolvers;
+  });
+
+  [
+    { type: 'arn', expected: `!GetAtt ${logicalId}.ARN` },
+    { type: 'name', expected: `!Ref ${logicalId}` }
+  ].forEach(test => {
+    it(`will generate function ${test.type} snippet`, async () => {
+      const resolver = `fn.${test.type}`;
+      const value = `${resolver}:${functionName}`;
+      const result = await variableResolvers[resolver](value);
+
+      expect(result).to.equal(test.expected);
+    });
+  });
+
+  it('will fail if function not found', async () => {
+    try {
+      await variableResolvers['fn.arn']('fn.arn:test');
+      throw new Error('should not get here');
+    } catch {
+      // success!
+    }
+  });
+});