fox-wamp/test/12.authorize.js

Web Application Message Router/Server WAMP/MQTT

'use strict'

const chai = require('chai')
const spies = require('chai-spies')
const expect = chai.expect

const WAMP         = require('../lib/wamp/protocol')
const { WampGate } = require('../lib/wamp/gate')
const { MqttGate } = require('../lib/mqtt/gate')
const Router       = require('../lib/router')
const FoxRouter    = require('../lib/fox_router')

chai.use(spies)

class TestAuth {
  authorize (session, funcClass, uniUri) {
    // console.log('!authorize', funcClass, uniUri)
    return uniUri[1] !== 'denied'
  }
}

const runs = [
  {it: 'zero', mkRouter: () => new Router()},
  {it: 'mem',  mkRouter: () => new FoxRouter()},
]

describe('12 authorize-topic', function () {
  runs.forEach(function (run) {
    describe('authorize:' + run.it, function () {
      var
        router,
        mqttGate,
        wampGate,
        realm,
        mqttSender,
        wampSender,
        wampCtx,
        mqttCtx,
        mqttCli,
        wampCli

      beforeEach(function () {
        let auth = new TestAuth()
        mqttSender = {}
        wampSender = {}
        router = run.mkRouter()
        realm = router.getRealm('test-realm')

        mqttGate = new MqttGate(router)
        mqttGate.setAuthHandler(auth)

        wampGate = new WampGate(router)
        wampGate.setAuthHandler(auth)

        mqttCli = router.createSession()
        mqttCtx = mqttGate.createContext(mqttCli, mqttSender)
        realm.joinSession(mqttCli)

        wampCli = router.createSession()
        wampCtx = wampGate.createContext(wampCli, wampSender)
        realm.joinSession(wampCli)
      })

      afterEach(function () {
      })

      it('wamp-subscribe:' + run.it, function () {
        wampSender.wampPkgWrite = chai.spy(
          function (msg) {
            expect(msg[0]).to.equal(WAMP.SUBSCRIBED)
            expect(msg[1]).to.equal(1234)
          }
        )
        wampGate.handle(wampCtx, wampCli, [WAMP.SUBSCRIBE, 1234, {}, 'topic1.passed'])
        expect(wampSender.wampPkgWrite, 'subscription confirmed').to.have.been.called.once()

        wampSender.wampPkgWrite = chai.spy(
          function (msg) {
            expect(msg[0]).to.equal(WAMP.ERROR)
            expect(msg[1]).to.equal(WAMP.SUBSCRIBE)
            expect(msg[2]).to.equal(1234)
            expect(msg[4]).to.equal('wamp.error.authorization_failed')
          }
        )
        wampGate.handle(wampCtx, wampCli, [WAMP.SUBSCRIBE, 1234, {}, 'topic1.denied'])
        expect(wampSender.wampPkgWrite, 'subscription confirmed').to.have.been.called.once()
      })

      it('mqtt-subscribe:' + run.it, function () {
        mqttSender.mqttPkgWrite = chai.spy((msg) => {
          expect(msg).to.deep.equal({cmd: 'suback', messageId: 321, granted: [ 128, 1 ]})
        })
        mqttGate.handle(mqttCtx, mqttCli, {
          cmd: 'subscribe',
          retain: false,
          qos: 1,
          dup: false,
          length: 17,
          topic: null,
          payload: null,
          subscriptions: [
            { topic: 'topic1/denied', qos: 0 },
            { topic: 'topic1/passed', qos: 2 }
          ],
          messageId: 321
        })
        expect(mqttSender.mqttPkgWrite).to.have.been.called.once()
      })
    })
  })
})