Skip to content

Commit faa5801

Browse files
authored
Use zero ports in tests (#216)
1 parent 299638d commit faa5801

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/service/test.rs

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@ use parking_lot::RwLock;
1919
use std::{collections::HashMap, sync::Arc, time::Duration};
2020
use tokio::sync::{mpsc, oneshot};
2121

22+
/// Default UDP port number to use for tests requiring UDP exposure
23+
pub const DEFAULT_UDP_PORT: u16 = 0;
24+
2225
fn _connected_state() -> NodeStatus {
2326
NodeStatus {
2427
state: ConnectionState::Connected,
@@ -105,14 +108,14 @@ async fn test_updating_connection_on_ping() {
105108
let ip = "127.0.0.1".parse().unwrap();
106109
let enr = EnrBuilder::new("v4")
107110
.ip4(ip)
108-
.udp4(10001)
111+
.udp4(DEFAULT_UDP_PORT)
109112
.build(&enr_key1)
110113
.unwrap();
111114
let ip2 = "127.0.0.1".parse().unwrap();
112115
let enr_key2 = CombinedKey::generate_secp256k1();
113116
let enr2 = EnrBuilder::new("v4")
114117
.ip4(ip2)
115-
.udp4(10002)
118+
.udp4(DEFAULT_UDP_PORT)
116119
.build(&enr_key2)
117120
.unwrap();
118121

@@ -141,7 +144,7 @@ async fn test_updating_connection_on_ping() {
141144
body: ResponseBody::Pong {
142145
enr_seq: 2,
143146
ip: ip2.into(),
144-
port: 10002,
147+
port: DEFAULT_UDP_PORT,
145148
},
146149
};
147150

@@ -173,15 +176,15 @@ async fn test_connection_direction_on_inject_session_established() {
173176
let ip = std::net::Ipv4Addr::LOCALHOST;
174177
let enr = EnrBuilder::new("v4")
175178
.ip4(ip)
176-
.udp4(10001)
179+
.udp4(DEFAULT_UDP_PORT)
177180
.build(&enr_key1)
178181
.unwrap();
179182

180183
let enr_key2 = CombinedKey::generate_secp256k1();
181184
let ip2 = std::net::Ipv4Addr::LOCALHOST;
182185
let enr2 = EnrBuilder::new("v4")
183186
.ip4(ip2)
184-
.udp4(10002)
187+
.udp4(DEFAULT_UDP_PORT)
185188
.build(&enr_key2)
186189
.unwrap();
187190

0 commit comments

Comments
 (0)