lambda-tdd/lib/modules/handler-executor.js

Test Framework for AWS Lambda

import wrapper from 'lambda-wrapper';
import { RequestRecorder } from 'node-tdd';
import rewriteObject from '../util/rewrite-object.js';

export default async (options) => {
  const handlers = await import(options.handlerFile);
  const runner = wrapper.wrap({ handler: handlers[options.handlerFunction] });
  const requestRecorder = RequestRecorder({
    cassetteFolder: options.cassetteFolder,
    stripHeaders: options.stripHeaders || false,
    strict: false,
    heal: options.nockHeal,
    modifiers: options.modifiers,
    reqHeaderOverwrite: options.reqHeaderOverwrite
  });
  await requestRecorder.inject(options.cassetteFile);

  const startTimestamp = process.hrtime();
  const startTime = (startTimestamp[0] * 1000) + (startTimestamp[1] / 1000000);
  const event = rewriteObject(options.event, options.modifiers);
  const [err, response] = await new Promise((resolve) => {
    runner.run(event, {
      ...options.context,
      getRemainingTimeInMillis: () => {
        const curTimeStamp = process.hrtime();
        const curTime = (curTimeStamp[0] * 1000) + (curTimeStamp[1] / 1000000);
        return (options.lambdaTimeout || 300000) - (curTime - startTime);
      }
    }, (...args) => resolve(args));
  });
  await requestRecorder.release();
  return { ...requestRecorder.get(), err, response };
};