From 51bc7e7a3d8c8ad3a013af8026fedfc395f1ad21 Mon Sep 17 00:00:00 2001
From: Kyle Benesch <4b796c65+github@gmail.com>
Date: Sun, 11 Jul 2021 16:14:24 -0700
Subject: [PATCH] Add tests for walk_library and walk_directory.

---
 delocate/tests/test_libsana.py | 63 +++++++++++++++++++++++++++++++---
 1 file changed, 58 insertions(+), 5 deletions(-)

diff --git a/delocate/tests/test_libsana.py b/delocate/tests/test_libsana.py
index 3daa53e2..c45214cb 100644
--- a/delocate/tests/test_libsana.py
+++ b/delocate/tests/test_libsana.py
@@ -5,13 +5,16 @@
 
 import os
 from os.path import (join as pjoin, dirname, realpath, relpath, split)
-from typing import Dict, Iterable, Text
+import shutil
+from typing import Any, Dict, Iterable, Text
 
 import pytest
 
+from ..delocating import filter_system_libs
 from ..libsana import (tree_libs, get_prefix_stripper, get_rp_stripper,
                        stripped_lib_dict, wheel_libs, resolve_rpath,
                        get_dependencies, resolve_dynamic_paths,
+                       walk_library, walk_directory,
                        DependencyNotFound)
 
 from ..tools import set_install_name
@@ -21,7 +24,7 @@
 from .pytest_tools import assert_equal
 
 from .test_install_names import (LIBA, LIBB, LIBC, TEST_LIB, _copy_libs,
-                                 EXT_LIBS, LIBSYSTEMB)
+                                 EXT_LIBS, LIBSYSTEMB, DATA_PATH)
 from .test_wheelies import (PLAT_WHEEL, PURE_WHEEL, STRAY_LIB_DEP)
 
 
@@ -218,8 +221,58 @@ def test_resolve_rpath():
     assert_equal(resolve_rpath(lib_rpath, []), lib_rpath)
 
 
-def test_get_dependencies():
-    # type: () -> None
+def test_get_dependencies(tmpdir):
+    # type: (Any) -> None
     with pytest.raises(DependencyNotFound):
         list(get_dependencies("nonexistent.lib"))
-    assert dict(get_dependencies(LIBA)) == {lib: lib for lib in EXT_LIBS}
+    ext_libs = {(lib, lib) for lib in EXT_LIBS}
+    assert set(get_dependencies(LIBA)) == ext_libs
+
+    os.symlink(
+        pjoin(DATA_PATH, "libextfunc_rpath.dylib"),
+        pjoin(tmpdir, "libextfunc_rpath.dylib"),
+    )
+    assert set(get_dependencies(pjoin(tmpdir, "libextfunc_rpath.dylib"),
+                                filt_func=filter_system_libs)) == {
+        (None, "@rpath/libextfunc2_rpath.dylib"), (LIBSYSTEMB, LIBSYSTEMB),
+    }
+
+    assert set(get_dependencies(pjoin(tmpdir, "libextfunc_rpath.dylib"),
+                                executable_path=DATA_PATH,
+                                filt_func=filter_system_libs)) == {
+        (pjoin(DATA_PATH, "libextfunc2_rpath.dylib"),
+         "@rpath/libextfunc2_rpath.dylib"
+         ),
+         (LIBSYSTEMB, LIBSYSTEMB),
+    }
+
+
+def test_walk_library():
+    # type: () -> None
+    with pytest.raises(DependencyNotFound):
+        list(walk_library("nonexistent.lib"))
+    assert set(walk_library(LIBA, filt_func=filter_system_libs)) == {LIBA, }
+    assert set(walk_library(pjoin(DATA_PATH, "libextfunc_rpath.dylib"),
+                            filt_func=filter_system_libs)) == {
+        pjoin(DATA_PATH, "libextfunc_rpath.dylib"),
+        pjoin(DATA_PATH, "libextfunc2_rpath.dylib"),
+    }
+
+
+def test_walk_directory(tmpdir):
+    # type: (Any) -> None
+    assert set(walk_directory(tmpdir)) == set()
+
+    shutil.copy(pjoin(DATA_PATH, "libextfunc_rpath.dylib"), tmpdir)
+    assert set(walk_directory(tmpdir, filt_func=filter_system_libs)) == {
+        pjoin(tmpdir, "libextfunc_rpath.dylib"),
+    }
+
+    assert set(
+        walk_directory(
+            tmpdir, executable_path=DATA_PATH, filt_func=filter_system_libs
+        )
+    ) == {
+        pjoin(tmpdir, "libextfunc_rpath.dylib"),
+        pjoin(DATA_PATH, "libextfunc2_rpath.dylib"),
+    }