From 0cfaa3325f8f07af17bca053951567f2eeaf1a42 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adri=C3=A0=20Casaj=C3=BAs?= <adria.casajus@proton.ch>
Date: Tue, 23 Dec 2025 23:27:01 +0100
Subject: [PATCH] Fix mailbox test

---
 tests/proton/test_proton_callback_handler.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/tests/proton/test_proton_callback_handler.py b/tests/proton/test_proton_callback_handler.py
index 2b1e18e7..6e96ca24 100644
--- a/tests/proton/test_proton_callback_handler.py
+++ b/tests/proton/test_proton_callback_handler.py
@@ -1,22 +1,32 @@
+from typing import Optional
+
 from arrow import Arrow
 
+from app import config
 from app.account_linking import (
     SLPlan,
     SLPlanType,
 )
 from app.constants import JobType
-from app.proton.proton_client import ProtonClient, UserInformation
+from app.models import User, PartnerUser, Job, JobState
 from app.proton.proton_callback_handler import (
     ProtonCallbackHandler,
     generate_account_not_allowed_to_log_in,
 )
-from app.models import User, PartnerUser, Job, JobState
+from app.proton.proton_client import ProtonClient, UserInformation
 from app.proton.proton_partner import get_proton_partner
 from app.utils import random_string
-from typing import Optional
 from tests.utils import random_email
 
 
+def setup_module():
+    config.SKIP_MX_LOOKUP_ON_CHECK = True
+
+
+def teardown_module():
+    config.SKIP_MX_LOOKUP_ON_CHECK = False
+
+
 class MockProtonClient(ProtonClient):
     def __init__(self, user: Optional[UserInformation]):
         self.user = user