From 117a59737b56727be57ac1538263e3c4676ecf36 Mon Sep 17 00:00:00 2001
From: Turan Ege Caner <turan.caner@nethermind.io>
Date: Sat, 15 Feb 2025 00:20:19 +0800
Subject: [PATCH] MockMailbox Zero Address handling for custom hook

---
 cairo/crates/mocks/src/mock_mailbox.cairo | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/cairo/crates/mocks/src/mock_mailbox.cairo b/cairo/crates/mocks/src/mock_mailbox.cairo
index a347315..35e71a6 100644
--- 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 {