nim-libp2p/tests/pubsub/utils.nim

import options, tables
import chronos
import ../../libp2p/standard_setup
export standard_setup

proc generateNodes*(num: Natural, gossip: bool = false): seq[Switch] =
  for i in 0..<num:
    result.add(newStandardSwitch(gossip = gossip))

proc subscribeNodes*(nodes: seq[Switch]) {.async.} =
  var dials: seq[Future[void]]
  for dialer in nodes:
    for node in nodes:
      if dialer.peerInfo.peerId != node.peerInfo.peerId:
        dials.add(dialer.subscribeToPeer(node.peerInfo))
  await sleepAsync(100.millis)
  await allFutures(dials)