From 1497cadc5d2837642d6d100042ead099d5745bec Mon Sep 17 00:00:00 2001
From: Ken Odegard <kodegard@anaconda.com>
Date: Wed, 27 Mar 2024 16:16:33 -0500
Subject: [PATCH] Add test_prefix_files

---
 tests/test_utils.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/tests/test_utils.py b/tests/test_utils.py
index d245e65796..ae2939ba4d 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -433,3 +433,19 @@ def test_is_conda_pkg(tmpdir, value: str, expected: bool, is_dir: bool, create:
                 fp.write("test")
 
     assert utils.is_conda_pkg(value) == expected
+
+
+def test_prefix_files(tmp_path: Path):
+    (prefix := tmp_path / "prefix1").mkdir()
+    (dirA := prefix / "dirA").mkdir()
+    (dirB := prefix / "dirB").mkdir()
+    (file1 := dirA / "file1").touch()
+    (file2 := dirB / "file2").touch()
+
+    (prefix2 := tmp_path / "prefix2").mkdir()
+    (dirC := prefix2 / "dirC").mkdir()
+    (link1 := prefix / "dirC").symlink_to(dirC)
+    (link1 / "file3").touch()
+
+    paths = {str(path.relative_to(prefix)) for path in (file1, file2, link1)}
+    assert paths == utils.prefix_files(str(prefix))