Add basic node discovery test

This commit is contained in:
kdeme 2020-02-17 16:36:04 +01:00 committed by zah
parent c679b9d436
commit bb6a3c2ce1
3 changed files with 54 additions and 1 deletions

View File

@ -56,6 +56,7 @@ proc runP2pTests() =
"test_waku_mail", "test_waku_mail",
"test_protocol_handlers", "test_protocol_handlers",
"test_enr", "test_enr",
"test_discoveryv5"
]: ]:
runTest("tests/p2p/" & filename) runTest("tests/p2p/" & filename)

View File

@ -8,7 +8,7 @@ import nimcrypto except toHex
type type
Protocol* = ref object Protocol* = ref object
transp: DatagramTransport transp: DatagramTransport
localNode: Node localNode*: Node
privateKey: PrivateKey privateKey: PrivateKey
whoareyouMagic: array[32, byte] whoareyouMagic: array[32, byte]
idHash: array[32, byte] idHash: array[32, byte]

View File

@ -0,0 +1,52 @@
import
unittest, chronos, sequtils, chronicles,
eth/keys, eth/p2p/enode, eth/trie/db,
eth/p2p/discoveryv5/[discovery_db, enr, node, types],
eth/p2p/discoveryv5/protocol as discv5_protocol,
./p2p_test_helper
proc startDiscoveryv5Node*(privKey: PrivateKey, address: Address,
bootnodes: seq[Record]): discv5_protocol.Protocol =
var db = DiscoveryDB.init(newMemoryDB())
result = newProtocol(privKey, db, address.udpPort)
for node in bootnodes:
result.addNode(node)
result.open()
result.start()
proc nodeIdInNodes(id: NodeId, nodes: openarray[Node]): bool =
for n in nodes:
if id == n.id: return true
suite "Discovery v5 Tests":
asyncTest "Discover nodes":
let
bootNodeKey = initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a617")
bootNodeAddr = localAddress(20301)
bootNode = startDiscoveryv5Node(bootNodeKey, bootNodeAddr, @[])
bootNodeRecord = initRecord(1, bootNodeKey,
{"udp": bootNodeAddr.udpPort.uint16, "ip": [byte 127, 0, 0, 1]})
let nodeKeys = [
initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a618"),
initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a619"),
initPrivateKey("a2b50376a79b1a8c8a3296485572bdfbf54708bb46d3c25d73d2723aaaf6a620")
]
var nodeAddrs = newSeqOfCap[Address](nodeKeys.len)
for i in 0 ..< nodeKeys.len: nodeAddrs.add(localAddress(20302 + i))
var nodes = zip(nodeKeys, nodeAddrs).mapIt(
startDiscoveryv5Node(it.a, it.b, @[bootNodeRecord]))
nodes.add(bootNode)
for node in nodes:
let discovered = await node.lookupRandom()
debug "Lookup from random id", node=node.localNode, discovered
# Check for each node if the other nodes shows up in the routing table
for i in nodes:
for j in nodes:
if j != i:
check(nodeIdInNodes(i.localNode.id, j.randomNodes(nodes.len - 1)))