diff --git a/tests/unit/python/plugins/lookup/test_sso.py b/tests/unit/python/plugins/lookup/test_sso.py index 050fd488b0..9f01f26824 100644 --- a/tests/unit/python/plugins/lookup/test_sso.py +++ b/tests/unit/python/plugins/lookup/test_sso.py @@ -19,25 +19,26 @@ from . import PROJECT_ROOT def _load_module(rel_path: str, name: str): - for key in ( - "utils.roles.applications.config", - "utils.roles.applications", - "utils", - "utils.cache.applications", - "utils.cache", - "utils.roles.applications.services.sso", - "utils.roles.applications.services", - ): - sys.modules.pop(key, None) - importlib.import_module("utils.roles.applications.config") - importlib.import_module("utils.cache.applications") - importlib.import_module("utils.roles.applications.services.sso") + with mock.patch.dict(sys.modules): + for key in ( + "utils.roles.applications.config", + "utils.roles.applications", + "utils", + "utils.cache.applications", + "utils.cache", + "utils.roles.applications.services.sso", + "utils.roles.applications.services", + ): + sys.modules.pop(key, None) + importlib.import_module("utils.roles.applications.config") + importlib.import_module("utils.cache.applications") + importlib.import_module("utils.roles.applications.services.sso") - path = PROJECT_ROOT / rel_path - spec = importlib.util.spec_from_file_location(name, str(path)) - mod = importlib.util.module_from_spec(spec) - assert spec and spec.loader - spec.loader.exec_module(mod) + path = PROJECT_ROOT / rel_path + spec = importlib.util.spec_from_file_location(name, str(path)) + mod = importlib.util.module_from_spec(spec) + assert spec and spec.loader + spec.loader.exec_module(mod) return mod diff --git a/tests/unit/python/plugins/lookup/test_sso_oidc_plugin.py b/tests/unit/python/plugins/lookup/test_sso_oidc_plugin.py index af94622a39..30eb3ec8e3 100644 --- a/tests/unit/python/plugins/lookup/test_sso_oidc_plugin.py +++ b/tests/unit/python/plugins/lookup/test_sso_oidc_plugin.py @@ -13,22 +13,23 @@ def _load_module(rel_path: str, name: str): """Import ``rel_path`` fresh, evicting sibling-test stubs of ``utils.roles.applications.config`` from ``sys.modules`` first. """ - for key in ( - "utils.roles.applications.config", - "utils.roles.applications", - "utils", - "utils.cache.applications", - "utils.cache", - ): - sys.modules.pop(key, None) - importlib.import_module("utils.roles.applications.config") - importlib.import_module("utils.cache.applications") + with mock.patch.dict(sys.modules): + for key in ( + "utils.roles.applications.config", + "utils.roles.applications", + "utils", + "utils.cache.applications", + "utils.cache", + ): + sys.modules.pop(key, None) + importlib.import_module("utils.roles.applications.config") + importlib.import_module("utils.cache.applications") - path = PROJECT_ROOT / rel_path - spec = importlib.util.spec_from_file_location(name, str(path)) - mod = importlib.util.module_from_spec(spec) - assert spec and spec.loader - spec.loader.exec_module(mod) + path = PROJECT_ROOT / rel_path + spec = importlib.util.spec_from_file_location(name, str(path)) + mod = importlib.util.module_from_spec(spec) + assert spec and spec.loader + spec.loader.exec_module(mod) return mod diff --git a/tests/unit/python/plugins/lookup/test_sso_proxy_consumers.py b/tests/unit/python/plugins/lookup/test_sso_proxy_consumers.py index 3fd055e309..7d17a9d5ae 100644 --- a/tests/unit/python/plugins/lookup/test_sso_proxy_consumers.py +++ b/tests/unit/python/plugins/lookup/test_sso_proxy_consumers.py @@ -19,25 +19,26 @@ from . import PROJECT_ROOT def _load_module(rel_path: str, name: str): - for key in ( - "utils.roles.applications.config", - "utils.roles.applications", - "utils", - "utils.cache.applications", - "utils.cache", - "utils.roles.applications.services.sso", - "utils.roles.applications.services", - ): - sys.modules.pop(key, None) - importlib.import_module("utils.roles.applications.config") - importlib.import_module("utils.cache.applications") - importlib.import_module("utils.roles.applications.services.sso") + with mock.patch.dict(sys.modules): + for key in ( + "utils.roles.applications.config", + "utils.roles.applications", + "utils", + "utils.cache.applications", + "utils.cache", + "utils.roles.applications.services.sso", + "utils.roles.applications.services", + ): + sys.modules.pop(key, None) + importlib.import_module("utils.roles.applications.config") + importlib.import_module("utils.cache.applications") + importlib.import_module("utils.roles.applications.services.sso") - path = PROJECT_ROOT / rel_path - spec = importlib.util.spec_from_file_location(name, str(path)) - mod = importlib.util.module_from_spec(spec) - assert spec and spec.loader - spec.loader.exec_module(mod) + path = PROJECT_ROOT / rel_path + spec = importlib.util.spec_from_file_location(name, str(path)) + mod = importlib.util.module_from_spec(spec) + assert spec and spec.loader + spec.loader.exec_module(mod) return mod