nim-eth/tests/p2p/test_discovery.nim

#
#                 Ethereum P2P
#              (c) Copyright 2018
#       Status Research & Development GmbH
#
#    See the file "LICENSE", included in this
#    distribution, for details about the copyright.
#

import sequtils, logging
import eth/keys, chronos, byteutils
import eth/p2p/[discovery, kademlia, enode]

const clientId = "nim-eth-p2p/0.0.1"

addHandler(newConsoleLogger())

proc startDiscoveryNode(privKey: PrivateKey, address: Address, bootnodes: seq[ENode]): Future[DiscoveryProtocol] {.async.} =
  result = newDiscoveryProtocol(privKey, address, bootnodes)
  result.open()
  await result.bootstrap()

proc localAddress(port: int): Address =
  let port = Port(port)
  result = Address(udpPort: port, tcpPort: port, ip: parseIpAddress("127.0.0.1"))

let
  bootNodeKey = initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a617")
  bootNodeAddr = localAddress(20301)
  bootENode = initENode(bootNodeKey.getPublicKey, bootNodeAddr)

  nodeKeys = [
    initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a618"),
    initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a619"),
    initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a620")
  ]
proc nodeIdInNodes(id: NodeId, nodes: openarray[Node]): bool =
  for n in nodes:
    if id == n.id: return true

proc test() {.async.} =
  let bootNode = await startDiscoveryNode(bootNodeKey, bootNodeAddr, @[])

  var nodeAddrs = newSeqOfCap[Address](nodeKeys.len)
  for i in 0 ..< nodeKeys.len: nodeAddrs.add(localAddress(20302 + i))

  var nodes = await all(zip(nodeKeys, nodeAddrs).mapIt(
    startDiscoveryNode(it.a, it.b, @[bootENode]))
  )
  nodes.add(bootNode)

  for i in nodes:
    for j in nodes:
      if j != i:
        doAssert(nodeIdInNodes(i.thisNode.id, j.randomNodes(nodes.len - 1)))

waitFor test()