wikimedia-kad-fork/test/hooks/whitelist.unit.js

implementation of the kademlia dht for node

'use strict';

var expect = require('chai').expect;
var WhitelistFactory = require('../../lib/hooks/whitelist');
var Message = require('../../lib/message');
var AddressPortContact = require('../../lib/contacts/address-port-contact');

describe('Hooks/Whitelist', function() {

  var node1 = AddressPortContact({
    address: '127.0.0.1',
    port: 8080
  });
  var node2 = AddressPortContact({
    address: '127.0.0.1',
    port: 8081
  });
  var whitelist = WhitelistFactory([node1.nodeID]);

  it('should pass an error if contact is not in the whitelist', function(done) {
    whitelist(Message({
      method: 'PING',
      params: {
        contact: node2
      },
      id: 'message1'
    }), node2, function(err) {
      expect(err).to.be.instanceOf(Error);
      done();
    });
  });

  it('should pass if contact is in the whitelist', function(done) {
    whitelist(Message({
      method: 'PING',
      params: {
        contact: node1
      },
      id: 'message1'
    }), node1, function(err) {
      expect(err).to.equal(undefined);
      done();
    });
  });

});