MongoDB连接与集合删除实现及Jest单元测试Mock问题

一、全局state单例方案的评价与优化建议 #

先聊聊你最开始用全局state存储数据库连接的方案：

• 优点：实现了单例模式，避免重复创建MongoDB连接，逻辑简单直接，适合小型项目或简单场景。
• 存在的问题：
  1. 测试污染风险：全局变量在测试用例之间是共享的，如果你在一个测试里初始化了state.db，下一个测试如果不重置，就会继承这个状态，导致测试结果不可靠。
  2. 缺乏连接状态的异步处理：当前connection()函数是同步的，但Monk的连接其实是异步建立的（虽然Monk实例会立即返回，但连接成功/失败是后续的），如果在连接还没建立时调用dropDB()，可能会出现意外错误。
  3. 扩展性差：如果后续需要切换数据库、多实例连接，全局单例的结构会变得难以维护。

优化建议：

• 如果你想保留单例模式，可以把connection()改成异步函数，等待连接就绪后再返回实例；同时给测试提供一个重置state的方法，方便测试用例之间隔离。
• 更推荐你后来用类的方式重构，类实例化的方式更灵活，每个实例可以管理自己的连接，测试时也更容易mock和隔离。

二、初始全局state方案的Jest测试修复 #

针对你第一个测试文件里的问题：如何mockremove()并初始化state.db，可以这样修改测试代码：

import monk from 'monk'
import { connection, dropDB } from './db'

// 先mock monk模块，替换成我们自己的mock实现
jest.mock('monk', () => {
  // 定义mock的remove方法
  const mockRemove = jest.fn(() => Promise.resolve({ n: 1, nRemoved: 1, ok: 1 }))
  // mock get方法，返回包含remove的对象
  const mockGet = jest.fn(() => ({ remove: mockRemove }))
  // mock monk构造函数，返回包含get方法的对象
  return jest.fn(() => ({ get: mockGet }))
})

describe('dropDB()', () => {
  // 每个测试前重置mock和state
  beforeEach(() => {
    jest.clearAllMocks()
    // 重置全局state.db
    require('./db').state.db = null
  })

  test('should throw error if db connection is missing', async () => {
    expect.assertions(1)
    await expect(dropDB()).rejects.toEqual(Error('Missing database connection'))
  })

  test('should call remove() methods twice', async () => {
    // 调用connection初始化state.db
    connection()
    // 获取mock的remove方法（从monk的mock里取）
    const mockRemove = monk().get().remove
    // 执行dropDB
    await dropDB()
    // 验证调用次数
    expect(mockRemove).toHaveBeenCalledTimes(2)
    // 还可以验证调用参数
    expect(mockRemove).toHaveBeenCalledWith({})
  })
})

关键说明：

1. 我们直接在jest.mock('monk')里定义完整的mock实现，让Monk实例的get()方法返回带有mockremove的对象。
2. 每个测试前重置mock和全局state，避免测试之间的污染。
3. 调用connection()后，state.db就会被设置为我们mock的Monk实例，后续dropDB()调用时就会使用这个mock实例的get()和remove()。

三、类重构后的Jest Mock实现 #

针对你用Db类重构后的测试问题，需要mockmonk模块，并让类的connection属性返回带有mockget()的对象，具体代码如下：

import monk from 'monk'
import { Db } from './db'

// mock monk模块
jest.mock('monk', () => {
  const mockRemove = jest.fn(() => Promise.resolve({ n: 1, nRemoved: 1, ok: 1 }))
  const mockGet = jest.fn(() => ({ remove: mockRemove }))
  return jest.fn(() => ({ get: mockGet }))
})

describe('DB class', () => {
  let dbInstance
  let mockRemove

  beforeEach(() => {
    jest.clearAllMocks()
    // 获取mock的remove方法
    mockRemove = monk().get().remove
    // 创建Db实例
    dbInstance = new Db()
  })

  describe('dropDB()', () => {
    test('should call remove method twice', async () => {
      await dbInstance.dropDB()
      // 验证remove被调用两次
      expect(mockRemove).toHaveBeenCalledTimes(2)
      // 验证每次调用的参数是空对象
      expect(mockRemove).toHaveBeenCalledWith({})
      // 验证get方法被调用了两次，分别是'users'和'content'
      expect(monk().get).toHaveBeenCalledTimes(2)
      expect(monk().get).toHaveBeenNthCalledWith(1, 'users')
      expect(monk().get).toHaveBeenNthCalledWith(2, 'content')
    })
  })
})

关键说明：

1. 不需要mockDb类本身，而是mock它依赖的monk模块——因为我们要测试的是Db类的逻辑，而不是Monk的功能。
2. 通过mockmonk的返回值，让this.connection（也就是Monk实例）的get()方法返回带有mockremove的对象，这样当dropDB()调用Users.remove()和Content.remove()时，实际调用的是我们的mock方法。
3. 每个测试前重置mock，保证测试独立性。

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