Add Discv5 banned nodes enabled tests.

status-im · Jan 24, 2025 · aab3ca0 · aab3ca0
1 parent 854f17c
commit aab3ca0
Show file tree

Hide file tree

Showing 3 changed files with 599 additions and 3 deletions.
diff --git a/tests/p2p/all_discv5_tests.nim b/tests/p2p/all_discv5_tests.nim
@@ -6,4 +6,5 @@ import
   ./test_ip_vote,
   ./test_routing_table,
   ./test_discoveryv5_encoding,
-  ./test_discoveryv5
+  ./test_discoveryv5,
+  ./test_discoveryv5_bannodes
diff --git a/tests/p2p/discv5_test_helper.nim b/tests/p2p/discv5_test_helper.nim
@@ -22,7 +22,8 @@ proc initDiscoveryNode*(
     address: Address,
     bootstrapRecords: openArray[Record] = [],
     localEnrFields: openArray[(string, seq[byte])] = [],
-    previousRecord = Opt.none(enr.Record)):
+    previousRecord = Opt.none(enr.Record),
+    banNodes = false):
     discv5_protocol.Protocol =
   # set bucketIpLimit to allow bucket split
   let config = DiscoveryConfig.init(1000, 24, 5)
@@ -36,7 +37,8 @@ proc initDiscoveryNode*(
     localEnrFields = localEnrFields,
     previousRecord = previousRecord,
     config = config,
-    rng = rng)
+    rng = rng,
+    banNodes = banNodes)
 
   protocol.open()