summaryrefslogtreecommitdiff
path: root/tests/test_pyscript.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_pyscript.py')
-rw-r--r--tests/test_pyscript.py19
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