如何对使用MongoDB的NodeJS代码进行隔离单元测试？

下午好！针对你提到的NodeJS + MongoDB代码的单元测试问题，我来分享几个靠谱的隔离测试方案，完全不需要启动真实数据库～

这种方式通过Mock mongodb库的核心方法（比如MongoClient.connect），模拟数据库连接的成功/失败场景，完全不需要接触真实数据库。常用的Mock工具可以用sinon或者Jest自带的Mock功能，下面以Jest为例演示：

步骤1：安装测试依赖 #

npm install jest --save-dev

步骤2：编写测试用例 #

假设你的DatabaseService.js完整代码如下（补全了connect方法）：

var mongodb = require('mongodb');
var exports = module.exports = (function () {
  var MongoClient = mongodb.MongoClient;
  var db = null;
  function getDB(){ return db; }
  const connect = (options) => {
    return MongoClient.connect(options.url)
      .then(client => {
        db = client.db(options.dbName);
        return db;
      });
  };

  return {
    connect,
    getDB
  };
})();

对应的测试文件DatabaseService.test.js：

const DatabaseService = require('./DatabaseService');

describe('DatabaseService', () => {
  // 每次测试前重置Mock和服务状态
  beforeEach(() => {
    jest.resetModules();
    // Mock MongoDB的MongoClient
    jest.mock('mongodb', () => ({
      MongoClient: {
        connect: jest.fn()
      }
    }));
  });

  test('connect 成功时应正确初始化db实例', async () => {
    const mockDb = { name: 'testDB' };
    const mockClient = { db: jest.fn().mockReturnValue(mockDb) };
    // 模拟connect成功返回client
    require('mongodb').MongoClient.connect.mockResolvedValue(mockClient);

    const options = { url: 'mongodb://localhost:27017', dbName: 'testDB' };
    await DatabaseService.connect(options);

    // 验证connect方法被正确调用
    expect(require('mongodb').MongoClient.connect).toHaveBeenCalledWith(options.url);
    // 验证db被正确赋值
    expect(DatabaseService.getDB()).toBe(mockDb);
  });

  test('connect 失败时应抛出错误且db保持为null', async () => {
    const mockError = new Error('数据库连接失败');
    // 模拟connect失败抛出错误
    require('mongodb').MongoClient.connect.mockRejectedValue(mockError);

    const options = { url: 'mongodb://localhost:27017', dbName: 'testDB' };
    // 验证抛出错误
    await expect(DatabaseService.connect(options)).rejects.toThrow(mockError);
    // 验证db未被赋值
    expect(DatabaseService.getDB()).toBeNull();
  });
});

如果你的代码耦合度较高（比如硬依赖mongodb库），可以通过依赖注入重构代码，让测试时能轻松替换真实的MongoClient为Mock对象，代码的可测试性会大幅提升：

重构后的DatabaseService.js #

var mongodb = require('mongodb');
// 允许外部传入MongoClient，默认使用真实的mongodb客户端
var exports = module.exports = (function (MongoClient = mongodb.MongoClient) {
  var db = null;
  function getDB(){ return db; }
  const connect = (options) => {
    return MongoClient.connect(options.url)
      .then(client => {
        db = client.db(options.dbName);
        return db;
      });
  };

  return {
    connect,
    getDB
  };
})();

重构后的测试用例 #

const DatabaseService = require('./DatabaseService');

describe('DatabaseService', () => {
  test('使用Mock MongoClient测试connect成功场景', async () => {
    const mockDb = { name: 'testDB' };
    const mockMongoClient = {
      connect: jest.fn().mockResolvedValue({ db: jest.fn().mockReturnValue(mockDb) })
    };
    // 传入Mock的MongoClient初始化服务
    const service = require('./DatabaseService')(mockMongoClient);

    const options = { url: 'mongodb://localhost:27017', dbName: 'testDB' };
    await service.connect(options);

    expect(mockMongoClient.connect).toHaveBeenCalledWith(options.url);
    expect(service.getDB()).toBe(mockDb);
  });
});

这种方式的好处是不需要全局Mock模块，测试更加灵活，代码的可维护性也更好。

如果你想验证真实的数据库操作逻辑（比如插入、查询、更新），但又不想依赖真实的MongoDB实例，可以用mongodb-memory-server——它会在内存中启动一个临时的MongoDB服务，测试结束后自动销毁，非常适合集成测试：

步骤1：安装依赖 #

npm install mongodb-memory-server jest --save-dev

步骤2：编写测试用例 #

const { MongoMemoryServer } = require('mongodb-memory-server');
const DatabaseService = require('./DatabaseService');

describe('DatabaseService 集成测试', () => {
  let mongoServer;
  let testDbUrl;

  // 测试前启动内存MongoDB
  beforeAll(async () => {
    mongoServer = await MongoMemoryServer.create();
    testDbUrl = mongoServer.getUri();
  });

  // 测试后停止内存MongoDB
  afterAll(async () => {
    await mongoServer.stop();
  });

  test('连接内存数据库并执行插入查询操作', async () => {
    await DatabaseService.connect({ url: testDbUrl, dbName: 'testDB' });
    const db = DatabaseService.getDB();

    // 插入测试数据
    const testCollection = db.collection('users');
    await testCollection.insertOne({ name: '张三', age: 25 });

    // 查询验证
    const user = await testCollection.findOne({ name: '张三' });
    expect(user.age).toBe(25);
  });
});

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