From: Oleg Broytman <phd@phdru.name>
Date: Sun, 16 Apr 2017 07:32:23 +0000 (+0300)
Subject: Refactor tests: allow assert_files(one_file_name)
X-Git-Tag: 0.1.0~3
X-Git-Url: https://git.phdru.name/?a=commitdiff_plain;h=f753377d712c7af64ae840d82923d566b2c7bc8c;p=ppu.git

Refactor tests: allow assert_files(one_file_name)
---

diff --git a/tests/test_remove_old_files.py b/tests/test_remove_old_files.py
index 7c0c3c3..e50b6fa 100755
--- a/tests/test_remove_old_files.py
+++ b/tests/test_remove_old_files.py
@@ -30,11 +30,15 @@ def create_files(files, subdirectory=None):
 
 
 def assert_files_exist(files):
+    if isinstance(files, str):
+        files = [files]
     for fname in files:
         assert os.path.exists(fname)
 
 
 def assert_files_not_exist(files):
+    if isinstance(files, str):
+        files = [files]
     for fname in files:
         assert not os.path.exists(fname)
 
@@ -44,8 +48,8 @@ def test_remove_old_files():
     assert_files_exist(['test1', 'test2'])
     os.utime('test2', (0, 0))
     assert os.system("remove-old-files.py --older 100 .") == 0
-    assert_files_exist(['test1'])
-    assert_files_not_exist(['test2'])
+    assert_files_exist('test1')
+    assert_files_not_exist('test2')
 
 
 def test_recursive():
@@ -55,5 +59,5 @@ def test_recursive():
     assert_files_exist([test3, test4])
     os.utime(test4, (0, 0))
     assert os.system("remove-old-files.py --older 100 .") == 0
-    assert_files_exist([test3])
-    assert_files_not_exist([test4])
+    assert_files_exist(test3)
+    assert_files_not_exist(test4)