logos-messaging-js/packages/tests/src/utils/nodes.ts

import {
  CreateNodeOptions,
  DefaultNetworkConfig,
  IWaku,
  LightNode,
  NetworkConfig,
  Protocols
} from "@waku/interfaces";
import { createLightNode } from "@waku/sdk";
import { delay, derivePubsubTopicsFromNetworkConfig } from "@waku/utils";
import { Context } from "mocha";
import pRetry from "p-retry";

import { NOISE_KEY_1 } from "../constants.js";
import { ServiceNodesFleet } from "../lib/index.js";
import { verifyServiceNodesConnected } from "../lib/service_node.js";
import { Args } from "../types.js";

import { waitForConnections } from "./waitForConnections.js";

export async function runMultipleNodes(
  context: Context,
  networkConfig: NetworkConfig = DefaultNetworkConfig,
  customArgs?: Args,
  strictChecking: boolean = false,
  numServiceNodes = 2,
  withoutFilter = false
): Promise<[ServiceNodesFleet, LightNode]> {
  // create numServiceNodes nodes
  const serviceNodes = await ServiceNodesFleet.createAndRun(
    context,
    numServiceNodes,
    strictChecking,
    networkConfig,
    customArgs,
    withoutFilter
  );

  if (numServiceNodes > 1) {
    const success = await verifyServiceNodesConnected(serviceNodes.nodes);
    if (!success) {
      throw new Error("Failed to verify that service nodes are connected");
    }
  }

  const wakuOptions: CreateNodeOptions = {
    staticNoiseKey: NOISE_KEY_1,
    libp2p: {
      addresses: { listen: ["/ip4/0.0.0.0/tcp/0/ws"] }
    },
    networkConfig
  };

  const waku = await createLightNode(wakuOptions);
  await waku.start();

  if (!waku) {
    throw new Error("Failed to initialize waku");
  }

  //TODO: reinvestigate the need for these delays with nwaku:0.35.0: https://github.com/waku-org/js-waku/issues/2264
  await delay(2000);

  for (const node of serviceNodes.nodes) {
    await waku.dial(await node.getMultiaddrWithId());
    await waku.waitForPeers([Protocols.Filter, Protocols.LightPush]);
    const success = await node.ensureSubscriptions(
      derivePubsubTopicsFromNetworkConfig(networkConfig)
    );
    if (!success) {
      throw new Error("Failed to ensure subscriptions");
    }

    //TODO: reinvestigate the need for these delays with nwaku:0.35.0: https://github.com/waku-org/js-waku/issues/2264
    await delay(2000);

    await node.waitForLog(waku.libp2p.peerId.toString(), 100);
  }

  await waitForConnections(numServiceNodes, waku);

  const wakuConnections = waku.libp2p.getConnections();
  if (wakuConnections.length < numServiceNodes) {
    throw new Error(
      `Expected at least ${numServiceNodes} connections for js-waku.`
    );
  }

  return [serviceNodes, waku];
}

export async function teardownNodesWithRedundancy(
  serviceNodes: ServiceNodesFleet,
  wakuNodes: IWaku | IWaku[]
): Promise<void> {
  const wNodes = Array.isArray(wakuNodes) ? wakuNodes : [wakuNodes];

  const stopNwakuNodes = serviceNodes.nodes.map(async (node) => {
    await pRetry(
      async () => {
        await node.stop();
      },
      { retries: 3 }
    );
  });

  const stopWakuNodes = wNodes.map(async (waku) => {
    if (waku) {
      await pRetry(
        async () => {
          await waku.stop();
        },
        { retries: 3 }
      );
    }
  });

  await Promise.all([...stopNwakuNodes, ...stopWakuNodes]);
}