serverless/lib/plugins/aws/package/lib/saveServiceState.test.js

'use strict';

const expect = require('chai').expect;
const sinon = require('sinon');
const path = require('path');
const AwsPackage = require('../index');
const Serverless = require('../../../../Serverless');
const AwsProvider = require('../../provider/awsProvider');

describe('#saveServiceState()', () => {
  let serverless;
  let awsPackage;
  let getServiceStateFileNameStub;
  let writeFileSyncStub;

  beforeEach(() => {
    const options = {};
    serverless = new Serverless();
    serverless.setProvider('aws', new AwsProvider(serverless, options));
    awsPackage = new AwsPackage(serverless, options);
    serverless.config.servicePath = 'my-service';
    serverless.service = {
      provider: {
        compiledCloudFormationTemplate: 'compiled content',
      },
      package: {
        individually: false,
        artifactDirectoryName: 'artifact-directory',
        artifact: 'service.zip',
      },
    };
    getServiceStateFileNameStub = sinon
      .stub(awsPackage.provider.naming, 'getServiceStateFileName')
      .returns('service-state.json');
    writeFileSyncStub = sinon.stub(awsPackage.serverless.utils, 'writeFileSync').returns();
  });

  afterEach(() => {
    awsPackage.provider.naming.getServiceStateFileName.restore();
    awsPackage.serverless.utils.writeFileSync.restore();
  });

  it('should write the service state file template to disk', () => {
    const filePath = path.join(
      awsPackage.serverless.config.servicePath,
      '.serverless',
      'service-state.json'
    );

    return awsPackage.saveServiceState().then(() => {
      const expectedStateFileContent = {
        service: {
          provider: {
            compiledCloudFormationTemplate: 'compiled content',
          },
        },
        package: {
          individually: false,
          artifactDirectoryName: 'artifact-directory',
          artifact: 'service.zip',
        },
      };

      expect(getServiceStateFileNameStub.calledOnce).to.equal(true);
      expect(
        writeFileSyncStub.calledWithExactly(filePath, expectedStateFileContent, true)
      ).to.equal(true);
    });
  });

  it('should remove self references correctly', () => {
    const filePath = path.join(
      awsPackage.serverless.config.servicePath,
      '.serverless',
      'service-state.json'
    );

    serverless.service.custom = {
      mySelfRef: serverless.service,
    };

    return awsPackage.saveServiceState().then(() => {
      const expectedStateFileContent = {
        service: {
          provider: {
            compiledCloudFormationTemplate: 'compiled content',
          },
          custom: {
            mySelfRef: '${self:}',
          },
        },
        package: {
          individually: false,
          artifactDirectoryName: 'artifact-directory',
          artifact: 'service.zip',
        },
      };

      expect(getServiceStateFileNameStub.calledOnce).to.equal(true);
      expect(
        writeFileSyncStub.calledWithExactly(filePath, expectedStateFileContent, true)
      ).to.equal(true);
    });
  });
});