add tests

Author: BruceZhang1993

index_generator/__main__.py

@@ -8,7 +8,7 @@
 
 from jinja2.exceptions import TemplateNotFound
 from index_generator.models.entries import Entry
-from index_generator.models.exceptions import IndexGeneratorTemplateNotFound
+from index_generator.models.exceptions import IndexGeneratorTemplateNotFound, IndexGeneratorPathNotExists
 from . import APP_NAME, APP_URL, APP_VERSION
 
 indexIgnore = ('index.html', 'images', 'favicon.ico')
@@ -51,6 +51,8 @@ def app(args):
         print('Usage: index-generator [OPTIONS] PATH.')
         print('See: index-generator --help')
         sys.exit(0)
+    if not os.path.exists(args.path):
+        raise IndexGeneratorPathNotExists('Path does not exists')
     if args.no_recursive:
         os.chdir(args.path)
         generate_once(args.theme, '.', os.listdir('.'), args.name, args.print, base=args.root, human=args.human,

index_generator/models/exceptions.py

@@ -7,3 +7,8 @@ def __init__(self, message, hint='Something went wrong.'):
 class IndexGeneratorTemplateNotFound(IndexGeneratorBaseException):
     def __init__(self, message):
         super().__init__(message, hint='Template folder does not exists or template file is not found.')
+
+
+class IndexGeneratorPathNotExists(IndexGeneratorBaseException):
+    def __init__(self, message):
+        super().__init__(message, hint='Target path does not exists.')

tests/test_main.py

@@ -19,3 +19,21 @@ def test_app_missing_argument(capfd):
     main()
     out, _ = capfd.readouterr()
     assert out == APP_NAME + ' ' + APP_VERSION + ' ' + APP_URL + "\nUsage: index-generator [OPTIONS] PATH.\nSee: index-generator --help\n"
+
+
+def test_app_template_not_found(capfd):
+    sys.argv = ['index_generator', '--template', '/tmp/not_existed', '/tmp']
+    main()
+    out, _ = capfd.readouterr()
+    assert 'IndexGeneratorTemplateNotFound' in out
+    sys.argv = ['index_generator', '-T', '/tmp/not_existed', '/tmp']
+    main()
+    out, _ = capfd.readouterr()
+    assert 'IndexGeneratorTemplateNotFound' in out
+
+
+def test_app_path_not_exists(capfd):
+    sys.argv = ['index_generator', '/tmp/not_existed']
+    main()
+    out, _ = capfd.readouterr()
+    assert 'IndexGeneratorPathNotExists' in out