occaecatidicta/packages/omelox-rpc/test/rpc-server/client/mock-ws-client.js

var sioClient = require'socket.io-client';
var EventEmitter = require('events').EventEmitter;
var util = require'util';
var utils = require('../../../lib/util/utils');

var Client = function() {
  EventEmitter.call(this);
  this.requests = {};
  this.curId = 0;
};
util.inherits(Client, EventEmitter);

var pro = Client.prototype;

pro.connect = function(host, port, cb) {
  this.socket = sioClient.connect(host + ':' + port, {'force new connection': true});

  var self = this;
  this.socket.on('message', function(pkg) {
    var cb = self.requests[pkg.id];
    delete self.requests[pkg.id];

    if(!cb) {
      return;
    }

    cb.apply(null, pkg.resp);
  });

  this.socket.on('connect', function() {
    utils.invokeCallback(cb);
  });
};

pro.send = function(msg, cb) {
  var id = this.curId++;
  this.requests[id] = cb;
  this.socket.emit('message', {id: id, msg: msg});
};

pro.close = function() {
  this.socket.disconnect();
};

module.exports.create = function(opts) {
  return new Client();
};