vertex/test/unit/client/client.js

distributed pub/sub using hash rings

var path = require('path');
var filename = path.basename(__filename);
var expect = require('expect.js');
const EventEmitter = require('events').EventEmitter;
const Protocol = require('../../../lib/common/protocol');
const protocol = new Protocol();

xdescribe(filename, function () {

  it('initializes a client with a config', function (done) {

    var Client = require('../../..').Client;

    var config = {url: 'ws://127.0.0.1:3737'};

    var client = new Client(config);

    expect(client.config).to.eql({
      connectionAttemptInterval: 5000,
      connectionOptions: {},
      url: 'ws://127.0.0.1:3737',
      callTimeout: 120000
    });

    done();
  });

  class Connection extends EventEmitter {

    constructor() {
      super();
    }

    send(message) {

      if (message.action == 'subscribe') {

        var subscribeResponse = protocol.createReply(message, {status: 1});

        this.emit('message', subscribeResponse);
      }

      if (message.action == 'unsubscribe') {

        var unsubscribeResponse = protocol.createReply(message, {status: 1});

        this.emit('message', unsubscribeResponse);
      }

      if (message.action == 'publish') {

        var publishMessage = protocol.createMessage('pub', message.payload);

        this.emit('message', publishMessage);

        var publishResponse = protocol.createReply(message, {status: 1});

        this.emit('message', publishResponse);
      }
    }

    close(code, reason) {
      this.emit('close', {code: code, reason: reason});
    }

    connect(url, options) {
      return new Promise((resolve) => {
        resolve();
      });
    }
  }

  it('mocks the connection uses it to test the client connect', function (done) {

    this.timeout(10000);

    var Client = require('../../..').Client;

    var config = {url: 'ws://127.0.0.1:3737'};

    var client = new Client(config, {
      Connection: Connection
    });

    // client.on('all', function(event){
    //   console.log(event.key, event.data);
    // });
    //
    client.on('connected', function () {
      done();
      //setTimeout(done, 5000);
    });

    client.connect();
  });

  it('mocks the connection uses it to test the client disconnect', function (done) {

    this.timeout(10000);

    var Client = require('../../..').Client;

    var config = {url: 'ws://127.0.0.1:3737'};

    var client = new Client(config, {
      Connection: Connection
    });

    // client.on('all', function(event){
    //   console.log(event.key, event.data);
    // });
    //
    client.on('connected', function () {

      client.on('close', function (info) {
        expect(info.code).to.be(1);
        expect(info.reason).to.be('intentional disconnect test');
        done();
      });

      client.disconnect({code: 1, reason: 'intentional disconnect test'});
    });

    client.connect();
  });

  it('mocks the connection uses it to test the client subscribe and publish', function (done) {

    this.timeout(10000);

    var Client = require('../../..').Client;

    var config = {url: 'ws://127.0.0.1:3737'};

    var client = new Client(config, {
      Connection: Connection
    });

    client.on('connected', function () {

      client.on('close', function (info) {
        done();
      });

      client.subscribe('/a/test/path', function (data) {

        expect(data).to.eql({some: "data"});

        client.disconnect({code: 1, reason: 'intentional disconnect test'});

      }).then(function () {

        client.publish('/a/test/path', {some: "data"})
          .then(function (response) {
            //do nothing
          }).catch(done);

      }).catch(done);
    });

    client.connect();

  });

  it('mocks the connection uses it to test the client subscribe, publish and unsubscribe', function (done) {

    this.timeout(10000);

    var Client = require('../../..').Client;

    var config = {url: 'ws://127.0.0.1:3737'};

    var client = new Client(config, {
      Connection: Connection
    });

    var emitCount = 0;

    client.on('connected', function () {

      client.on('close', function (info) {
        done();
      });

      client.subscribe('/a/test/path', function (data) {
        emitCount++;
      }).then(function (subKey) {

        client.publish('/a/test/path', {some: "data"})
          .then((response) => {

            client.unsubscribe(subKey).then(function (response) {

              client.publish('/a/test/path', {some: "data"})
                .then(function (response) {

                  setTimeout(() => {
                    if (emitCount > 1) return done(new Error('unsub failed'));
                    else client.disconnect({code: 1, reason: 'intentional disconnect test'});
                  }, 5000);

                }).catch(done);
            });

          }).catch(done);

      }).catch(done);
    });

    client.connect();

  });
});