js-libp2p/test/connection-manager/max-peer-per-protocol.js

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

const chai = require('chai')
chai.use(require('dirty-chai'))
const expect = chai.expect

const Prepare = require('./utils/prepare')

const PEER_COUNT = 3

describe('maxPeers', function () {
  const prepare = Prepare(PEER_COUNT, [{
    maxPeersPerProtocol: {
      tcp: 1
    }
  }])
  before(prepare.create)
  after(prepare.after)

  it('kicks out peers in excess', function (done) {
    this.timeout(10000)

    let disconnects = 0
    const manager = prepare.connManagers()[0]
    manager.on('disconnected', () => {
      disconnects++
      expect(disconnects).to.be.most(PEER_COUNT - 2)
      manager.removeAllListeners('disconnected')
      done()
    })

    prepare.tryConnectAll((err) => {
      expect(err).to.not.exist()
    })
  })
})