abstract-extension/test.js

Small abstraction to help build out user defined extension messages in an RPC system

const Message = require('./')
const tape = require('tape')

tape('basic', function (t) {
  t.plan(4 * 3)

  const a = Message.createLocal({
    onextensionupdate () {
      pb.update(a.names())
    }
  })

  const b = Message.createLocal({
    onextensionupdate () {
      pa.update(b.names())
    }
  })

  const pa = a.remote()
  const pb = b.remote()

  const ma = a.add('b', {
    encoding: 'utf-8',
    onmessage (str) {
      t.same(str, 'to a')
    }
  })

  const mb = b.add('b', {
    encoding: 'utf-8',
    onmessage (str) {
      t.same(str, 'to b')
    }
  })

  t.same(ma.id, 0)
  t.same(mb.id, 0)
  pb.onmessage(ma.id, ma.encode('to b'))
  pa.onmessage(mb.id, mb.encode('to a'))

  a.add('a')

  t.same(ma.id, 1)
  t.same(mb.id, 0)
  pb.onmessage(ma.id, ma.encode('to b'))
  pa.onmessage(mb.id, mb.encode('to a'))

  b.add('c')

  t.same(ma.id, 1)
  t.same(mb.id, 0)
  pb.onmessage(ma.id, ma.encode('to b'))
  pa.onmessage(mb.id, mb.encode('to a'))
})

tape('extendable', function (t) {
  t.plan(3)

  class Extended extends Message {
    extended () {
      t.pass('was extended')
    }
  }

  const a = Extended.createLocal()
  const m = a.add('hello')

  t.ok(m instanceof Extended)
  t.ok(m instanceof Message)
  m.extended()
})

tape('create from function', function (t) {
  const a = Message.createLocal()

  const m = a.add('hello', function (ext) {
    t.ok(ext instanceof Message)
  })

  t.ok(m instanceof Message)
  t.end()
})