Use setup_loader for cmdmod test

dwoz · Mar 10, 2021 · 355fbc2 · 355fbc2
1 parent 149153a
commit 355fbc2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/pytests/unit/modules/test_cmdmod.py b/tests/pytests/unit/modules/test_cmdmod.py
@@ -25,9 +25,11 @@
 MOCK_SHELL_FILE = "# List of acceptable shells\n" "\n" "/bin/bash\n"
 
 
-@pytest.fixture
-def configure_loader_modules():
-    return {cmdmod: {}}
+@pytest.fixture(autouse=True)
+def setup_loader():
+    setup_loader_modules = {cmdmod: {}}
+    with pytest.helpers.loader_mock(setup_loader_modules) as loader_mock:
+        yield loader_mock
 
 
 @pytest.fixture(scope="module")