0

我在尝试模拟lambda.invoke我从另一个 lambda 函数中调用的函数时遇到问题。

  • 函数被包装（我不能使用 sinon，它会告诉我它已经被包装了）。
  • 测试一直在 AWS 上调用 lambda 函数，而不是调用 mock。
  • 如果我使用 sinon 而不是 aws-sdk-mock，它会做同样的事情。

test.js

const { handler1 } = require('../handler');
const sinon = require('sinon');
const AWSMock = require('aws-sdk-mock');

describe('invoke', () => {
  beforeEach(() => {
    invokeMock = jest.fn();
    AWSMock.mock('Lambda', 'invoke', invokeMock);
    // const mLambda = { invoke: sinon.stub().returnsThis(), promise: sinon.stub() };
    // sinon.stub(AWS, 'Lambda').callsFake(() => mLambda);
  });

  afterEach(() => {
    AWSMock.restore();
    sinon.restore();
  });

  test('test1', async () => {
    const event = { test: 'ok'};
    const handler = await handler1(event);
    expect(handler.statusCode).toBe(204);
  });
});

我的 lambda 函数是：

handler.js

const AWS = require('aws-sdk');

module.exports.handler1 = (event) => {
  // The initialisation bellow has to be in the handler not outside.
  const lambda = new AWS.Lambda({
    region: 'us-west-2' //change to your region
  });
  let params = {
    InvocationType: 'Event',
    LogType: 'Tail',
    FunctionName: 'handler2', // the lambda function we are going to invoke
    Payload: JSON.stringify(event)
  };
  return new Promise((resolve, reject) => {
    lambda.invoke(params, function(error, data) {
      if(error) return reject(error);
      const payload = JSON.parse(data.Payload);
      if(!payload.success){
        return resolve({ statusCode: 400});
      }
      return resolve({ statusCode: 204});
    });
  });
};

编辑：所以我遇到的问题是因为我的 lambda 初始化 ( const lambda = new AWS.Lambda({})) 在处理程序之外而不是在内部。感谢 stijndepestel 的回答。

4

1 回答 1

1

从您共享的代码中并不完全清楚，但大概在您将函数包装到测试中之前，您在您的代码中有一个lambda引用handler.js。您可以在处理程序函数内而不是在它之外添加该const lambda = new AWS.Lamda({})行吗？

于 2021-08-04T08:36:20.827 回答