Backport: pytest.skip fix from #2952 (#3007)

fix from #2952
mamba-org · Nov 23, 2023 · 3994cff · 3994cff
1 parent 91c41db
commit 3994cff
Showing 1 changed file with 14 additions and 13 deletions.
diff --git a/micromamba/tests/test_menuinst.py b/micromamba/tests/test_menuinst.py
@@ -1,28 +1,21 @@
-import json
 import os
-import re
 import shutil
-import subprocess
 import sys
 from pathlib import Path
 
 import pytest
 
-if not sys.platform.startswith("win"):
-    pytest.skip("skipping windows-only tests", allow_module_level=True)
-
-import menuinst
-import win32com.client
-
 from .helpers import create, get_env, get_umamba, random_string, remove, umamba_list
 
+if sys.platform.startswith("win"):
+    import menuinst
+    import win32com.client
+
 
 class TestMenuinst:
     root_prefix = os.environ["MAMBA_ROOT_PREFIX"]
     current_prefix = os.environ["CONDA_PREFIX"]
 
-    dirs = menuinst.win32.dirs_src
-
     @classmethod
     def setup_class(cls):
         pass
@@ -31,12 +24,16 @@ def setup_class(cls):
     def teardown_class(cls):
         pass
 
+    @pytest.mark.skipif(
+        not sys.platform.startswith("win"),
+        reason="skipping windows-only tests",
+    )
     def test_simple_shortcut(self):
         env_name = random_string()
         # "--json"
         create("miniforge_console_shortcut=1.0", "-n", env_name, no_dry_run=True)
         prefix = os.path.join(self.root_prefix, "envs", env_name)
-        d = self.dirs["user"]["start"][0]
+        d = menuinst.win32.dirs_src["user"]["start"][0]
         lnk = os.path.join(d, "Miniforge", "Miniforge Prompt (" + env_name + ").lnk")
 
         assert os.path.exists(lnk)
@@ -60,6 +57,10 @@ def test_simple_shortcut(self):
         remove("miniforge_console_shortcut", "-n", env_name, no_dry_run=True)
         assert not os.path.exists(lnk)
 
+    @pytest.mark.skipif(
+        not sys.platform.startswith("win"),
+        reason="skipping windows-only tests",
+    )
     def test_shortcut_weird_env(self):
         # note Umlauts do not work yet
         os.environ["MAMBA_ROOT_PREFIX"] = str(Path("./compl i c ted").absolute())
@@ -69,7 +70,7 @@ def test_shortcut_weird_env(self):
         # "--json"
         create("miniforge_console_shortcut=1.0", "-n", env_name, no_dry_run=True)
         prefix = os.path.join(root_prefix, "envs", env_name)
-        d = self.dirs["user"]["start"][0]
+        d = menuinst.win32.dirs_src["user"]["start"][0]
         lnk = os.path.join(d, "Miniforge", "Miniforge Prompt (" + env_name + ").lnk")
 
         assert os.path.exists(lnk)