test: strengthen as_contract_with_func auth coverage

Author: shinzoxD

soroban-sdk/src/env.rs

@@ -1788,7 +1788,17 @@ impl Env {
     ///
     /// Used to write or read contract data, or take other actions in tests for
     /// setting up tests or asserting on internal state.
-    pub fn as_contract_with_func<T>(&self, id: &Address, func: &crate::Symbol, f: impl FnOnce() -> T) -> T {
+    ///
+    /// Unlike [`as_contract`][Self::as_contract], `func` is recorded as the
+    /// current contract function for authorization. This lets
+    /// [`mock_auths`][Self::mock_auths] and [`auths`][Self::auths] observe the
+    /// same function name that [`Address::require_auth`] uses.
+    pub fn as_contract_with_func<T>(
+        &self,
+        id: &Address,
+        func: &crate::Symbol,
+        f: impl FnOnce() -> T,
+    ) -> T {
         let id = id.contract_id();
         let mut t: Option<T> = None;
         self.env_impl
@@ -1886,6 +1896,11 @@ impl Env {
     ///
     /// Used to write or read contract data, or take other actions in tests for
     /// setting up tests or asserting on internal state.
+    ///
+    /// Unlike [`try_as_contract`][Self::try_as_contract], `func` is recorded as
+    /// the current contract function for authorization. This lets
+    /// [`mock_auths`][Self::mock_auths] and [`auths`][Self::auths] observe the
+    /// same function name that [`Address::require_auth`] uses.
     pub fn try_as_contract_with_func<T, E>(
         &self,
         id: &Address,
@@ -1898,10 +1913,12 @@ impl Env {
     {
         let id = id.contract_id();
         let mut t: Option<T> = None;
-        let result = self.env_impl.try_with_test_contract_frame(id, func.to_symbol_val(), || {
-            t = Some(f());
-            Ok(().into())
-        });
+        let result = self
+            .env_impl
+            .try_with_test_contract_frame(id, func.to_symbol_val(), || {
+                t = Some(f());
+                Ok(().into())
+            });
 
         match result {
             Ok(_) => Ok(t.unwrap()),

soroban-sdk/src/tests/env.rs

@@ -386,16 +386,30 @@ fn test_as_contract_with_func() {
 
 #[test]
 fn test_try_as_contract_with_func() {
+    use crate::testutils::{MockAuth, MockAuthInvoke};
+    use crate::IntoVal;
+
     let env = Env::default();
 
     let addr = Address::generate(&env);
     env.register_at(&addr, Contract, ());
 
     let func_name = Symbol::new(&env, "custom_fn");
+    let user = Address::generate(&env);
+
+    env.mock_auths(&[MockAuth {
+        address: &user,
+        invoke: &MockAuthInvoke {
+            contract: &addr,
+            fn_name: "custom_fn",
+            args: ().into_val(&env),
+            sub_invokes: &[],
+        },
+    }]);
 
     let result = env.try_as_contract_with_func::<Symbol, Error>(&addr, &func_name, || {
+        user.require_auth();
         Symbol::new(&env, "hello")
     });
     assert_eq!(result, Ok(Symbol::new(&env, "hello")));
 }
-

soroban-sdk/test_snapshots/tests/env/test_try_as_contract_with_func.1.json

@@ -1,12 +1,27 @@
 {
   "generators": {
-    "address": 1,
-    "nonce": 0,
+    "address": 2,
+    "nonce": 1,
     "mux_id": 0
   },
   "auth": [
     [],
-    []
+    [],
+    [
+      [
+        "CAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAFCT4",
+        {
+          "function": {
+            "contract_fn": {
+              "contract_address": "CAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD2KM",
+              "function_name": "custom_fn",
+              "args": []
+            }
+          },
+          "sub_invocations": []
+        }
+      ]
+    ]
   ],
   "ledger": {
     "protocol_version": 27,
@@ -41,6 +56,49 @@
         },
         "live_until": 4095
       },
+      {
+        "entry": {
+          "last_modified_ledger_seq": 0,
+          "data": {
+            "contract_data": {
+              "ext": "v0",
+              "contract": "CAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAFCT4",
+              "key": "ledger_key_contract_instance",
+              "durability": "persistent",
+              "val": {
+                "contract_instance": {
+                  "executable": {
+                    "wasm": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"
+                  },
+                  "storage": null
+                }
+              }
+            }
+          },
+          "ext": "v0"
+        },
+        "live_until": 4095
+      },
+      {
+        "entry": {
+          "last_modified_ledger_seq": 0,
+          "data": {
+            "contract_data": {
+              "ext": "v0",
+              "contract": "CAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAFCT4",
+              "key": {
+                "ledger_key_nonce": {
+                  "nonce": "1"
+                }
+              },
+              "durability": "temporary",
+              "val": "void"
+            }
+          },
+          "ext": "v0"
+        },
+        "live_until": 6311999
+      },
       {
         "entry": {
           "last_modified_ledger_seq": 0,