MockMailbox Zero Address handling for custom hook

astraly-labs · Feb 14, 2025 · 117a597 · 117a597
1 parent 83a814a
commit 117a597
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cairo/crates/mocks/src/mock_mailbox.cairo b/cairo/crates/mocks/src/mock_mailbox.cairo
@@ -226,7 +226,13 @@ pub mod MockMailbox {
             hook: Option<ContractAddress>
         ) -> u256 {
             let hook = match hook {
-                Option::Some(hook) => { hook },
+                Option::Some(hook) => {
+                    if hook != contract_address_const::<0>() {
+                        hook
+                    } else {
+                        self.default_hook.read()
+                    }
+                },
                 Option::None(()) => { self.default_hook.read() }
             };
             let hook_metadata = match metadata {