test iter_find_files max depth

mahmoud · Jun 30, 2024 · d9a927b · d9a927b
1 parent ee3c660
commit d9a927b
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/tests/test_fileutils.py b/tests/test_fileutils.py
@@ -1,4 +1,10 @@
-from boltons.fileutils import FilePerms
+import os.path
+
+from boltons import fileutils
+from boltons.fileutils import FilePerms, iter_find_files
+
+
+BOLTONS_PATH = os.path.dirname(os.path.abspath(fileutils.__file__))
 
 
 def test_fileperms():
@@ -17,3 +23,14 @@ def test_fileperms():
     assert oct(int(up)) == '0o770'
 
     assert int(FilePerms()) == 0
+
+
+def test_iter_find_files():
+    def _to_baseless_list(paths):
+        return [p.lstrip(BOLTONS_PATH) for p in paths]
+
+    assert 'fileutils.py' in _to_baseless_list(iter_find_files(BOLTONS_PATH, patterns=['*.py']))
+
+    boltons_parent = os.path.dirname(BOLTONS_PATH)
+    assert 'fileutils.py' in _to_baseless_list(iter_find_files(boltons_parent, patterns=['*.py']))
+    assert 'fileutils.py' not in _to_baseless_list(iter_find_files(boltons_parent, patterns=['*.py'], max_depth=0))