peer-queue.spec.js

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

const chai = require('chai')
chai.use(require('dirty-chai'))
const expect = chai.expect
const PeerId = require('peer-id')
const { Buffer } = require('buffer')
const PeerQueue = require('../src/peer-list/peer-queue')

describe('PeerQueue', () => {
  it('basics', async () => {
    const p1 = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31'))
    const p2 = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a32'))
    const p3 = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33'))
    const p4 = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a34'))
    const p5 = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31'))

    const peer = new PeerId(Buffer.from('11140beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a31'))

    const pq = await PeerQueue.fromPeerId(peer)

    await pq.enqueue(p3)
    await pq.enqueue(p1)
    await pq.enqueue(p2)
    await pq.enqueue(p4)
    await pq.enqueue(p5)
    await pq.enqueue(p1)

    expect([
      pq.dequeue(),
      pq.dequeue(),
      pq.dequeue(),
      pq.dequeue(),
      pq.dequeue(),
      pq.dequeue()
    ].map((m) => m.toB58String())).to.be.eql([
      p1, p1, p1, p4, p3, p2
    ].map((m) => m.toB58String()))

    expect(pq.length).to.be.eql(0)
  })
})