03-transport-websockets.node.js

/* eslint-env mocha */
'use strict'

const chai = require('chai')
const dirtyChai = require('dirty-chai')
const expect = chai.expect
chai.use(dirtyChai)
const parallel = require('async/parallel')
const multiaddr = require('multiaddr')
const WebSockets = require('libp2p-websockets')
const pull = require('pull-stream')
const goodbye = require('pull-goodbye')

const utils = require('./utils')
const Swarm = require('../src')

describe('transport - websockets', function () {
  var swarmA
  var swarmB
  var peerA
  var peerB

  before((done) => {
    utils.createInfos(2, (err, infos) => {
      if (err) {
        return done(err)
      }
      peerA = infos[0]
      peerB = infos[1]

      peerA.multiaddr.add(multiaddr('/ip4/127.0.0.1/tcp/9888/ws'))
      peerB.multiaddr.add(multiaddr('/ip4/127.0.0.1/tcp/9999/ws/ipfs/QmcgpsyWgH8Y8ajJz1Cu72KnS5uo2Aa2LpzU7kinSupNKC'))
      swarmA = new Swarm(peerA)
      swarmB = new Swarm(peerB)
      done()
    })
  })

  it('add', (done) => {
    swarmA.transport.add('ws', new WebSockets())
    expect(Object.keys(swarmA.transports).length).to.equal(1)
    swarmB.transport.add('ws', new WebSockets(), () => {
      expect(Object.keys(swarmB.transports).length).to.equal(1)
      done()
    })
  })

  it('listen', (done) => {
    parallel([
      (cb) => swarmA.transport.listen('ws', {}, (conn) => {
        pull(conn, conn)
      }, cb),
      (cb) => swarmB.transport.listen('ws', {}, (conn) => {
        pull(conn, conn)
      }, cb)
    ], () => {
      expect(peerA.multiaddrs.length).to.equal(1)
      expect(
        peerA.multiaddrs[0].equals(multiaddr('/ip4/127.0.0.1/tcp/9888/ws'))
      ).to.be.equal(
        true
      )
      expect(peerB.multiaddrs.length).to.equal(1)
      expect(
        peerB.multiaddrs[0].equals(multiaddr('/ip4/127.0.0.1/tcp/9999/ws/ipfs/QmcgpsyWgH8Y8ajJz1Cu72KnS5uo2Aa2LpzU7kinSupNKC'))
      ).to.equal(
        true
      )
      done()
    })
  })

  it('dial', (done) => {
    const conn = swarmA.transport.dial('ws', multiaddr('/ip4/127.0.0.1/tcp/9999/ws'), (err, conn) => {
      expect(err).to.not.exist()
    })

    const s = goodbye({
      source: pull.values([Buffer('hey')]),
      sink: pull.collect((err, data) => {
        expect(err).to.not.exist()
        expect(data).to.be.eql([Buffer('hey')])
        done()
      })
    })
    pull(s, conn, s)
  })

  it('dial (conn from callback)', (done) => {
    swarmA.transport.dial('ws', multiaddr('/ip4/127.0.0.1/tcp/9999/ws'), (err, conn) => {
      expect(err).to.not.exist()

      const s = goodbye({
        source: pull.values([Buffer('hey')]),
        sink: pull.collect((err, data) => {
          expect(err).to.not.exist()
          expect(data).to.be.eql([Buffer('hey')])
          done()
        })
      })
      pull(s, conn, s)
    })
  })

  it('dial to set of multiaddr, none is available', (done) => {
    swarmA.transport.dial('ws', [
      multiaddr('/ip4/127.0.0.1/tcp/9320/ws'),
      multiaddr('/ip4/127.0.0.1/tcp/9359/ws')
    ], (err, conn) => {
      expect(err).to.exist()
      expect(err.errors).to.have.length(2)
      expect(conn).to.not.exist()
      done()
    })
  })

  it('close', (done) => {
    parallel([
      (cb) => swarmA.transport.close('ws', cb),
      (cb) => swarmB.transport.close('ws', cb)
    ], done)
  })
})