Avoid mocking Anki modules globally

test/config/config_settings_spec.py

@@ -3,9 +3,10 @@
 from expects import expect, contain
 from mamba import describe, it, context
 
-from test_utils.anki import mock_anki_modules
+from test_utils.anki import MockAnkiModules
+
+mock_anki_modules = MockAnkiModules()
 
-mock_anki_modules()
 from aqt import mw
 
 from crowd_anki.config.config_settings import ConfigSettings, NoteSortingMethods
@@ -94,3 +95,5 @@
 
             expect(config._config.items()).to(contain(*list(new_settings.items())))
             addon_manager_mock.writeConfig.assert_called_once()
+
+mock_anki_modules.unmock()

test/export/anki_exporter_wrapper_spec.py

@@ -2,6 +2,10 @@
 
 from mamba import describe, it, context
 
+from test_utils.anki import MockAnkiModules
+
+mock_anki_modules = MockAnkiModules(["win32file", "win32pipe", "pywintypes", "winerror"]) # Anki on Windows uses pywin32
+
 from crowd_anki.export.anki_exporter_wrapper import AnkiJsonExporterWrapper
 
 DUMMY_EXPORT_DIRECTORY = "/tmp"
@@ -23,3 +27,5 @@
 
             notifier_mock.warning.assert_called_once()
             exporter_mock.export_to_directory.assert_not_called()
+
+mock_anki_modules.unmock()

test_utils/anki/__init__.py

@@ -1,16 +1,31 @@
+from typing import List
+from typing import Optional
 import sys
 
 from unittest.mock import MagicMock
 
-
-def mock_anki_modules():
+class MockAnkiModules:
     """
     I'd like to get rid of the situation when this is required, but for now this helps with the situation that
     anki modules are not available during test runtime.
     """
+    module_names_list = ['anki', 'anki.hooks', 'anki.exporting', 'anki.decks', 'anki.utils', 'anki.cards', 'anki.models',
+                         'anki.notes', 'aqt', 'aqt.qt', 'aqt.exporting', 'aqt.utils']
+
+    def __init__(self, module_names_list: Optional[List[str]] = None):
+        if module_names_list is None:
+            module_names_list = self.module_names_list
+
+        self.shadowed_modules = {}
 
-    modules_list = ['anki', 'anki.hooks', 'anki.exporting', 'anki.decks', 'anki.utils', 'anki.cards', 'anki.models',
-                    'anki.notes', 'aqt''', 'aqt.qt', 'aqt.exporting', 'aqt.utils']
+        for module_name in module_names_list:
+            self.shadowed_modules[module_name] = sys.modules.get(module_name)
+            sys.modules[module_name] = MagicMock()
 
-    for module in modules_list:
-        sys.modules[module] = MagicMock()
+    def unmock(self):
+        for module_name, module in self.shadowed_modules.items():
+            if module is not None:
+                sys.modules[module_name] = module
+            else:
+                if module_name in sys.modules:
+                    del sys.modules[module_name]