diff --git a/mullvad-relay-selector/src/relay_selector/helpers.rs b/mullvad-relay-selector/src/relay_selector/helpers.rs index cd02dae11856..fd30c4704388 100644 --- a/mullvad-relay-selector/src/relay_selector/helpers.rs +++ b/mullvad-relay-selector/src/relay_selector/helpers.rs @@ -220,6 +220,7 @@ mod tests { #[test] fn test_shadowsocks_no_extra_addrs() { const PORT_RANGES: &[(u16, u16)] = &[(100, 200), (1000, 2000)]; + const WITHIN_RANGE_PORT: u16 = 100; const OUT_OF_RANGE_PORT: u16 = 1; let wg_in_ip: IpAddr = "1.2.3.4".parse().unwrap(); @@ -233,6 +234,20 @@ mod tests { "expected port in port range" ); + let selected_addr = get_shadowsocks_obfuscator_inner( + wg_in_ip, + PORT_RANGES, + &[], + Constraint::Only(WITHIN_RANGE_PORT), + ) + .expect("should find within-range port"); + + assert_eq!(selected_addr.ip(), wg_in_ip); + assert!( + port_in_range(selected_addr.port(), PORT_RANGES), + "expected port in port range" + ); + let selected_addr = get_shadowsocks_obfuscator_inner( wg_in_ip, PORT_RANGES,