serverless/test/unit/lib/plugins/invoke.test.js

'use strict';

const chai = require('chai');
const overrideEnv = require('process-utils/override-env');
const Invoke = require('../../../../lib/plugins/invoke');
const Serverless = require('../../../../lib/serverless');

chai.use(require('chai-as-promised'));

const expect = chai.expect;

describe('Invoke', () => {
  let invoke;
  let serverless;
  let restoreEnv;

  beforeEach(() => {
    ({ restoreEnv } = overrideEnv());
    serverless = new Serverless({ commands: [], options: {} });
    invoke = new Invoke(serverless);
  });

  afterEach(() => restoreEnv());

  describe('#constructor()', () => {
    it('should have commands', () => expect(invoke.commands).to.be.not.empty);
    it('should have hooks', () => expect(invoke.hooks).to.be.not.empty);
  });

  describe('#loadEnvVarsForLocal()', () => {
    it('should set IS_LOCAL', () => {
      invoke.loadEnvVarsForLocal();
      expect(process.env.IS_LOCAL).to.equal('true');
      expect(serverless.service.provider.environment.IS_LOCAL).to.equal('true');
    });
    it('should leave provider env variable untouched if already defined', () => {
      serverless.service.provider.environment = { IS_LOCAL: 'false' };
      invoke.loadEnvVarsForLocal();
      expect(serverless.service.provider.environment.IS_LOCAL).to.equal('false');
    });
  });

  describe('hooks', () => {
    describe('invoke:local:loadEnvVars', () => {
      it('should be an event', () => {
        expect(invoke.commands.invoke.commands.local.lifecycleEvents).to.contain('loadEnvVars');
      });

      it('should set IS_LOCAL', async () =>
        expect(invoke.hooks['invoke:local:loadEnvVars']()).to.be.fulfilled.then(() => {
          expect(process.env.IS_LOCAL).to.equal('true');
          expect(serverless.service.provider.environment.IS_LOCAL).to.equal('true');
        }));

      it('should leave provider env variable untouched if already defined', async () => {
        serverless.service.provider.environment = { IS_LOCAL: 'false' };
        return expect(invoke.hooks['invoke:local:loadEnvVars']()).to.be.fulfilled.then(() => {
          expect(serverless.service.provider.environment.IS_LOCAL).to.equal('false');
        });
      });

      it('should accept a single env option', async () => {
        invoke.options = { env: 'NAME=value' };
        return expect(invoke.hooks['invoke:local:loadEnvVars']()).to.be.fulfilled.then(() =>
          expect(process.env.NAME).to.equal('value')
        );
      });

      it('should accept multiple env options', async () => {
        invoke.options = { env: ['NAME1=val1', 'NAME2=val2'] };

        return expect(invoke.hooks['invoke:local:loadEnvVars']())
          .to.be.fulfilled.then(() => expect(process.env.NAME1).to.equal('val1'))
          .then(() => expect(process.env.NAME2).to.equal('val2'));
      });
    });
  });
});