rocket.chat.mqtt/node_modules/aedes/test/not-blocking.js

It's a MQTT Server, using redis to scale horizontally.

'use strict'

var mqtt = require('mqtt')
var test = require('tape')
var aedes = require('../')
var net = require('net')
var port = 4883

test('do not block after a subscription', function (t) {
  var instance = aedes()
  var server = net.createServer(instance.handle)
  var total = 10000
  var sent = 0
  var received = 0

  server.listen(4883, function (err) {
    t.error(err, 'no error')

    var publisher = mqtt.connect({
      port: port,
      keepalive: 0
    })

    var subscriber

    function immediatePublish () {
      setImmediate(publish)
    }

    function publish () {
      if (sent === total) {
        publisher.end()
      } else {
        sent++
        publisher.publish('test', 'payload', immediatePublish)
      }
    }

    function startSubscriber () {
      subscriber = mqtt.connect({
        port: port,
        keepalive: 0
      })

      subscriber.subscribe('test', publish)

      subscriber.on('message', function () {
        if (received % (total / 10) === 0) {
          console.log('sent / received', sent, received)
        }

        if (++received === total) {
          finish()
        }
      })
    }

    publisher.on('connect', startSubscriber)

    var timer = setTimeout(finish, 10000)

    function finish () {
      clearTimeout(timer)
      subscriber.end()
      publisher.end()
      instance.close()
      server.close()
      t.equal(total, sent, 'messages sent')
      t.equal(total, received, 'messages received')
      t.end()
    }
  })
})

test('do not block with overlapping subscription', function (t) {
  var instance = aedes({ concurrency: 15 })
  var server = net.createServer(instance.handle)
  var total = 10000
  var sent = 0
  var received = 0

  server.listen(4883, function (err) {
    t.error(err, 'no error')

    var publisher = mqtt.connect({
      port: port,
      keepalive: 0
    })

    var subscriber

    function immediatePublish (e) {
      setImmediate(publish)
    }

    function publish () {
      if (sent === total) {
        publisher.end()
      } else {
        sent++
        publisher.publish('test', 'payload', immediatePublish)
      }
    }

    function startSubscriber () {
      subscriber = mqtt.connect({
        port: port,
        keepalive: 0
      })

      subscriber.subscribe('#', function () {
        subscriber.subscribe('test', function () {
          immediatePublish()
        })
      })

      subscriber.on('message', function () {
        // console.log('received', arguments)
        if (received % (total / 10) === 0) {
          console.log('sent / received', sent, received)
        }

        if (++received === total) {
          finish()
        }
      })
    }

    publisher.on('connect', startSubscriber)

    var timer = setTimeout(finish, 10000)

    function finish () {
      clearTimeout(timer)
      subscriber.end()
      publisher.end()
      instance.close()
      server.close()
      t.equal(total, sent, 'messages sent')
      t.equal(total, received, 'messages received')
      t.end()
    }
  })
})