@@ -301,7 +301,11 @@ impl<'a> TraitClient<'a> {
301301 self . env . mock_auths ( mock_auths) ;
302302 }
303303 if self . mock_all_auths {
304- self . env . mock_all_auths ( ) ;
304+ if self . allow_non_root_auth {
305+ self . env . mock_all_auths_allowing_non_root_auth ( ) ;
306+ } else {
307+ self . env . mock_all_auths ( ) ;
308+ }
305309 }
306310 }
307311 use soroban_sdk:: { FromVal , IntoVal } ;
@@ -457,7 +461,11 @@ impl<'a> ContractClient<'a> {
457461 self . env . mock_auths ( mock_auths) ;
458462 }
459463 if self . mock_all_auths {
460- self . env . mock_all_auths ( ) ;
464+ if self . allow_non_root_auth {
465+ self . env . mock_all_auths_allowing_non_root_auth ( ) ;
466+ } else {
467+ self . env . mock_all_auths ( ) ;
468+ }
461469 }
462470 }
463471 use soroban_sdk:: { FromVal , IntoVal } ;
@@ -630,7 +638,11 @@ impl<'a> ContractClient<'a> {
630638 self . env . mock_auths ( mock_auths) ;
631639 }
632640 if self . mock_all_auths {
633- self . env . mock_all_auths ( ) ;
641+ if self . allow_non_root_auth {
642+ self . env . mock_all_auths_allowing_non_root_auth ( ) ;
643+ } else {
644+ self . env . mock_all_auths ( ) ;
645+ }
634646 }
635647 }
636648 use soroban_sdk:: { FromVal , IntoVal } ;
@@ -955,7 +967,11 @@ mod test_with_wasm {
955967 self . env . mock_auths ( mock_auths) ;
956968 }
957969 if self . mock_all_auths {
958- self . env . mock_all_auths ( ) ;
970+ if self . allow_non_root_auth {
971+ self . env . mock_all_auths_allowing_non_root_auth ( ) ;
972+ } else {
973+ self . env . mock_all_auths ( ) ;
974+ }
959975 }
960976 }
961977 use soroban_sdk:: { FromVal , IntoVal } ;
@@ -1036,7 +1052,11 @@ mod test_with_wasm {
10361052 self . env . mock_auths ( mock_auths) ;
10371053 }
10381054 if self . mock_all_auths {
1039- self . env . mock_all_auths ( ) ;
1055+ if self . allow_non_root_auth {
1056+ self . env . mock_all_auths_allowing_non_root_auth ( ) ;
1057+ } else {
1058+ self . env . mock_all_auths ( ) ;
1059+ }
10401060 }
10411061 }
10421062 use soroban_sdk:: { FromVal , IntoVal } ;
0 commit comments