te_nsqjs/test/lookupd_test.coffee

NodeJS client for NSQ

_ = require 'underscore'

nock = require 'nock'
should = require 'should'
url = require 'url'

lookup = require '../src/lookupd'

NSQD_1 =
  address: 'localhost'
  broadcast_address: 'localhost'
  hostname: 'localhost'
  http_port: 4151
  remote_address: 'localhost:12345'
  tcp_port: 4150
  topics: ['sample_topic']
  version: '0.2.23'
NSQD_2 =
  address: 'localhost'
  broadcast_address: 'localhost'
  hostname: 'localhost'
  http_port: 5151
  remote_address: 'localhost:56789'
  tcp_port: 5150
  topics: ['sample_topic']
  version: '0.2.23'
NSQD_3 =
  address: 'localhost'
  broadcast_address: 'localhost'
  hostname: 'localhost'
  http_port: 6151
  remote_address: 'localhost:23456'
  tcp_port: 6150
  topics: ['sample_topic']
  version: '0.2.23'
NSQD_4 =
  address: 'localhost'
  broadcast_address: 'localhost'
  hostname: 'localhost'
  http_port: 7151
  remote_address: 'localhost:34567'
  tcp_port: 7150
  topics: ['sample_topic']
  version: '0.2.23'

LOOKUPD_1 = '127.0.0.1:4161'
LOOKUPD_2 = '127.0.0.1:5161'
LOOKUPD_3 = 'http://127.0.0.1:6161/'
LOOKUPD_4 = 'http://127.0.0.1:7161/path/lookup'

nockUrlSplit = (url) ->
  match = url.match(/^(https?:\/\/[^\/]+)(\/.*$)/i)
  if match
    baseUrl: match[1]
    path: match[2]

registerWithLookupd = (lookupdAddress, nsqd) ->
  producers = if nsqd? then [nsqd] else []

  if nsqd?
    for topic in nsqd.topics
      if lookupdAddress.indexOf('://') is -1
        nock("http://#{lookupdAddress}")
          .get("/lookup?topic=#{topic}")
          .reply 200,
            status_code: 200
            status_txt: 'OK'
            data:
              producers: producers
      else
        {baseUrl, path} = nockUrlSplit(lookupdAddress)
        if not path or path is '/'
          path = '/lookup'
        nock(baseUrl)
          .get("#{path}?topic=#{topic}")
          .reply 200,
            status_code: 200
            status_txt: 'OK'
            data:
              producers: producers

setFailedTopicReply = (lookupdAddress, topic) ->
  nock("http://#{lookupdAddress}")
    .get("/lookup?topic=#{topic}")
    .reply 200,
      status_code: 500
      status_txt: 'INVALID_ARG_TOPIC'
      data: null


describe 'lookupd.lookup', ->
  afterEach ->
    nock.cleanAll()

  describe 'querying a single lookupd for a topic', ->
    it 'should return an empty list if no nsqd nodes', (done) ->
      setFailedTopicReply LOOKUPD_1, 'sample_topic'

      lookup LOOKUPD_1, 'sample_topic', (err, nodes) ->
        nodes.should.be.empty
        done()

    it 'should return a list of nsqd nodes for a success reply', (done) ->
      registerWithLookupd LOOKUPD_1, NSQD_1

      lookup LOOKUPD_1, 'sample_topic', (err, nodes) ->
        nodes.should.have.length 1
        for key in ['address', 'broadcast_address', 'tcp_port', 'http_port']
          should.ok key in _.keys(nodes[0])
        done()

  describe 'querying a multiple lookupd', ->
    it 'should combine results from multiple lookupds', (done) ->
      registerWithLookupd LOOKUPD_1, NSQD_1
      registerWithLookupd LOOKUPD_2, NSQD_2
      registerWithLookupd LOOKUPD_3, NSQD_3
      registerWithLookupd LOOKUPD_4, NSQD_4

      lookupdAddresses = [LOOKUPD_1, LOOKUPD_2, LOOKUPD_3, LOOKUPD_4]
      lookup lookupdAddresses, 'sample_topic', (err, nodes) ->
        nodes.should.have.length 4
        _.chain(nodes)
          .pluck('tcp_port')
          .sort()
          .value().should.be.eql [4150, 5150, 6150, 7150]
        done()

    it 'should dedupe combined results', (done) ->
      registerWithLookupd LOOKUPD_1, NSQD_1
      registerWithLookupd LOOKUPD_2, NSQD_1
      registerWithLookupd LOOKUPD_3, NSQD_1
      registerWithLookupd LOOKUPD_4, NSQD_1

      lookupdAddresses = [LOOKUPD_1, LOOKUPD_2, LOOKUPD_3, LOOKUPD_4]
      lookup lookupdAddresses, 'sample_topic', (err, nodes) ->
        nodes.should.have.length 1
        done()

    it 'should succeed inspite of failures to query a lookupd', (done) ->
      registerWithLookupd LOOKUPD_1, NSQD_1
      nock("http://#{LOOKUPD_2}")
        .get('/lookup?topic=sample_topic')
        .reply 500

      lookupdAddresses = [LOOKUPD_1, LOOKUPD_2]
      lookup lookupdAddresses, 'sample_topic', (err, nodes) ->
        nodes.should.have.length 1
        done()