factor out symlink_or_skip

Author: blueyed

testing/acceptance_test.py

@@ -762,21 +762,13 @@ def test():
         result = testdir.runpytest(str(p) + "::test", "--doctest-modules")
         result.stdout.fnmatch_lines(["*1 passed*"])
 
-    def test_cmdline_python_package_symlink(self, testdir, monkeypatch):
+    def test_cmdline_python_package_symlink(
+        self, testdir, monkeypatch, symlink_or_skip
+    ):
         """
         test --pyargs option with packages with path containing symlink can
         have conftest.py in their package (#2985)
         """
-        # dummy check that we can actually create symlinks: on Windows `os.symlink` is available,
-        # but normal users require special admin privileges to create symlinks.
-        if sys.platform == "win32":
-            try:
-                os.symlink(
-                    str(testdir.tmpdir.ensure("tmpfile")),
-                    str(testdir.tmpdir.join("tmpfile2")),
-                )
-            except OSError as e:
-                pytest.skip(str(e.args[0]))
         monkeypatch.delenv("PYTHONDONTWRITEBYTECODE", raising=False)
 
         dirname = "lib"
@@ -794,7 +786,7 @@ def test_cmdline_python_package_symlink(self, testdir, monkeypatch):
 
         d_local = testdir.mkdir("local")
         symlink_location = os.path.join(str(d_local), "lib")
-        os.symlink(str(d), symlink_location, target_is_directory=True)
+        symlink_or_skip(str(d), symlink_location, target_is_directory=True)
 
         # The structure of the test directory is now:
         # .

testing/conftest.py

@@ -1,3 +1,4 @@
+import os
 import re
 import sys
 from typing import List
@@ -136,6 +137,24 @@ def testdir(testdir: Testdir) -> Testdir:
     return testdir
 
 
+@pytest.fixture
+def symlink_or_skip():
+    """Return a function to create symlinks or skip the test.
+
+    On Windows `os.symlink` is available, but normal users require special
+    admin privileges to create symlinks.
+    """
+
+    def wrap_os_symlink(src, dst, *args, **kwargs):
+        try:
+            os.symlink(src, dst, *args, **kwargs)
+        except OSError as e:
+            pytest.skip("os.symlink({!r}) failed: {!r}".format((src, dst), e))
+        assert os.path.islink(dst)
+
+    return wrap_os_symlink
+
+
 @pytest.fixture(scope="session")
 def color_mapping():
     """Returns a utility class which can replace keys in strings in the form "{NAME}"