Jest - 如何在可用函数中模拟函数

Question

我在一个文件中有如下函数：

import logger from 'logger-module';
import spir from '../spir';

const spirA = new spir(
  process.env.USERNAME,
  process.env.PASSWORD,
  process.env.API_KEY,
  process.env.SMS_WORKSPACE_ID,
  process.env.SMS_TEMPLATE_ID
);

const spirSMS = params => {
  return new Promise((resolve, reject) => {
    spirA.sms(params).then(resolve, reject);
  });
};

export const send = params => {
  logger.info('Sending new sms...');
  const { to: recipients, invalidNumber } = params;
  const promises = recipients.map(number =>
   spirSMS({ to: number, subject: params.subject, body: params.body })
  );
  return Promise.all(promises).then(() => {
    logger.info(`SMS has been sent to ${recipients.toString()}!`);
    return {
      message: `SMS has been sent to ${recipients.toString()}`,
      data: {
        recipients,
        invalid_recipients: invalidNumber
      }
    };
  });
};

如何在 recipients.map

中模拟 spirSMS

我发现它是一个循环函数，所以我不确定如何模拟它。 非常感谢！

Answer 1

我可能会在这里使用类来更好地测试您的代码。有这样的想法

class SomeClass {
  constructor(spirSms) {
    this.spirSms_ = spirSms;
  }

  send(params) {
    logger.info('Sending new sms...');
    const { to: recipients, invalidNumber } = params;
    const promises = recipients.map(number =>
     this.spirSms_({ to: number, subject: params.subject, body: params.body })
    );
    return Promise.all(promises).then(() => {
      logger.info(`SMS has been sent to ${recipients.toString()}!`);
      return {
        message: `SMS has been sent to ${recipients.toString()}`,
        data: {
          recipients,
          invalid_recipients: invalidNumber
        }
      };
    });
  }
}

module.exports = SomeClass;

然后你会得到一些main.js，它将实例化你的类并注入spirSms

const spirA = new spir(
  process.env.USERNAME,
  process.env.PASSWORD,
  process.env.API_KEY,
  process.env.SMS_WORKSPACE_ID,
  process.env.SMS_TEMPLATE_ID
);

const spirSMS = params => {
  return new Promise((resolve, reject) => {
    spirA.sms(params).then(resolve, reject);
  });
};

const someClass = new SomeClass(spirSMS);
someClass.send(params);

现在在您测试SomeClass 时，您可以注入任何模拟spirSms，它基本上可以满足您的需求。

希望这会有所帮助。