add test

Author: casperdcl

tests/test_shtab.py

@@ -6,6 +6,7 @@
 """
 import logging
 import subprocess
+from argparse import ArgumentParser
 
 import pytest
 
@@ -69,6 +70,14 @@ def test_choices():
     assert "" not in shtab.Required.FILE
 
 
+@pytest.mark.parametrize("shell", SUPPORTED_SHELLS)
+def test_main(shell, caplog):
+    with caplog.at_level(logging.INFO):
+        main(["-s", shell, "shtab.main.get_main_parser"])
+
+    assert not caplog.record_tuples
+
+
 @pytest.mark.parametrize("shell", SUPPORTED_SHELLS)
 def test_complete(shell, caplog):
     parser = get_main_parser()
@@ -84,8 +93,17 @@ def test_complete(shell, caplog):
 
 
 @pytest.mark.parametrize("shell", SUPPORTED_SHELLS)
-def test_main(shell, caplog):
+def test_positional_choices(shell, caplog):
+    parser = ArgumentParser(prog="test")
+    parser.add_argument("posA", choices=["one", "two"])
     with caplog.at_level(logging.INFO):
-        main(["-s", shell, "shtab.main.get_main_parser"])
+        completion = shtab.complete(parser, shell=shell)
+    print(completion)
+
+    if shell == "bash":
+        shell = Bash(completion)
+        # ideally should check completion
+        # https://github.com/iterative/shtab/issues/11
+        shell.compgen('-W "$_shtab_test_options_"', "", "-h --help")
 
     assert not caplog.record_tuples