unit-testing - 使用 mocha 进行所有测试后在哪里删除数据库并关闭连接

Question

我试图找出在所有测试运行后将删除数据库和关闭连接的函数放在哪里。

这是我的嵌套测试：

//db.connection.db.dropDatabase();
//db.connection.close();

describe('User', function(){
    beforeEach(function(done){
    });

    after(function(done){
    });

    describe('#save()', function(){
        beforeEach(function(done){
        });

        it('should have username property', function(done){
            user.save(function(err, user){
                done();
            });
        });

        // now try a negative test
        it('should not save if username is not present', function(done){
            user.save(function(err, user){
                done();
            });
        });
    });

    describe('#find()', function(){
        beforeEach(function(done){
            user.save(function(err, user){
                done();
            });
        });

        it('should find user by email', function(done){
            User.findOne({email: fakeUser.email}, function(err, user){
                done();
            });
        });

        it('should find user by username', function(done){
            User.findOne({username: fakeUser.username}, function(err, user){
                done();
            });
        });
    });
});

似乎没有任何效果。我得到错误：超过 2000 毫秒的超时

Answer 1

after()您可以在第一个describe()处理清理之前定义一个“根”钩子：

after(function (done) {
    db.connection.db.dropDatabase(function () {
        db.connection.close(function () {
            done();
        });
    });
});

describe('User', ...);

不过，您得到的错误可能来自 3 个异步钩子，它们没有通知 Mocha 继续。这些需要调用done()或跳过参数，以便它们可以被视为同步：

describe('User', function(){
    beforeEach(function(done){ // arg = asynchronous
        done();
    });

    after(function(done){
        done()
    });

    describe('#save()', function(){
        beforeEach(function(){ // no arg = synchronous
        });

        // ...
    });
});

从文档：

通过向 Mocha 添加一个回调（通常命名为done）it()就会知道它应该等待完成。

Answer 2

我实现它有点不同。

1. 我删除了“之前”钩子中的所有文档 - 发现它比 dropDatabase() 快得多。

2. 我使用 Promise.all() 确保在退出钩子之前删除了所有文档。

   beforeEach(function (done) {
   
       function clearDB() {
           var promises = [
               Model1.remove().exec(),
               Model2.remove().exec(),
               Model3.remove().exec()
           ];
   
           Promise.all(promises)
               .then(function () {
                   done();
               })
       }
   
       if (mongoose.connection.readyState === 0) {
           mongoose.connect(config.dbUrl, function (err) {
               if (err) {
                   throw err;
               }
               return clearDB();
           });
       } else {
           return clearDB();
       }
   });