使用mocha.js完成测试后关闭mongoos.js中的连接

const assert = require('assert');
const mongoose = require('mongoose');
const { describe, it, after } = require('mocha');

before((done) => {
  mongoose.connect('mongodb://localhost/test', { useNewUrlParser: true });
  const db = mongoose.connection;
  db.on('error', console.error.bind(console, 'connection error'));
  db.once('open', () => {
    console.log('Connected to the database');
    done();
  });
});

describe('Model', () => {
  it('should save a document', (done) => {
    const TestModel = mongoose.model('Test', new mongoose.Schema({ name: String }));
    const testDoc = new TestModel({ name: 'Test' });
    testDoc.save((err, doc) => {
      assert.equal(err, null);
      assert.equal(doc.name, 'Test');
      done();
    });
  });
});

after(() => {
  mongoose.connection.close(() => {
    console.log('Disconnected from the database');
  });
});