diff options
| author | Sebastian Holmin <sebastian.holmin@mullvad.net> | 2024-08-21 17:45:23 +0200 |
|---|---|---|
| committer | David Lönnhager <david.l@mullvad.net> | 2024-08-22 20:24:38 +0200 |
| commit | dfe166d16fd87198546bb631fca38e08d473fb7c (patch) | |
| tree | 33d6d998be2b0e805ec9be0848bb9d4d3a4e8d20 | |
| parent | b42d608e4f666c0d3cf020395db2e7b42963e6f2 (diff) | |
| download | mullvadvpn-dfe166d16fd87198546bb631fca38e08d473fb7c.tar.xz mullvadvpn-dfe166d16fd87198546bb631fca38e08d473fb7c.zip | |
Improve IP override testing
| -rw-r--r-- | mullvad-relay-selector/tests/relay_selector.rs | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/mullvad-relay-selector/tests/relay_selector.rs b/mullvad-relay-selector/tests/relay_selector.rs index f0e48641c9..9b178c3a44 100644 --- 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!( |