如何使用 Jest 测试 mongoose.connect 回调方法

Question

我已经开始制作一个新的 Express 应用程序，选择的测试框架是 Jest。

我设法涵盖了每一行代码，除了mongoose.connect方法的回调：

我试过窥探对象的connect方法mongoose，所以我可以指定返回的对象，但无济于事。

有没有办法测试 的回调mongoose.connect？

Answer 1

这是解决方案，您的代码不容易测试，因此我进行了一些重构。

index.ts：

import mongoose, { Mongoose } from 'mongoose';
import { MongoError } from 'mongodb';

function callback(err?: MongoError) {
  if (err) {
    console.log(err.message);
  } else {
    console.log('Succesfully Connected!');
  }
}

function connectDatabase(): Promise<Mongoose> {
  const mongoUrl = 'localhost';

  return mongoose.connect(mongoUrl, { useCreateIndex: true, useNewUrlParser: true }, exports.callback);
}

exports.callback = callback;
exports.connectDatabase = connectDatabase;

单元测试：

import mongoose, { Mongoose, ConnectionOptions } from 'mongoose';
import { MongoError } from 'mongodb';

jest.mock('mongoose');

describe('connectDatabase', () => {
  const dbModule = require('./');

  it('should connect database succesfully', done => {
    const consoleLogSpyOn = jest.spyOn(console, 'log');
    const mongooseConnectSpyOn = jest
      .spyOn<Mongoose, 'connect'>(mongoose, 'connect')
      .mockImplementationOnce((uris: string, options?: ConnectionOptions, callback?: (err?: MongoError) => void) => {
        if (callback) {
          callback();
          done();
        }
        return Promise.resolve(mongoose);
      });

    dbModule.connectDatabase();
    expect(mongooseConnectSpyOn).toBeCalledWith(
      'localhost',
      { useCreateIndex: true, useNewUrlParser: true },
      dbModule.callback
    );
    expect(consoleLogSpyOn).toBeCalledWith('Succesfully Connected!');
    consoleLogSpyOn.mockRestore();
  });

  it('connect database error', done => {
    const consoleLogSpyOn = jest.spyOn(console, 'log');
    const mongooseConnectSpyOn = jest
      .spyOn<Mongoose, 'connect'>(mongoose, 'connect')
      .mockImplementationOnce((uris: string, options?: ConnectionOptions, callback?: (err?: MongoError) => void) => {
        if (callback) {
          callback(new Error('connect error'));
          done();
        }
        return Promise.resolve(mongoose);
      });

    dbModule.connectDatabase();
    expect(mongooseConnectSpyOn).toBeCalledWith(
      'localhost',
      { useCreateIndex: true, useNewUrlParser: true },
      dbModule.callback
    );
    expect(consoleLogSpyOn).toBeCalledWith('connect error');
    consoleLogSpyOn.mockRestore();
  });
});

单元测试结果和覆盖率报告：

 PASS  src/stackoverflow/56132437/index.spec.ts
  connectDatabase
    ? should connect database succesfully (12ms)
    ? connect database error (1ms)

  console.log node_modules/jest-mock/build/index.js:860
    Succesfully Connected!

  console.log node_modules/jest-mock/build/index.js:860
    connect error

----------|----------|----------|----------|----------|-------------------|
File      |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
----------|----------|----------|----------|----------|-------------------|
All files |      100 |      100 |      100 |      100 |                   |
 index.ts |      100 |      100 |      100 |      100 |                   |
----------|----------|----------|----------|----------|-------------------|
Test Suites: 1 passed, 1 total
Tests:       2 passed, 2 total
Snapshots:   0 total
Time:        1.853s, estimated 5s

这是完整的演示：https : //github.com/mrdulin/jest-codelab/blob/master/src/stackoverflow/56132437/index.spec.ts