diff --git a/tests/test_utils.py b/tests/test_utils.py index 482f2f7..e6c141d 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -67,6 +67,13 @@ class Test_which(unittest.TestCase): class Test_statfiles(unittest.TestCase): + def test_statfiles_dir_in_result(self): + '''Assumes that directory ./testdata/ exists here''' + here = os.path.dirname(os.path.abspath(__file__)) + statinfo = statfiles(here, relative=here) + self.assertIsInstance(statinfo, dict) + self.assertTrue(os.path.basename('testdata') in statinfo) + def test_statfiles_dir_rel(self): here = os.path.dirname(os.path.abspath(__file__)) statinfo = statfiles(here, relative=here) diff --git a/tldp/utils.py b/tldp/utils.py index d119962..e0fbcb2 100644 --- a/tldp/utils.py +++ b/tldp/utils.py @@ -194,7 +194,10 @@ def statfiles(name, relative=None): del statinfo[relpath] else: for root, dirs, files in os.walk(name): - for x in files: + inodes = list() + inodes.extend(dirs) + inodes.extend(files) + for x in inodes: foundpath = os.path.join(root, x) if relative: relpath = os.path.relpath(foundpath, start=relative)