如何在测试前清理基于Mongoose的服务状态？

我正在为一个简单的基于Mongoose的服务编写测试代码：

describe('users service', () => { 
  const service = app.service('users'); 
  it('registered the service', () => { 
    assert.ok(service, 'Registered the service'); 
  }); 
});

问题在于，如何在每个测试前清理数据？我目前的实现如下：

// Reset service state before each test.
beforeEach((done) => { 
  service.find().then((items) => { 
    items.data.forEach((item) => { 
      service.remove(item._id) 
    }); 
    done(); 
  }) 
});

这种方式十分繁琐，是否有更优的实现方案？

几种更高效的优化方案 #

1. 直接用Mongoose模型批量删除 #

既然是基于Mongoose的服务，跳过服务层直接操作对应的User模型会高效很多。deleteMany()方法可以一次性清空整个集合，省去先查询再逐个删除的冗余步骤：

const User = require('../models/user'); // 引入你的User模型

beforeEach(async () => {
  // 清空users集合的所有文档
  await User.deleteMany({});
});

这里用async/await替代了回调函数done，代码更简洁，也避免了异步操作未完成就执行测试的问题。

2. 利用MongoDB事务回滚（MongoDB 4.0+支持） #

如果你的测试环境使用的是MongoDB 4.0及以上版本，可以用事务来自动回滚测试数据，根本不需要手动清理：

let session;

beforeEach(async () => {
  // 开启新会话和事务
  session = await mongoose.startSession();
  session.startTransaction();
  
  // 让User模型的所有操作都绑定这个会话
  User.createCollection({ session });
});

afterEach(async () => {
  // 回滚事务，丢弃所有测试产生的数据
  await session.abortTransaction();
  session.endSession();
});

这种方式的优势是测试数据完全不会污染数据库，适合复杂的测试场景，也省去了清理代码的维护成本。

3. 为测试单独配置数据库并重置 #

你可以给测试环境单独配置一个专用数据库，在每个测试前直接删除整个数据库再重建：

beforeEach(async () => {
  // 删除当前测试数据库
  await mongoose.connection.dropDatabase();
});

这个方案最彻底，适合小型项目或者独立的测试环境，但一定要注意配置好测试数据库，绝对不能误操作生产库！

4. 修复现有代码的异步问题（如果坚持用服务层） #

如果必须通过服务层的方法来清理，那首先要解决你当前代码的bug：forEach里的service.remove是异步操作，但你直接调用done()会导致删除还没完成就开始测试，数据清理不彻底。应该用Promise.all等待所有删除操作完成：

beforeEach(async () => {
  const { data } = await service.find();
  // 等待所有删除请求完成
  await Promise.all(data.map(item => service.remove(item._id)));
});

这样能保证所有数据都被清理后才进入测试，同时代码也比原来的回调写法更清晰。

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