Improve IP override testing

mullvad · Aug 22, 2024 · dfe166d · dfe166d
1 parent b42d608
commit dfe166d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mullvad-relay-selector/tests/relay_selector.rs b/mullvad-relay-selector/tests/relay_selector.rs
@@ -804,9 +804,11 @@ fn test_selecting_wireguard_over_shadowsocks_extra_ips() {
     match relay {
         GetRelay::Wireguard {
             obfuscator: Some(SelectedObfuscator { config: ObfuscatorConfig::Shadowsocks { endpoint }, .. }),
-            inner: WireguardConfig::Singlehop { .. },
+            inner: WireguardConfig::Singlehop { exit },
             ..
         } => {
+            assert!(!exit.overridden_ipv4);
+            assert!(!exit.overridden_ipv6);
             assert!(SHADOWSOCKS_RELAY_EXTRA_ADDRS.contains(&endpoint.ip()), "{} is not an additional IP", endpoint);
         }
         wrong_relay => panic!(
@@ -849,6 +851,7 @@ fn test_selecting_wireguard_ignore_extra_ips_override_v4() {
             ..
         } => {
             assert!(exit.overridden_ipv4);
+            assert!(!exit.overridden_ipv6);
             assert_eq!(endpoint.ip(), IpAddr::from(OVERRIDE_IPV4));
         }
         wrong_relay => panic!(
@@ -887,9 +890,11 @@ fn test_selecting_wireguard_ignore_extra_ips_override_v6() {
     match relay {
         GetRelay::Wireguard {
             obfuscator: Some(SelectedObfuscator { config: ObfuscatorConfig::Shadowsocks { endpoint }, .. }),
-            inner: WireguardConfig::Singlehop { .. },
+            inner: WireguardConfig::Singlehop { exit },
             ..
         } => {
+            assert!(exit.overridden_ipv6);
+            assert!(!exit.overridden_ipv4);
             assert_eq!(endpoint.ip(), IpAddr::from(OVERRIDE_IPV6));
         }
         wrong_relay => panic!(