javascript - 测试nodejs和谐生成器方法

Question

假设你有以下 JS 函数：

function YourProxy($orm, $usr) {

     this.addToDB = function(obj) {
     /* Do some validation on obj */
        return function(callback){
            var oo = $orm.createNew(obj);
            oo.save(options, function(err,ok){
                if(err) callback(err);
                callback(null,ok);
            }
        }
     }
}

您可以在带有 ES6 生成器的 node.js 上使用它来等待该操作发生，例如：

function *(){
    var yourProxy = new YourProxy();
    try {
        var result = yield yourProxy.addToDB(anObject);
    } catch(e) { 
        /* Something went wrong sync. Here you have err from save's callback */
    }
    /* result contains ok, the one from save's callback */
}

为了测试我做了这样的事情，使用 mocha 和 sinon（和 mocha-sinon）：

describe('addToDB', function(){
    it('adds the object to the db', function(){
        var callback = sinon.spy();
        myProxy.addToDB(anObject)(callback);
        expect( callback ).to.be.calledOnce;
    });
 });

但我得到的只是回调永远不会被调用，因为 addToDB() 在调用 save 的回调之前退出。

你将如何测试它？

Answer 1

尝试co-mocha像在示例中那样使用并生成生成器。

describe('addToDB', function(){
    it('adds the object to the db', function* (){
        var callback = sinon.spy();
        yield myProxy.addToDB(anObject)(callback);
        expect( callback ).to.be.calledOnce;
    });
 });