Create argparse type "existing_dir[_empty]"

[pyi2ncommon] / test / test_argparse_helpers.py

diff --git a/test/test_argparse_helpers.py b/test/test_argparse_helpers.py
index 175219e..f0c77e2 100644 (file)
--- a/test/test_argparse_helpers.py
+++ b/test/test_argparse_helpers.py
@@ -27,8 +27,9 @@ For help see :py:mod:`unittest`
 
 import unittest
 import os
-from os.path import isfile
-from tempfile import mkstemp
+from os.path import isfile, isdir
+from shutil import rmtree
+from tempfile import mkstemp, mkdtemp
 
 # relative import of tested module ensures we do not test installed version
 try:
@@ -61,6 +62,8 @@ class ArgparseTester(unittest.TestCase):
             # test with existing file
             parser = argparse_helpers.NonExitingParser()
             parser.add_argument('--input', type=argparse_helpers.existing_file)
+            parser.add_argument('--input2', type=argparse_helpers.existing_file_or_empty)
+            parser.parse_args(['--input', temp_file, '--input2', temp_file])
         finally:
             if temp_handle:
                 os.close(temp_handle)
@@ -73,5 +76,45 @@ class ArgparseTester(unittest.TestCase):
                                parser.parse_args,
                                ['--input', temp_file, ])
 
+    def test_existing_dir(self):
+        temp_dir = None
+        try:
+            # create a dir
+            temp_dir = mkdtemp()
+
+            # test with existing dir
+            parser = argparse_helpers.NonExitingParser()
+            parser.add_argument('--input', type=argparse_helpers.existing_dir)
+            parser.add_argument('--input2', type=argparse_helpers.existing_dir_or_empty)
+            parser.parse_args(['--input', temp_dir, '--input2', temp_dir])
+        finally:
+            if temp_dir and isdir(temp_dir):
+                rmtree(temp_dir)
+
+        # test with non-existing dir
+        self.assertRaisesRegex(argparse_helpers.ArgParserWantsExit,
+                               'is not an existing directory',
+                               parser.parse_args,
+                               ['--input', temp_dir, ])
+
+    def test_allow_empty(self):
+        parser = argparse_helpers.NonExitingParser()
+        parser.add_argument('--file', type=argparse_helpers.existing_file_or_empty)
+        parser.add_argument('--dir', type=argparse_helpers.existing_dir_or_empty)
+
+        # both emtpy: no problem
+        parser.parse_args(['--file', '', '--dir', ''])
+
+        # test with non-existing file/dir
+        self.assertRaisesRegex(argparse_helpers.ArgParserWantsExit,
+                               'is not an existing file',
+                               parser.parse_args,
+                               ['--dir', '', '--file', 'not-an-existing-file'])
+        self.assertRaisesRegex(argparse_helpers.ArgParserWantsExit,
+                               'is not an existing directory',
+                               parser.parse_args,
+                               ['--file', '', '--dir', 'not-an-existing-dir'])
+
+
 if __name__ == '__main__':
     unittest.main()