X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Ftest_remove_old_files.py;h=67da601952463f0c5379af8614b0b371aa7332f7;hb=0d59edf64d7d550a41fc723475eeb6fb3e65537e;hp=6f615ba0d57331eaba41cfa3b9528fc3c2802771;hpb=947977ce851e15e0065630f265b4047cd47ea1f1;p=ppu.git

diff --git a/tests/test_remove_old_files.py b/tests/test_remove_old_files.py
index 6f615ba..67da601 100755
--- a/tests/test_remove_old_files.py
+++ b/tests/test_remove_old_files.py
@@ -29,31 +29,49 @@ def create_files(files, subdirectory=None):
             pass
 
 
-def _test_files_exist(files):
+def assert_files_exist(files):
+    if isinstance(files, str):
+        files = [files]
     for fname in files:
         assert os.path.exists(fname)
 
 
-def _test_files_not_exist(files):
+def assert_files_not_exist(files):
+    if isinstance(files, str):
+        files = [files]
     for fname in files:
         assert not os.path.exists(fname)
 
 
 def test_remove_old_files():
     create_files(['test1', 'test2'])
-    _test_files_exist(['test1', 'test2'])
+    assert_files_exist(['test1', 'test2'])
     os.utime('test2', (0, 0))
     assert os.system("remove-old-files.py --older 100 .") == 0
-    _test_files_exist(['test1'])
-    _test_files_not_exist(['test2'])
+    assert_files_exist('test1')
+    assert_files_not_exist('test2')
 
 
 def test_recursive():
     create_files(['test3', 'test4'], 'subdir')
     test3 = os.path.join('subdir', 'test3')
     test4 = os.path.join('subdir', 'test4')
-    _test_files_exist([test3, test4])
+    assert_files_exist([test3, test4])
     os.utime(test4, (0, 0))
     assert os.system("remove-old-files.py --older 100 .") == 0
-    _test_files_exist([test3])
-    _test_files_not_exist([test4])
+    assert_files_exist(test3)
+    assert_files_not_exist(test4)
+
+
+def test_remove_empty_directory():
+    create_files(['test3', 'test4'], 'subdir')
+    test3 = os.path.join('subdir', 'test3')
+    test4 = os.path.join('subdir', 'test4')
+    assert_files_exist([test3, test4])
+    os.utime(test3, (0, 0))
+    os.utime(test4, (0, 0))
+    assert os.system("remove-old-files.py --older 100 .") == 0
+    assert_files_exist('subdir')
+    assert_files_not_exist([test3, test4])
+    assert os.system("remove-old-files.py -e --older 100 .") == 0
+    assert_files_not_exist('subdir')