Question

我在测试ldapjs客户端搜索操作时遇到问题。它返回一个EventEmitter，您必须让它侦听某些特定事件。我包装了此操作以证明它并定义我的逻辑，我想对其进行单元测试。

findUser(username) {
    return new Promise((resolve, reject) => {
        logger.debug('Searching user: ', username);
        this.ldapClient.bind(user.name, .user.password, err => {
            if (err) return reject(err);
            else
                this.ldapClient.search(root, {filter: `(cn=${username})`}, (errSearch, resSearch) => {
                    if (errSearch) return reject(errSearch);
                    const entries = [];
                    resSearch.on('searchEntry', entry => entries.push(entry.object));
                    resSearch.on('searchReference', referral => reject(new Error(`Received search referall: ${referral}`)));
                    resSearch.on('error', err => reject((err));
                    resSearch.on('end', result => {
                        if (result.status === 0 && entries.length === 1) {
                            return resolve({
                                cn: entries[0].cn,
                                objectclass: entries[0].objectclass,
                                password: entries[0].password
                            });
                        } else {
                            return reject(new Error(`Wrong search result: ${result}`));
                        }
                    });
                });
        });
    });
}

我正在使用mockery和Sinon替换模块内部的ldapjs依赖项：

beforeEach(function () {
    searchEM = new EventEmitter();
    sandbox = sinon.createSandbox();
    ldapClientStub = Stubs.getLdapClientStub(sandbox);
    ldapClientStub.bind.yields(null);
    ldapClientStub.search.withArgs('o=ldap', {filter: `(cn=${findParam})`}).yields(null, searchEM);

    mockery.registerMock('ldapjs', Stubs.getLdapStub(ldapClientStub));
    mockery.registerAllowable('../src/client');

    UserClientCls = require('../src/client').default;
    userClient = new UserClientCls(config.get());
});

it('should return user with given username', function (done) {
    setTimeout(() => {
        searchEM.emit('searchEntry', users[1]);
        searchEM.emit('end', {status: 0});
        console.log('emitted');
    }, 500);
    searchEM.on('end', res => console.log(res));

    userClient.findUser(findParam)
        .then(user => {
            user.cn.should.equal(users[1].attributes.cn);
            user.objectclass.should.equal(users[1].attributes.objectclass);
            user.password.should.equal(users[1].attributes.password);
            return done();
        })
        .catch(err => done(err));
});

问题在于findUser中定义的侦听器永远不会被调用（但函数本身会被调用）。我在测试中定义的侦听器（只是为了调试行为）被正确调用。

我不知道我是否错过了一些有关EventEmitters工作原理的信息，或者我是否以错误的方式进行了测试。也许我写了一段不好的代码，无法测试。

Answer 1

我找到了解决问题的方法。我扩展了基本的EventEmitter：我添加了存储要发出的事件的逻辑，并使用了发出假事件的逻辑来覆盖其on方法。

class TestEventEmitter extends EventEmitter {
    constructor() {
        super();
    }

    setFakeEmit(fakeEmit) {
        this.fakeEmit = fakeEmit;
    }

    on(eventName, cb) {
        super.on(eventName, cb);
        if (super.eventNames().length === 4)
            this.fakeEmit.forEach(f => this.emit(f.name, f.obj));
    }
}

因此，在beforeEach中，我可以对ldapClientStub.search进行存根以使其返回我的TestEventEmitter：

beforeEach(function() {
    searchEM = new TestEventEmitter();
    searchEM.setFakeEmit([{
        name: 'searchEntry',
        obj: { object: users[1].attributes }
    }, {
        name: 'end',
        obj: { status: 0 }
    }]);

    ...

    ldapClientStub.search.withArgs('o=ldap', { filter: `(&(cn=${findParam})(objectclass=astsUser))` }).yields(null, searchEM);
})

此解决方案可能不是很好，但它可以工作。如果有人可以发布更好的解决方案，我将很高兴看看。

使用Mocha进行NodeJS EventEmitter测试

1 个答案: