如何在测试中让await语句按条件执行？适配生产异步测试同步场景

这场景我太熟了！生产环境里异步发邮件不阻塞主流程是完全合理的，但测试时必须等它跑完才能验证结果——不然你根本没法确认邮件到底发没发，整个流程是不是真的按预期走完了。给你几个实用的解决方案，按灵活性排序：

方案一：依赖注入（最推荐） #

核心思路是把邮件发送的逻辑抽成独立的服务，根据环境切换实现：生产用“发完就不管”的异步版本，测试用需要等待完成的版本（甚至可以直接mock，不用真发邮件）。

举个JavaScript的例子：

// 生产环境邮件服务：异步发送，不阻塞主流程
class ProductionMailer {
  sendEmail(payload) {
    // 异步请求，不等待结果
    fetch('/mail/send', {
      method: 'POST',
      body: JSON.stringify(payload)
    }).catch(err => console.error('邮件发送失败:', err));
    
    // 立即返回resolved的Promise，让主流程继续
    return Promise.resolve();
  }
}

// 测试环境邮件服务：等待发送完成，还能记录发送内容方便断言
class TestMailer {
  constructor() {
    this.sentEmails = []; // 存储所有发送的邮件，供测试验证
  }

  async sendEmail(payload) {
    // 等待发送完成（或直接mock请求，不用真调用邮件API）
    const res = await fetch('/mail/send', {
      method: 'POST',
      body: JSON.stringify(payload)
    });

    if (!res.ok) throw new Error('邮件发送失败');
    
    this.sentEmails.push(payload);
    return Promise.resolve();
  }
}

// 根据环境注入对应的服务
const mailer = process.env.NODE_ENV === 'test' 
  ? new TestMailer() 
  : new ProductionMailer();

// 业务逻辑里调用，无需关心具体实现
async function updateObjectStatus(obj, targetStatus) {
  // 更新对象状态的核心逻辑...
  await mailer.sendEmail({
    to: obj.ownerEmail,
    subject: `对象状态已更新为${targetStatus}`
  });
}

这个方案的好处是完全解耦业务与邮件逻辑，测试时你甚至可以把TestMailer换成纯mock版本，不用真调用邮件API，直接验证调用次数、参数是否正确，测试速度快还不会乱发测试邮件。

方案二：环境变量直接控制等待逻辑 #

如果项目比较小，不想搞复杂的依赖注入，直接在原函数里加个环境判断就行：

async function sendEmail(payload) {
  const sendPromise = fetch('/mail/send', {
    method: 'POST',
    body: JSON.stringify(payload)
  }).catch(err => console.error('邮件发送失败:', err));

  // 测试环境强制等待发送完成，生产环境直接跳过
  if (process.env.NODE_ENV === 'test') {
    await sendPromise;
  }

  return;
}

// 业务逻辑调用方式不变
async function updateObjectStatus(obj, targetStatus) {
  // 更新状态逻辑...
  await sendEmail({ to: obj.ownerEmail, subject: '状态更新通知' });
}

这个方案简单直接，但耦合度比依赖注入高，适合快速解决问题的小型项目。

方案三：测试框架层面强制等待 #

如果你的测试框架支持（比如Jest、Mocha），也可以在测试用例里强制等待异步任务完成。不过前提是你的邮件函数在测试环境要返回Promise：

// 原邮件函数修改：测试环境返回发送Promise，生产环境直接resolve
function sendEmail(payload) {
  const sendPromise = fetch('/mail/send', {
    method: 'POST',
    body: JSON.stringify(payload)
  }).catch(err => console.error('邮件发送失败:', err));

  return process.env.NODE_ENV === 'test' ? sendPromise : Promise.resolve();
}

// 测试用例（以Jest为例）
test('更新对象状态后成功发送邮件', async () => {
  // Mock fetch，避免真发邮件
  jest.mock('node-fetch', () => jest.fn().mockResolvedValue({ ok: true }));
  
  const testObj = { ownerEmail: 'test@example.com' };
  await updateObjectStatus(testObj, 'completed');

  // 断言fetch是否被正确调用
  expect(fetch).toHaveBeenCalledWith('/mail/send', expect.objectContaining({
    body: JSON.stringify({
      to: 'test@example.com',
      subject: '状态更新通知'
    })
  }));
});

额外提示：测试时别真发邮件！ #

不管用哪个方案，测试阶段一定要mock邮件发送的API，不然每次跑测试都发一堆测试邮件，不仅慢还容易骚扰别人。用测试框架的mock功能就能轻松实现，比如Jest的jest.mock，或者Sinon的stub。

内容的提问来源于stack exchange，提问作者Stas Arshanski