fixup: test within-range port

mullvad · Jul 31, 2024 · bbc96e8 · bbc96e8
1 parent d7698d5
commit bbc96e8
Showing 1 changed file with 15 additions and 0 deletions.
diff --git 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,