serverless/test/utils/misc.js

'use strict';

const awsRequest = require('@serverless/test/aws-request');
const CloudWatchLogsService = require('aws-sdk').CloudWatchLogs;
const wait = require('timers-ext/promise/sleep');

const logger = console;

function replaceEnv(values) {
  const originals = {};
  for (const key of Object.keys(values)) {
    if (process.env[key]) {
      originals[key] = process.env[key];
    } else {
      originals[key] = 'undefined';
    }
    if (values[key] === 'undefined') {
      delete process.env[key];
    } else {
      process.env[key] = values[key];
    }
  }
  return originals;
}

/**
 * Cloudwatch logs when turned on, are usually take some time for being effective
 * This function allows to confirm that new setting (turned on cloudwatch logs)
 * is effective after stack deployment
 */
async function confirmCloudWatchLogs(logGroupName, trigger, options = {}) {
  const startTime = Date.now();
  const timeout = options.timeout || 3 * 60 * 1000;
  return trigger()
    .then(() => wait(1000))
    .then(() => awsRequest(CloudWatchLogsService, 'filterLogEvents', { logGroupName }))
    .then(({ events }) => {
      if (events.length) {
        if (options.checkIsComplete) {
          if (options.checkIsComplete(events)) return events;
        } else {
          return events;
        }
      }
      const duration = Date.now() - startTime;
      if (duration > timeout) throw new Error('Log items not found');
      return confirmCloudWatchLogs(
        logGroupName,
        trigger,
        Object.assign({}, options, { timeout: timeout - duration })
      );
    });
}

module.exports = {
  confirmCloudWatchLogs,
  logger,
  replaceEnv,
};