serverless/test/unit/lib/cli/conditionally-load-dotenv.test.js

'use strict';

const chai = require('chai');
const path = require('path');
const overrideEnv = require('process-utils/override-env');
const fs = require('fs');
const conditionallyLoadDotenv = require('../../../../lib/cli/conditionally-load-dotenv');

chai.use(require('chai-as-promised'));
const expect = require('chai').expect;

describe('test/unit/lib/cli/conditionally-load-dotenv.test.js', () => {
  let restoreEnv;

  before(async () => {
    const defaultFileContent = 'DEFAULT_ENV_VARIABLE=valuefromdefault';
    await fs.promises.writeFile(path.resolve('.env'), defaultFileContent);

    const stage = 'testing';
    const stageFileContent = 'STAGE_ENV_VARIABLE=valuefromstage';
    await fs.promises.writeFile(path.resolve(`.env.${stage}`), stageFileContent);
  });

  beforeEach(() => {
    conditionallyLoadDotenv.clear();
    ({ restoreEnv } = overrideEnv());
  });
  afterEach(() => {
    restoreEnv && restoreEnv();
  });

  it('should load environment variables from default .env file if no matching stage', async () => {
    await conditionallyLoadDotenv({}, { useDotenv: true });

    expect(process.env.DEFAULT_ENV_VARIABLE).to.equal('valuefromdefault');
    expect(process.env.STAGE_ENV_VARIABLE).to.be.undefined;
  });

  it('should load environment variables from stage .env file if matching stage', async () => {
    await conditionallyLoadDotenv({ stage: 'testing' }, { useDotenv: true });
    expect(process.env.DEFAULT_ENV_VARIABLE).to.be.undefined;
    expect(process.env.STAGE_ENV_VARIABLE).to.equal('valuefromstage');
  });
});