@@ -12,7 +12,7 @@ use mullvad_types::relay_constraints::{
12
12
} ;
13
13
use mullvad_types:: wireguard;
14
14
use talpid_types:: net:: {
15
- proxy:: { CustomProxy , Socks5Remote } ,
15
+ proxy:: { CustomProxy , Socks5Local , Socks5Remote } ,
16
16
TransportProtocol , TunnelType ,
17
17
} ;
18
18
use test_macro:: test_function;
@@ -676,6 +676,11 @@ pub async fn test_local_socks_bridge(
676
676
rpc : ServiceClient ,
677
677
mut mullvad_client : MullvadProxyClient ,
678
678
) -> Result < ( ) , Error > {
679
+ let socks_server = rpc
680
+ . start_socks_server ( "127.0.0.1:0" . parse ( ) . unwrap ( ) )
681
+ . await
682
+ . expect ( "failed to start SOCKS server" ) ;
683
+
679
684
mullvad_client
680
685
. set_bridge_state ( relay_constraints:: BridgeState :: On )
681
686
. await
@@ -685,10 +690,13 @@ pub async fn test_local_socks_bridge(
685
690
. set_bridge_settings ( BridgeSettings {
686
691
bridge_type : BridgeType :: Custom ,
687
692
normal : BridgeConstraints :: default ( ) ,
688
- custom : Some ( CustomProxy :: Socks5Remote ( Socks5Remote :: new ( (
689
- crate :: vm:: network:: NON_TUN_GATEWAY ,
690
- crate :: vm:: network:: SOCKS5_PORT ,
691
- ) ) ) ) ,
693
+ custom : Some ( CustomProxy :: Socks5Local ( Socks5Local :: new (
694
+ (
695
+ crate :: vm:: network:: NON_TUN_GATEWAY ,
696
+ crate :: vm:: network:: SOCKS5_PORT ,
697
+ ) ,
698
+ socks_server. bind_addr ( ) . port ( ) ,
699
+ ) ) ) ,
692
700
} )
693
701
. await
694
702
. expect ( "failed to update bridge settings" ) ;
0 commit comments