tests for main entry point

Author: BruceZhang1993

index_generator/__main__.py

@@ -36,9 +36,10 @@ def main():
         except TemplateNotFound as e:
             raise IndexGeneratorTemplateNotFound(str(e))
     except BaseException as e:
-        print('[Exception] ' + e.__class__.__name__ + ': ' + str(e))
-        if hasattr(e, 'hint'):
-            print(e.hint)
+        if e.__class__.__name__ != 'SystemExit':
+            print('[Exception] ' + e.__class__.__name__ + ': ' + str(e))
+            if hasattr(e, 'hint'):
+                print(e.hint)
 
 
 def app(args):

tests/test_main.py

@@ -0,0 +1,14 @@
+import sys
+from index_generator.__main__ import main
+from index_generator import *
+
+
+def test_app_version(capfd):
+    sys.argv = ['index_generator', '--version']
+    main()
+    out, err = capfd.readouterr()
+    assert out == APP_NAME + ' ' + APP_VERSION + ' ' + APP_URL + "\n"
+    sys.argv = ['index_generator', '-V']
+    main()
+    out, err = capfd.readouterr()
+    assert out == APP_NAME + ' ' + APP_VERSION + ' ' + APP_URL + "\n"