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

'use strict';

const expect = require('chai').expect;
const sinon = require('sinon');
const os = require('os');
const path = require('path');
const AWS = require('aws-sdk');
const BbPromise = require('bluebird');
const AwsDeploy = require('../awsDeploy');
const Serverless = require('../../../Serverless');

const serverless = new Serverless();
serverless.init();
const awsDeploy = new AwsDeploy(serverless);
const tmpDirPath = path.join(os.tmpdir(), (new Date).getTime().toString());
const serverlessEnvYamlPath = path.join(tmpDirPath, 'serverless.env.yaml');
const serverlessEnvYaml = {
  vars: {},
  stages: {
    dev: {
      vars: {},
      regions: {
        aws_useast1: {
          vars: {},
        },
      },
    },
  },
};
serverless.service.service = `service-${(new Date).getTime().toString()}`;
serverless.config.servicePath = tmpDirPath;
serverless.utils.writeFileSync(serverlessEnvYamlPath, serverlessEnvYaml);
awsDeploy.options = {
  stage: 'dev',
  region: 'us-east-1',
};
awsDeploy.CloudFormation = new AWS.CloudFormation({ region: 'us-east-1' });
BbPromise.promisifyAll(awsDeploy.CloudFormation, { suffix: 'Promised' });

describe('#update()', () => {
  it('should update stack', () => {
    const updateStackStub = sinon
      .stub(awsDeploy.CloudFormation, 'updateStackPromised').returns(BbPromise.resolve());
    return awsDeploy.update().then(() => {
      expect(updateStackStub.calledOnce).to.be.equal(true);
      awsDeploy.CloudFormation.updateStackPromised.restore();
    });
  });
});

describe('#monitorUpdate()', () => {
  it('should keep monitoring until UPDATE_COMPLETE stack status', () => {
    const describeStub = sinon.stub(awsDeploy.CloudFormation, 'describeStacksPromised');
    const cfDataMock = {
      StackId: 'new-service-dev',
    };
    const DescribeReturn = {
      Stacks: [
        {
          StackStatus: 'UPDATE_IN_PROGRESS',
        },
      ],
    };
    const finalDescribeReturn = {
      Stacks: [
        {
          StackStatus: 'UPDATE_COMPLETE',
        },
      ],
    };

    describeStub.onCall(0).returns(BbPromise.resolve(DescribeReturn));
    describeStub.onCall(1).returns(BbPromise.resolve(DescribeReturn));
    describeStub.onCall(2).returns(BbPromise.resolve(finalDescribeReturn));

    return awsDeploy.monitorUpdate(cfDataMock, 10).then((stack) => {
      expect(describeStub.callCount).to.be.equal(3);
      expect(stack.StackStatus).to.be.equal('UPDATE_COMPLETE');
      awsDeploy.CloudFormation.describeStacksPromised.restore();
    });
  });

  it('should throw an error if CloudFormation returned unusual stack status', () => {
    const describeStub = sinon.stub(awsDeploy.CloudFormation, 'describeStacksPromised');
    const cfDataMock = {
      StackId: 'new-service-dev',
    };
    const DescribeReturn = {
      Stacks: [
        {
          StackStatus: 'UPDATE_IN_PROGRESS',
        },
      ],
    };
    const finalDescribeReturn = {
      Stacks: [
        {
          StackStatus: 'UNUSUAL_STATUS',
        },
      ],
    };

    describeStub.onCall(0).returns(BbPromise.resolve(DescribeReturn));
    describeStub.onCall(1).returns(BbPromise.resolve(DescribeReturn));
    describeStub.onCall(2).returns(BbPromise.resolve(finalDescribeReturn));

    return awsDeploy.monitorUpdate(cfDataMock, 10).catch((e) => {
      expect(e.name).to.be.equal('ServerlessError');
      expect(describeStub.callCount).to.be.equal(3);
      awsDeploy.CloudFormation.describeStacksPromised.restore();
    });
  });
});

describe('#updateStack()', () => {
  it('should run promise chain in order', () => {
    const updateStub = sinon
      .stub(awsDeploy, 'update').returns(BbPromise.resolve());
    const monitorStub = sinon
      .stub(awsDeploy, 'monitorUpdate').returns(BbPromise.resolve());

    return awsDeploy.updateStack().then(() => {
      expect(updateStub.calledOnce).to.be.equal(true);
      expect(monitorStub.calledAfter(updateStub)).to.be.equal(true);

      awsDeploy.update.restore();
      awsDeploy.monitorUpdate.restore();
    });
  });
});