Add failing test for StashedUninstallPathSet symlink

tests/unit/test_req_uninstall.py

@@ -306,3 +306,67 @@ class TestStashedUninstallPathSet(object):
         for old_path, new_path in stashed_paths:
             assert os.path.exists(old_path)
             assert not os.path.exists(new_path)
+
+    @pytest.mark.skipif("sys.platform == 'win32'")
+    def test_commit_symlinks(self, tmpdir):
+        adir = tmpdir / "dir"
+        adir.mkdir()
+        dirlink = tmpdir / "dirlink"
+        dirlink.symlink_to(adir)
+        afile = tmpdir / "file"
+        afile.write_text("...")
+        filelink = tmpdir / "filelink"
+        filelink.symlink_to(afile)
+
+        pathset = StashedUninstallPathSet()
+        stashed_paths = []
+        stashed_paths.append(pathset.stash(dirlink))
+        stashed_paths.append(pathset.stash(filelink))
+        for stashed_path in stashed_paths:
+            assert os.path.lexists(stashed_path)
+        assert not os.path.exists(dirlink)
+        assert not os.path.exists(filelink)
+
+        pathset.commit()
+
+        # stash removed, links removed
+        for stashed_path in stashed_paths:
+            assert not os.path.lexists(stashed_path)
+        assert not os.path.lexists(dirlink) and not os.path.isdir(dirlink)
+        assert not os.path.lexists(filelink) and not os.path.isfile(filelink)
+
+        # link targets untouched
+        assert os.path.isdir(adir)
+        assert os.path.isfile(afile)
+
+    @pytest.mark.skipif("sys.platform == 'win32'")
+    def test_rollback_symlinks(self, tmpdir):
+        adir = tmpdir / "dir"
+        adir.mkdir()
+        dirlink = tmpdir / "dirlink"
+        dirlink.symlink_to(adir)
+        afile = tmpdir / "file"
+        afile.write_text("...")
+        filelink = tmpdir / "filelink"
+        filelink.symlink_to(afile)
+
+        pathset = StashedUninstallPathSet()
+        stashed_paths = []
+        stashed_paths.append(pathset.stash(dirlink))
+        stashed_paths.append(pathset.stash(filelink))
+        for stashed_path in stashed_paths:
+            assert os.path.lexists(stashed_path)
+        assert not os.path.lexists(dirlink)
+        assert not os.path.lexists(filelink)
+
+        pathset.rollback()
+
+        # stash removed, links restored
+        for stashed_path in stashed_paths:
+            assert not os.path.lexists(stashed_path)
+        assert os.path.lexists(dirlink) and os.path.isdir(dirlink)
+        assert os.path.lexists(filelink) and os.path.isfile(filelink)
+
+        # link targets untouched
+        assert os.path.isdir(adir)
+        assert os.path.isfile(afile)