1
1
#![ cfg( test) ]
2
2
3
3
use crate :: { socket:: ListenConfig , Discv5 , * } ;
4
- use enr:: { k256, CombinedKey , Enr , EnrBuilder , EnrKey , NodeId } ;
4
+ use enr:: { k256, CombinedKey , Enr , EnrKey , NodeId } ;
5
5
use rand_core:: { RngCore , SeedableRng } ;
6
6
use std:: {
7
7
collections:: HashMap ,
@@ -28,11 +28,7 @@ async fn build_nodes(n: usize, base_port: u16) -> Vec<Discv5> {
28
28
let listen_config = ListenConfig :: Ipv4 { ip, port } ;
29
29
let config = ConfigBuilder :: new ( listen_config) . build ( ) ;
30
30
31
- let enr = EnrBuilder :: new ( "v4" )
32
- . ip4 ( ip)
33
- . udp4 ( port)
34
- . build ( & enr_key)
35
- . unwrap ( ) ;
31
+ let enr = Enr :: builder ( ) . ip4 ( ip) . udp4 ( port) . build ( & enr_key) . unwrap ( ) ;
36
32
// transport for building a swarm
37
33
let mut discv5 = Discv5 :: new ( enr, enr_key, config) . unwrap ( ) ;
38
34
discv5. start ( ) . await . unwrap ( ) ;
@@ -52,11 +48,7 @@ async fn build_nodes_from_keypairs(keys: Vec<CombinedKey>, base_port: u16) -> Ve
52
48
let listen_config = ListenConfig :: Ipv4 { ip, port } ;
53
49
let config = ConfigBuilder :: new ( listen_config) . build ( ) ;
54
50
55
- let enr = EnrBuilder :: new ( "v4" )
56
- . ip4 ( ip)
57
- . udp4 ( port)
58
- . build ( & enr_key)
59
- . unwrap ( ) ;
51
+ let enr = Enr :: builder ( ) . ip4 ( ip) . udp4 ( port) . build ( & enr_key) . unwrap ( ) ;
60
52
61
53
let mut discv5 = Discv5 :: new ( enr, enr_key, config) . unwrap ( ) ;
62
54
discv5. start ( ) . await . unwrap ( ) ;
@@ -77,7 +69,7 @@ async fn build_nodes_from_keypairs_ipv6(keys: Vec<CombinedKey>, base_port: u16)
77
69
} ;
78
70
let config = ConfigBuilder :: new ( listen_config) . build ( ) ;
79
71
80
- let enr = EnrBuilder :: new ( "v4" )
72
+ let enr = Enr :: builder ( )
81
73
. ip6 ( Ipv6Addr :: LOCALHOST )
82
74
. udp6 ( port)
83
75
. build ( & enr_key)
@@ -108,7 +100,7 @@ async fn build_nodes_from_keypairs_dual_stack(
108
100
} ;
109
101
let config = ConfigBuilder :: new ( listen_config) . build ( ) ;
110
102
111
- let enr = EnrBuilder :: new ( "v4" )
103
+ let enr = Enr :: builder ( )
112
104
. ip4 ( Ipv4Addr :: LOCALHOST )
113
105
. udp4 ( ipv4_port)
114
106
. ip6 ( Ipv6Addr :: LOCALHOST )
@@ -744,11 +736,7 @@ async fn test_table_limits() {
744
736
let mut keypairs = generate_deterministic_keypair ( 12 , 9487 ) ;
745
737
let ip: Ipv4Addr = "127.0.0.1" . parse ( ) . unwrap ( ) ;
746
738
let enr_key: CombinedKey = keypairs. remove ( 0 ) ;
747
- let enr = EnrBuilder :: new ( "v4" )
748
- . ip4 ( ip)
749
- . udp4 ( 9050 )
750
- . build ( & enr_key)
751
- . unwrap ( ) ;
739
+ let enr = Enr :: builder ( ) . ip4 ( ip) . udp4 ( 9050 ) . build ( & enr_key) . unwrap ( ) ;
752
740
let listen_config = ListenConfig :: Ipv4 {
753
741
ip : enr. ip4 ( ) . unwrap ( ) ,
754
742
port : enr. udp4 ( ) . unwrap ( ) ,
@@ -763,7 +751,7 @@ async fn test_table_limits() {
763
751
. map ( |i| {
764
752
let ip: Ipv4Addr = Ipv4Addr :: new ( 192 , 168 , 1 , i as u8 ) ;
765
753
let enr_key: CombinedKey = keypairs. remove ( 0 ) ;
766
- EnrBuilder :: new ( "v4" )
754
+ Enr :: builder ( )
767
755
. ip4 ( ip)
768
756
. udp4 ( 9050 + i as u16 )
769
757
. build ( & enr_key)
@@ -782,19 +770,15 @@ async fn test_table_limits() {
782
770
async fn test_bucket_limits ( ) {
783
771
let enr_key = CombinedKey :: generate_secp256k1 ( ) ;
784
772
let ip: Ipv4Addr = "127.0.0.1" . parse ( ) . unwrap ( ) ;
785
- let enr = EnrBuilder :: new ( "v4" )
786
- . ip4 ( ip)
787
- . udp4 ( 9500 )
788
- . build ( & enr_key)
789
- . unwrap ( ) ;
773
+ let enr = Enr :: builder ( ) . ip4 ( ip) . udp4 ( 9500 ) . build ( & enr_key) . unwrap ( ) ;
790
774
let bucket_limit: usize = 2 ;
791
775
// Generate `bucket_limit + 1` keypairs that go in `enr` node's 256th bucket.
792
776
let keys = {
793
777
let mut keys = Vec :: new ( ) ;
794
778
for _ in 0 ..bucket_limit + 1 {
795
779
loop {
796
780
let key = CombinedKey :: generate_secp256k1 ( ) ;
797
- let enr_new = EnrBuilder :: new ( "v4" ) . build ( & key) . unwrap ( ) ;
781
+ let enr_new = Enr :: empty ( & key) . unwrap ( ) ;
798
782
let node_key: Key < NodeId > = enr. node_id ( ) . into ( ) ;
799
783
let distance = node_key. log2_distance ( & enr_new. node_id ( ) . into ( ) ) . unwrap ( ) ;
800
784
if distance == 256 {
@@ -810,7 +794,7 @@ async fn test_bucket_limits() {
810
794
. map ( |i| {
811
795
let kp = & keys[ i - 1 ] ;
812
796
let ip: Ipv4Addr = Ipv4Addr :: new ( 192 , 168 , 1 , i as u8 ) ;
813
- EnrBuilder :: new ( "v4" )
797
+ Enr :: builder ( )
814
798
. ip4 ( ip)
815
799
. udp4 ( 9500 + i as u16 )
816
800
. build ( kp)
0 commit comments