init function is capable of returning value

OpenZeppelin · Jan 16, 2025 · 50c3971 · 50c3971
1 parent 8ae0ebd
commit 50c3971
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/crates/motsu/src/context.rs b/crates/motsu/src/context.rs
@@ -346,12 +346,12 @@ impl<ST: StorageType + TestRouter + 'static> Contract<ST> {
 
     /// Initialize the contract with an `initializer` function, and on behalf of
     /// the given `account`.
-    pub fn init<A: Into<Address>>(
+    pub fn init<A: Into<Address>, Output>(
         &self,
         account: A,
-        initializer: impl FnOnce(&mut ST),
-    ) {
-        initializer(&mut self.sender(account.into()));
+        initializer: impl FnOnce(&mut ST) -> Output,
+    ) -> Output {
+        initializer(&mut self.sender(account.into()))
     }
 
     /// Create a new contract with random address.