node.js - Jest Mock Promise with Params

Question

这是我试图用 Jest 编写单元测试的方法

async function getParameter(parameter: string,
  withDecryption: boolean = false): Promise<String> {

  const params = {
    Name: parameter,
    WithDecryption: withDecryption,
  };
  try {
    const request = await       ssmClient.getParameter(params).promise();

    return request.Parameter.Value;
  } catch (err) {
    logger.error(`Error ${err}`);
    throw Error(err);
  }
}

测试方法 ：

test('getParameterFromSystemManager', async () => {
  const mockedResponseData = {
    Parameter: {
      Value: 'parameterValue',
    },
  };

 ssmClient.getParameter(params).promise = jest.fn();

 ssmClient.getParameter(params).promise.mockImplementation(() => Promise.resolve(mockedResponseData));

 const data = 
 await SSMParameters.getParameterFromSystemManager('testurl', false, 
 'Test', 'elastic');
 expect(data).toEqual(mockedResponseData.Parameter.Value);
 expect(ssmClient.getParameter).toHaveBeenCalledTimes(1);
});

我得到错误：

TypeError: ssmClient.getParameter(...).promise.mockImplementation 不是函数

我们如何在 Jest 中模拟这样的 .promise() 函数？

Answer 1

不是函数的原因.promise.mockImplementation是调用getParameter(...)返回一个带有 new 的新实例/对象.promise()。

所以在你的第一行：ssmClient.getParameter(params).promise = jest.fn();你只用jest.fn().

在第二行中，您实际上是在调用.mockImplementation()一个全新的实例。

相反，您将不得不模拟getParameter()以确保它每次都返回您的模拟。

ssmClient.getParameter = jest.fn();
ssmClient.getParameter.mockImplementation(() => ({
    promise: jest.fn().mockImplementation(() => Promise.resolve(mockedResponseData));
}))