js-waku/src/lib/discovery/fetch_nodes.spec.ts

import { expect } from "chai";
import { Multiaddr } from "multiaddr";
import PeerId from "peer-id";

import { ENR, Waku2 } from "../enr";

import fetchNodesUntilCapabilitiesFulfilled from "./fetch_nodes";

async function createEnr(waku2: Waku2): Promise<ENR> {
  const peerId = await PeerId.create({ keyType: "secp256k1" });
  const enr = ENR.createFromPeerId(peerId);
  enr.setLocationMultiaddr(new Multiaddr("/ip4/18.223.219.100/udp/9000"));
  enr.multiaddrs = [
    new Multiaddr("/dns4/node1.do-ams.wakuv2.test.statusim.net/tcp/443/wss"),
    new Multiaddr("/dns6/node2.ac-chi.wakuv2.test.statusim.net/tcp/443/wss"),
    new Multiaddr(
      "/onion3/vww6ybal4bd7szmgncyruucpgfkqahzddi37ktceo3ah7ngmcopnpyyd:1234/wss"
    ),
  ];

  enr.waku2 = waku2;
  return enr;
}

const Waku2None = {
  relay: false,
  store: false,
  filter: false,
  lightPush: false,
};

describe("Fetch nodes until capabilities are fulfilled", function () {
  it("1 Relay, 1 fetch", async function () {
    const relayNode = await createEnr({ ...Waku2None, relay: true });

    const getNode = (): Promise<ENR> => Promise.resolve(relayNode);

    const res = await fetchNodesUntilCapabilitiesFulfilled(
      { relay: 1 },
      0,
      getNode
    );

    expect(res.length).to.eq(1);
    expect(res[0].peerId!.toB58String()).to.eq(relayNode.peerId?.toB58String());
  });

  it("1 Store, 2 fetches", async function () {
    const relayNode = await createEnr({ ...Waku2None, relay: true });
    const storeNode = await createEnr({ ...Waku2None, store: true });

    const retrievedNodes = [relayNode, storeNode];

    let fetchCount = 0;
    const getNode = (): Promise<ENR> => {
      const node = retrievedNodes[fetchCount];
      fetchCount++;
      return Promise.resolve(node);
    };

    const res = await fetchNodesUntilCapabilitiesFulfilled(
      { store: 1 },
      1,
      getNode
    );

    expect(res.length).to.eq(1);
    expect(res[0].peerId!.toB58String()).to.eq(storeNode.peerId?.toB58String());
  });

  it("1 Store, 2 relays, 2 fetches", async function () {
    const relayNode1 = await createEnr({ ...Waku2None, relay: true });
    const relayNode2 = await createEnr({ ...Waku2None, relay: true });
    const relayNode3 = await createEnr({ ...Waku2None, relay: true });
    const relayStoreNode = await createEnr({
      ...Waku2None,
      relay: true,
      store: true,
    });

    const retrievedNodes = [relayNode1, relayNode2, relayNode3, relayStoreNode];

    let fetchCount = 0;
    const getNode = (): Promise<ENR> => {
      const node = retrievedNodes[fetchCount];
      fetchCount++;
      return Promise.resolve(node);
    };

    const res = await fetchNodesUntilCapabilitiesFulfilled(
      { store: 1, relay: 2 },
      1,
      getNode
    );

    expect(res.length).to.eq(3);
    expect(res[0].peerId!.toB58String()).to.eq(
      relayNode1.peerId?.toB58String()
    );
    expect(res[1].peerId!.toB58String()).to.eq(
      relayNode2.peerId?.toB58String()
    );
    expect(res[2].peerId!.toB58String()).to.eq(
      relayStoreNode.peerId?.toB58String()
    );
  });

  it("1 Relay, 1 Filter, gives up", async function () {
    const relayNode = await createEnr({ ...Waku2None, relay: true });

    const getNode = (): Promise<ENR> => Promise.resolve(relayNode);

    const res = await fetchNodesUntilCapabilitiesFulfilled(
      { filter: 1, relay: 1 },
      5,
      getNode
    );

    expect(res.length).to.eq(1);
    expect(res[0].peerId!.toB58String()).to.eq(relayNode.peerId?.toB58String());
  });
});