diff --git a/backend/tests/unit/test_action_item_reminder_cancel_on_complete.py b/backend/tests/unit/test_action_item_reminder_cancel_on_complete.py index c187e59c55..de3b216015 100644 --- a/backend/tests/unit/test_action_item_reminder_cancel_on_complete.py +++ b/backend/tests/unit/test_action_item_reminder_cancel_on_complete.py @@ -53,11 +53,25 @@ def exec_module(self, m): pass +def _install_stub_module(name): + module = _AutoMock(name) + sys.modules[name] = module + parent_name, _, attr = name.rpartition(".") + parent = sys.modules.get(parent_name) + if parent is not None: + setattr(parent, attr, module) + return module + + def _load_real_notifications(): sys.path.insert(0, str(BACKEND_DIR)) finder = _Finder() sys.meta_path.insert(0, finder) try: + firebase_admin = _install_stub_module("firebase_admin") + firebase_admin.messaging = _install_stub_module("firebase_admin.messaging") + firebase_admin.auth = _install_stub_module("firebase_admin.auth") + import utils.notifications as notif return notif diff --git a/backend/tests/unit/test_batch_upload_storage.py b/backend/tests/unit/test_batch_upload_storage.py index 08045593a2..ace9265d05 100644 --- a/backend/tests/unit/test_batch_upload_storage.py +++ b/backend/tests/unit/test_batch_upload_storage.py @@ -17,6 +17,9 @@ # Mock heavy dependencies at sys.modules level before importing storage sys.modules.setdefault("database._client", MagicMock()) +sys.modules.setdefault("database.redis_db", MagicMock()) +sys.modules.setdefault("database.users", MagicMock()) +sys.modules.setdefault("opuslib", MagicMock()) _mock_gcs_storage = MagicMock() _mock_gcs_client_instance = MagicMock() @@ -586,6 +589,8 @@ def setup_class(cls): 'database.tasks', 'database.plugins', 'database.notifications', + 'database.vector_db', + 'pinecone', ]: sys.modules.setdefault(mod_name, MagicMock())