Jest Mongoose 多个连接 - JavaScript

在使用 Jest 编写 Mongoose 测试时，如果测试涉及多个数据库连接，则需要一些特殊配置。这篇文章将介绍如何在 Jest 中使用 Mongoose 连接多个数据库。

需求

我们假设有一个应用程序，需要连接两个 MongoDB 数据库。第一个数据库使用 mongodb://localhost/test1 URL，第二个数据库使用 mongodb://localhost/test2 URL。我们需要编写 Jest 测试来测试在这两个数据库中保存数据的功能。

安装依赖

首先，我们需要安装必要的依赖：

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

其中：

• jest 是测试框架；
• mongodb-memory-server 是一个 MongoDB 内存实例，用于 Jest 测试；
• mongoose 是 MongoDB 的 ODM。

编写测试用例

我们写两个测试用例来测试在两个数据库中保存数据的功能。

## 保存数据到 test1 数据库

### 步骤：

1. 连接 test1 数据库。
2. 定义 Mongoose 模型。
3. 创建模型实例。
4. 将模型实例保存到 test1 数据库。

### 期望：

模型实例在 test1 数据库中保存成功。

```javascript
const mongoose = require('mongoose');

describe('test1 数据库', () => {
  let connection;
  let Test1Model;

  beforeAll(async () => {
    const url = 'mongodb://localhost/test1';
    connection = await mongoose.createConnection(url, {
      useNewUrlParser: true,
      useUnifiedTopology: true,
    });
    Test1Model = connection.model('Test1', { name: String });
  });

  afterAll(async () => {
    await connection.close();
  });

  test('保存数据到 Test1Model', async () => {
    const data = new Test1Model({ name: 'test' });
    await data.save();

    const result = await Test1Model.findOne({ name: 'test' }).exec();
    expect(result).toBeTruthy();
  });
});

## 保存数据到 test2 数据库

### 步骤：

1. 连接 test2 数据库。
2. 定义 Mongoose 模型。
3. 创建模型实例。
4. 将模型实例保存到 test2 数据库。

### 期望：

模型实例在 test2 数据库中保存成功。

```javascript
describe('test2 数据库', () => {
  let connection;
  let Test2Model;

  beforeAll(async () => {
    const url = 'mongodb://localhost/test2';
    connection = await mongoose.createConnection(url, {
      useNewUrlParser: true,
      useUnifiedTopology: true,
    });
    Test2Model = connection.model('Test2', { name: String });
  });

  afterAll(async () => {
    await connection.close();
  });

  test('保存数据到 Test2Model', async () => {
    const data = new Test2Model({ name: 'test' });
    await data.save();

    const result = await Test2Model.findOne({ name: 'test' }).exec();
    expect(result).toBeTruthy();
  });
});

运行测试用例

如果两个测试用例在同一个测试文件中，可以使用以下命令运行测试：

npx jest --detectOpenHandles

如果两个测试用例在不同的测试文件中，需要将以下配置添加到 Jest 配置文件中（比如 jest.config.js）：

module.exports = {
  testEnvironment: 'node',
  globalSetup: './jest.setup.js',
  globalTeardown: './jest.teardown.js',
};

然后在 jest.setup.js 文件中初始化 MongoDB 内存实例：

const { MongoMemoryServer } = require('mongodb-memory-server');
const mongoose = require('mongoose');

const mongod = new MongoMemoryServer();

module.exports = async () => {
  jest.setTimeout(30000);

  const uri = await mongod.getConnectionString();

  await mongoose.connect(uri, {
    useNewUrlParser: true,
    useUnifiedTopology: true,
  });
};

在 jest.teardown.js 文件中关闭 MongoDB 内存实例：

const mongoose = require('mongoose');

module.exports = async () => {
  await mongoose.disconnect();
  await mongod.stop();
};

然后使用以下命令运行测试：

npx jest --config=jest.config.js --detectOpenHandles

结论

在 Jest 中使用 Mongoose 连接多个数据库需要进行一些特殊配置。本文提供了使用 MongoDB 内存实例在 Jest 中连接多个数据库的方案。