Question

这是我使用monk()连接到mongoDB的方式。我会将其存储在state。

假设我们要删除一些集合，我们称之为dropDB。

db.js

var state = {
  db: null
}

export function connection () {
  if (state.db) return
  state.db = monk('mongdb://localhost:27017/db')
  return state.db
}

export async function dropDB () {
  var db = state.db
  if (!db) throw Error('Missing database connection')

  const Users = db.get('users')
  const Content = db.get('content')

  await Users.remove({})
  await Content.remove({})
}

我不太确定使用state变量是否是一个好方法。也许有人可以对此发表评论或表现出改进。

现在我想使用JestJS为这个函数编写一个单元测试：

db.test.js

import monk from 'monk'
import { connection, dropDB } from './db'
jest.mock('monk')

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

这部分很简单，但下一部分给出了两个问题：

如何模拟remove()方法？

  test('should call remove() methods', async () => {
    connection() // should set `state.db`, but doesn't work
    const remove = jest.fn(() => Promise.resolve({ n: 1, nRemoved: 1, ok: 1 }))
    // How do I use this mocked remove()?
    expect(remove).toHaveBeenCalledTimes(2)
  })

在此之前？如何设置state.db？

更新

正如 poke 所解释的那样，全局变量会产生问题。所以我换了一个班：

db.js

export class Db {
  constructor() {
    this.connection = monk('mongdb://localhost:27017/db');
  }

  async dropDB() {
    const Users = this.connection.get('users');
    const Content = this.connection.get('content');

    await Users.remove({});
    await Content.remove({});
  }
}

导致此测试文件：

db.test.js

import { Db } from './db'
jest.mock('./db')

let db
let remove

describe('DB class', () => {
  beforeAll(() => {
    const remove = jest.fn(() => Promise.resolve({ n: 1, nRemoved: 1, ok: 1 }))
    Db.mockImplementation(() => {
      return { dropDB: () => {
        // Define this.connection.get() and use remove as a result of it
      } }
    })
  })
  describe('dropDB()', () => {
    test('should call remove method', () => {
      db = new Db()
      db.dropDB()
      expect(remove).toHaveBeenCalledTimes(2)
    })
  })
})

如何模拟任何this元素？在这种情况下，我需要模拟this.connection.get()

Answer 1

拥有全球状态绝对是您问题的根源。我建议寻找一个根本不涉及全局变量的解决方案。根据{{3}}，全局变量会导致紧密耦合并使测试难以测试（正如您已经注意到的那样）。

更好的解决方案是将数据库连接显式传递给dropDB函数，因此它具有作为显式依赖项的连接，或者引入一些保留的有状态对象连接并提供dropDB作为方法。

第一个选项如下：

export function openConnection() {
  return monk('mongdb://localhost:27017/db');
}

export async function dropDB(connection) {
  if (!connection) {
    throw Error('Missing database connection');
  }

  const Users = connection.get('users');
  const Content = connection.get('content');

  await Users.remove({});
  await Content.remove({});
}

这也可以让非常容易来测试dropDB，因为你现在可以直接为它传递一个模拟对象。

另一个选项可能如下所示：

export class Connection() {
  constructor() {
    this.connection = monk('mongdb://localhost:27017/db');
  }

  async dropDB() {
    const Users = this.connection.get('users');
    const Content = this.connection.get('content');

    await Users.remove({});
    await Content.remove({});
  }
}

第一个选项的测试可能如下所示：

test('should call remove() methods', async () => {
  const usersRemove = jest.fn().mockReturnValue(Promise.resolve(null));
  const contentRemove = jest.fn().mockReturnValue(Promise.resolve(null));
  const dbMock = {
    get(type) {
      if (type === 'users') {
        return { remove: usersRemove };
      }
      else if (type === 'content') {
        return { remove: contentRemove };
      }
    }
  };

  await dropDB(dbMock);

  expect(usersRemove).toHaveBeenCalledTimes(1);
  expect(contentRemove).toHaveBeenCalledTimes(1);
});

基本上，dropDB函数需要一个具有get方法的对象，该方法在调用时返回一个具有remove方法的对象。所以你只需要传递看起来像这样的的东西，这样函数就可以调用那些remove方法。

对于类，这有点复杂，因为构造函数依赖于monk模块。一种方法是再次显示该依赖关系（就像在第一个解决方案中一样），并在那里传递monk或其他工厂。但我们也可以使用Global Variables Are Bad来模拟整个monk模块。

请注意，我们不想要模拟包含Connection类型的模块。我们想测试一下，所以我们需要它处于未经模拟的状态。

要模拟monk，我们需要在__mocks__/monk.js创建一个模拟模块。该手册指出此__mocks__文件夹应与node_modules文件夹相邻。

在该文件中，我们只是导出自定义monk功能。这与我们在第一个示例中已经使用的几乎相同，因为我们只关心如何实现这些remove方法：

export default function mockedMonk (url) {
  return {
    get(type) {
      if (type === 'users') {
        return { remove: mockedMonk.usersRemove };
      }
      else if (type === 'content') {
        return { remove: mockedMonk.contentRemove };
      }
    }
  };
};

请注意，这会将功能称为mockedMonk.usersRemove和mockedMonk.contentRemove。我们将在测试中使用它在测试执行期间显式配置这些函数。

现在，在测试函数中，我们需要调用jest.mock('monk')以使Jest能够使用我们的模拟模块来模拟monk模块。然后，我们也可以导入它并在测试中设置我们的功能。基本上，就像上面一样：

import { Connection } from './db';
import monk from 'monk';

// enable mock
jest.mock('./monk');

test('should call remove() methods', async () => {
  monk.usersRemove = jest.fn().mockReturnValue(Promise.resolve(null));
  monk.contentRemove = jest.fn().mockReturnValue(Promise.resolve(null));

  const connection = new Connection();
  await connection.dropDB();

  expect(monk.usersRemove).toHaveBeenCalledTimes(1);
  expect(monk.contentRemove).toHaveBeenCalledTimes(1);
});

如何连接到mongoDB并测试drop collection？

1 个答案: