@sintaxi/dnsd/test/server.js

Dynamic authoritative name server for Iris services

// Copyright 2012 Iris Couch, all rights reserved.
//
// Test DNS server

var fs = require('fs')
var net = require('net')
var tap = require('tap')
var test = tap.test
var util = require('util')
var dgram = require('dgram')

var API = require('../named')

var PORT = 5321

test('Server API', function(t) {
  // The idea is to mimic the http or net server API.
  t.type(API.createServer, 'function', 'createServer() API call')
  t.equal(API.createServer.length, 1, 'createServer() takes one argument')

  var server
  t.doesNotThrow(function() { server = API.createServer(function(){}) }, 'Create a server')
  t.type(server.listen, 'function', 'Server has a .listen() method')
  t.equal(server.listen.length, 3, 'listen() method takes three parameters')

  t.type(server.unref, 'function', 'Server has a .unref() method')
  t.equal(server.unref.length, 0, 'unref() method takes zero parameters')

  t.type(server.ref, 'function', 'Server has a .ref() method')
  t.equal(server.ref.length, 0, 'ref() method takes zero parameters')

  t.end()
})

test('Network server', function(t) {
  var server = API.createServer(function(req, res) {
    console.log('Req: %j', req)
    console.log('Res: %j', res)
    res.end()
  })

  var events = {'listening':0, 'close':0, 'error':0}
  var listenCallback = false

  server.on('listening', function() { events.listening += 1 })
  server.on('close', function() { events.close += 1 })
  server.on('error', function() { events.error += 1 })

  server.listen(PORT, '127.0.0.1', function() { listenCallback = true })
  setTimeout(check_init, 150)
  setTimeout(check_stop, 200)

  function check_init() {
    t.ok(listenCallback, '"listen" callback called')
    t.equal(events.listening, 1, 'Fired "listening" event')
    t.equal(events.close, 0, 'No "close" events')
    t.equal(events.error, 0, 'No "error" events')

    server.close()
  }

  function check_stop() {
    t.equal(events.close, 1, 'Fired "close" event')
    t.equal(events.error, 0, 'Still no "error" events')

    t.end()
  }
})

test('Network queries', function(t) {
  var reqs = { 'ru.ac.th'                   : { id:36215, opcode:'update', recursion:false, name:'dynamic-update' }
             , 'oreilly.com'                : { id:45753, opcode:'query' , recursion:true , name:'oreilly.com-query' }
             , 'www.company.example'        : { id:62187, opcode:'query' , recursion:true , name:'www.company.example-query' }
             , 'www.microsoft.com.nsatc.net': { id:47096, opcode:'query' , recursion:true , name:'www.microsoft.com-query' }
             }

  var i = 0
  var server = API.createServer(check_req)
  server.listen(PORT, '127.0.0.1')
  server.on('listening', send_requests)

  function check_req(req, res) {
    t.type(req.question, 'Array', 'Got a question message')
    t.equal(req.question.length, 1, 'Got exactly one question')

    var question = req.question[0]
      , name = question.name
      , expected = reqs[name]

    t.ok(expected, 'Found expected request: ' + name)
    t.equal(req.id, expected.id, 'ID match: ' + name)
    t.equal(req.opcode, expected.opcode, 'Opcode match: ' + name)
    t.equal(req.recursion_desired, expected.recursion, 'Recursion match: ' + name)

    res.end()

    i += 1
    if(i == 4) {
      server.close()
      t.end()
    }
  }

  function send_requests() {
    var type = 'tcp'
    Object.keys(reqs).forEach(function(domain) {
      var data = fs.readFileSync(__dirname + '/../_test_data/' + reqs[domain].name)

      if(type == 'udp') {
        type = 'tcp'
        var sock = dgram.createSocket('udp4')
        sock.send(data, 0, data.length, PORT, '127.0.0.1', function() { sock.close() })
      } else {
        type = 'udp'
        console.error('TCP to %d', PORT)
        var sock = net.connect({'port':PORT}, function(er) {
          sock.write(new Buffer([data.length >> 8, data.length & 0xff]))
          sock.write(data)
          sock.end()
        })
      }
    })
  }
})