Add test_cli.py

References:
http://stackoverflow.com/questions/18160078/how-do-you-write-tests-for-the-argparse-portion-of-a-python-module
http://dustinrcollins.com/testing-python-command-line-apps
https://github.com/mdklatt/cookiecutter-python-app

2 files changed, 48 insertions, 1 deletions

diff --git a/setup.cfg b/setup.cfg
index abc0206..fea37fb 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -2,7 +2,7 @@
 universal = 1
 
 [pytest]
-xfail_strict=true
+xfail_strict = True
 
 [flake8]
 exclude =
@@ -10,3 +10,8 @@ exclude =
 ignore =
     W503,
     E731
+
+[coverage:run]
+branch = False
+omit =
+    sqlparse/__main__.py
diff --git a/tests/test_cli.py b/tests/test_cli.py
new file mode 100644
index 0000000..81d8449
--- /dev/null
+++ b/tests/test_cli.py
@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+import os
+import subprocess
+import sys
+
+import pytest
+
+import sqlparse
+import sqlparse.__main__
+from tests.utils import FILES_DIR
+
+path = os.path.join(FILES_DIR, 'function.sql')
+
+
+def test_cli_main_empty():
+    with pytest.raises(SystemExit):
+        sqlparse.cli.main([])
+
+
+def test_parser_empty():
+    with pytest.raises(SystemExit):
+        parser = sqlparse.cli.create_parser()
+        parser.parse_args([])
+
+
+def test_main_help():
+    # Call with the --help option as a basic sanity check.
+    with pytest.raises(SystemExit) as exinfo:
+        sqlparse.cli.main(["--help", ])
+    assert exinfo.value.code == 0
+
+
+def test_valid_args():
+    # test doesn't abort
+    assert sqlparse.cli.main([path, '-r']) is not None
+
+
+def test_script():
+    # Call with the --help option as a basic sanity check.
+    cmdl = "{0:s} -m sqlparse.cli --help".format(sys.executable)
+    assert subprocess.call(cmdl.split()) == 0