如何测试节点。JSWebSocket服务器？

Question

我使用标准配置的sockjs。

   var ws = sockjs.createServer();
    ws.on('connection', function(conn) {
        conn.on('data', function(message) {
            wsParser.parse(conn, message)
        });
        conn.on('close', function() {

        });
    });

    var server = http.createServer(app);
    ws.installHandlers(server, {prefix:'/ws'});
    server.listen(config.server.port, config.server.host);

wsParser。parse函数的工作原理如下：

function(conn, message) {

(...)

switch(message.action) {
    case "titleAutocomplete":
        titleAutocomplete(conn, message.data);
        break;
    (...) // a lot more of these
 }

交换机中调用的每个方法都会向客户端发回一条消息。

var titleAutocomplete = function(conn, data) {

    redis.hgetall("titles:"+data.query, function(err, titles){
        if(err) ERR(err);

        if(titles) {
            var response = JSON.stringify({"action": "titleAutocomplete", "data": {"titles": titles}});
            conn.write(response);
        } 
    })
};

现在我的问题是，我想对我的代码进行测试（我想晚做总比不做好），但我不知道如何做。我开始写正常的超文本传输协议测试在摩卡超级测试，但我只是不知道如何处理webSocket。

我希望通过所有测试只有一个webSocket连接可以重用，我在第一条消息后将webSocket连接与用户会话绑定，我也想测试这种持久性。

如何使用ws客户端的onMessage事件并在测试中使用它？测试如何区分收到的消息，并知道他们应该等待哪一个？

Answer 1

Collegue在工作中问，它是否真的需要一个客户端连接，或者是否可以只模拟它。事实证明，这是一条路要走。我编写了一个小助手类wsMockjs

var wsParser = require("../wsParser.js");

exports.createConnectionMock = function(id) {
    return {
        id: id,
        cb: null,
        write: function(message) {
            this.cb(message);
        },
        send: function(action, data, cb) {
            this.cb = cb;
            var obj = {
                action: action,
                data: data
            }
            var message = JSON.stringify(obj);
            wsParser.parse(this, message);
        },
        sendRaw: function(message, cb) {
            this.cb = cb;
            wsParser.parse(this, message);
        }
    }
}

现在，在我的摩卡咖啡测试中，我就是这么做的

var wsMock = require("./wsMock.js");
ws = wsMock.createConnectionMock("12345-67890-abcde-fghi-jklmn-opqrs-tuvwxyz");
(...)
describe('Websocket server', function () {

    it('should set sessionId variable after handshake', function (done) {
        ws.send('handshake', {token: data.token}, function (res) {
            var msg = JSON.parse(res);
            msg.action.should.equal('handshake');
            msg.data.should.be.empty;
            ws.should.have.property('sessionId');
            ws.should.not.have.property('session');
            done();
        })
    })

    it('should not return error when making request after handshake', function (done) {
        ws.send('titleAutocomplete', {query: "ter"}, function (res) {
            var msg = JSON.parse(res);
            msg.action.should.equal('titleAutocomplete');
            msg.data.should.be.an.Object;
            ws.should.have.property('session');
            done();
        })
    })
})

它就像一个符咒，在请求之间保持连接状态和变量。