使用mocha进行所有测试后删除数据库和关闭连接的位置

时间:2012-12-16 05:09:16

标签: unit-testing node.js mongoose mocha

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

以下是我的嵌套测试:

//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毫秒的超时

2 个答案:

答案 0 :(得分:22)

您可以在第一个after()之前定义“root”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
        });

        // ...
    });
});

From the docs

  

通过向done添加回调(通常名为it()},Mocha将知道它应该等待完成。

答案 1 :(得分:0)

我实施的有点不同。

  1. 我删除了"之前的所有文件" hook - 发现它比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();
        }
    });