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;
}