Merge pull request #571 from python-cmd2/argparse_remainder

Fixes related to handling of argparse.REMAINDER

1 files changed, 18 insertions, 1 deletions

diff --git a/tests/test_pyscript.py b/tests/test_pyscript.py
index 6b72e940..bcb72a3b 100644
--- a/tests/test_pyscript.py
+++ b/tests/test_pyscript.py
@@ -9,7 +9,7 @@ import os
 import pytest
 from cmd2.cmd2 import Cmd, with_argparser
 from cmd2 import argparse_completer
-from .conftest import run_cmd
+from .conftest import run_cmd, normalize
 from cmd2.utils import namedtuple_with_defaults, StdSim
 
 
@@ -234,3 +234,20 @@ def test_pyscript_custom_name(ps_echo, request):
     out = run_cmd(ps_echo, 'pyscript {}'.format(python_script))
     assert out
     assert message == out[0]
+
+
+def test_pyscript_argparse_checks(ps_app, capsys):
+    # Test command that has nargs.REMAINDER and make sure all tokens are accepted
+    run_cmd(ps_app, 'py app.alias.create("my_alias", "alias_command", "command_arg1", "command_arg2")')
+    out = run_cmd(ps_app, 'alias list my_alias')
+    assert out == normalize('alias create my_alias alias_command command_arg1 command_arg2')
+
+    # Specify flag outside of keyword argument
+    run_cmd(ps_app, 'py app.help("-h")')
+    _, err = capsys.readouterr()
+    assert '-h appears to be a flag' in err
+
+    # Specify list with flag outside of keyword argument
+    run_cmd(ps_app, 'py app.help(["--help"])')
+    _, err = capsys.readouterr()
+    assert '--help appears to be a flag' in err