Verify that an IPv6 address is used in e2e test

1 files changed, 14 insertions, 6 deletions

diff --git a/test/test-manager/src/tests/tunnel.rs b/test/test-manager/src/tests/tunnel.rs
index 828ced73a8..3d6ac6432e 100644
--- a/test/test-manager/src/tests/tunnel.rs
+++ b/test/test-manager/src/tests/tunnel.rs
@@ -22,7 +22,7 @@ use mullvad_types::{
     wireguard,
 };
 use std::{
-    net::{Ipv4Addr, Ipv6Addr, SocketAddr},
+    net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr},
     str::FromStr,
 };
 use talpid_types::net::{
@@ -117,17 +117,25 @@ pub async fn test_wireguard_tunnel_ipvx(
             .unwrap();
 
         let connection_result = connect_and_wait(&mut mullvad_client).await;
-        assert_eq!(
-            connection_result.is_ok(),
-            should_succeed,
-            "unexpected result for port {port}: {connection_result:?}",
-        );
 
         if should_succeed {
+            let Ok(connection_result) = &connection_result else {
+                panic!("connection must succeed for port {port}: {connection_result:?}");
+            };
+
+            let endpoint = connection_result.endpoint().expect("must have endpoint");
+            let endpoint = endpoint.entry_endpoint.unwrap_or(endpoint.endpoint);
+            assert!(matches!(endpoint.address.ip(), IpAddr::VX(..)));
+
             assert!(
                 helpers::using_mullvad_exit(&rpc).await,
                 "expected Mullvad exit IP"
             );
+        } else {
+            assert!(
+                connection_result.is_err(),
+                "connection must fail for port {port}: {connection_result:?}",
+            );
         }
 
         disconnect_and_wait(&mut mullvad_client).await?;