dht-flood/test.js

const test = require('tape')
const Hyperswarm = require('hyperswarm')
const sodium = require('sodium-universal')
const b4a = require('b4a')
const { default: DHTFlood } = require('./')
const crypto = require('crypto')

const topicName = crypto.randomBytes(10)

test('Broadcast through several peers', (t) => {
    const peer1 = createPeer()
    const peer2 = createPeer()
    const peer3 = createPeer()
    t.plan(2)

    Promise.all([peer1, peer2, peer3]).then((peers) => {
        const peer1 = peers.shift()
        const peer2 = peers.shift()
        const peer3 = peers.shift()

        const flood1 = new DHTFlood({ swarm: peer1 })
        const flood2 = new DHTFlood({ swarm: peer2 })
        const flood3 = new DHTFlood({ swarm: peer3 })
        const data = Buffer.from('Hello World')

        flood1.on('message', () => t.error('Got own message'))

        flood2.on('message', (message) => {
            t.deepEquals(message, data, 'Data got  broadcast')
        })
        flood3.on('message', (message) => {
            t.deepEquals(message, data, 'Data got  broadcast')
        })

        function maybeFlood() {
            if (peer1.peers.size === 2) {
                flood1.broadcast(data);
                peer1.removeListener("connection", maybeFlood);
            }
        }

        peer1.on("connection", maybeFlood);

        t.teardown(() => {
            [peer1, peer2, peer3].forEach((item) => item.destroy())
        })
    })
})

async function createPeer () {
    const swarm = new Hyperswarm()
    await swarm.dht.ready()
    await swarm.listen()

    const topic = b4a.allocUnsafe(32)
    sodium.crypto_generichash(topic, b4a.from(topicName))

    swarm.join(topic)

    return swarm
}