serverless/lib/plugins/awsDeploy/tests/deployFunctions.js

'use strict';

const expect = require('chai').expect;
const path = require('path');
const os = require('os');
const deployFunctions = require('../lib/deployFunctions');
const Serverless = require('../../../Serverless');

describe('deployFunctions', () => {
  let serverless;
  let awsDeployMock;

  class AwsDeployMock {
    constructor(serverless) {
      Object.assign(this, deployFunctions);
      this.serverless = serverless;
      this.options = {};
      this.deployedFunctions = [];
    }
  }

  const rawFunctionObjectsMock = {
    name_template: 'name-template-name',
    first: {
      handler: 'first.function.handler',
    },
    second: {
      handler: 'second.function.handler',
    },
  };

  const deployedFunctionsArrayMock = [
    {
      name: 'function',
      handler: 'function.handler',
    },
  ];

  const functionCodeMock = `
    'use strict';

    module.exports.handler = function(event, context, cb) {
      return cb(null, {
        message: 'First function'
      });
    };
  `;

  beforeEach(() => {
    serverless = new Serverless();
    awsDeployMock = new AwsDeployMock(serverless);
  });

  describe('#validateForDeployFunctions', () => {
    it('should throw an error if the service name is not set', () => {
      awsDeployMock.serverless.service.service = '';
      expect(() => awsDeployMock.validateForDeployFunctions().to.throw(Error));
    });

    it('should throw an error if the region is not set', () => {
      awsDeployMock.options.region = '';
      expect(() => awsDeployMock.validateForDeployFunctions().to.throw(Error));
    });
  });

  describe('#extractHandlers()', () => {
    it('should extract all the handlers in the function definitions', () => {
      serverless.service.functions = rawFunctionObjectsMock;

      return awsDeployMock.extractFunctionHandlers().then(() => {
        expect(
          awsDeployMock.deployedFunctions[0].handler
        ).to.equal(rawFunctionObjectsMock.first.handler);
        expect(
          awsDeployMock.deployedFunctions[1].handler
        ).to.equal(rawFunctionObjectsMock.second.handler);
      });
    });
  });

  describe('#zipFunctions()', () => {
    it('should zip the functions', () => {
      // set the deployedFunctions array
      awsDeployMock.deployedFunctions = deployedFunctionsArrayMock;

      // create a function in a temporary directory
      const tmpDirPath = path.join(os.tmpdir(), (new Date).getTime().toString());
      const tmpFilePath = path.join(tmpDirPath, 'function.js');
      serverless.utils.writeFileSync(tmpFilePath, functionCodeMock);

      // set the servicePath
      serverless.config.servicePath = tmpDirPath;

      return awsDeployMock.zipFunctions().then(() => {
        expect(awsDeployMock.deployedFunctions[0].zipFilePath).to.match(/.+\-\d+/);
      });
    });
  });

  describe('#uploadZipFilesToS3Bucket()', () => {
    it('should upload the zip files to the S3 bucket', () => {
      // set the deployedFunctions array
      awsDeployMock.deployedFunctions = deployedFunctionsArrayMock;

      awsDeployMock.serverless.service.service = 'first-service';
      awsDeployMock.options.region = 'eu-central-1';

      // create a mock zip file
      const tmpDirPath = path.join(os.tmpdir(), (new Date).getTime().toString());
      const tmpFilePath = path.join(tmpDirPath, 'function.zip');
      serverless.utils.writeFileSync(tmpFilePath, 'some binary data');

      awsDeployMock.deployedFunctions[0].zipFilePath = tmpFilePath;

      return awsDeployMock.uploadZipFilesToS3Bucket().then(() => {
        // TODO use sinon to mock S3
        expect(true).to.equal(true);
      });
    });
  });
});