Added more unit tests for py and pyscript commands

Author: tleonhardt

CHANGES.md

@@ -4,7 +4,7 @@ News
 0.7.4
 -----
 
-*Release date: TBD*
+*Release date: 2017-07-TBD*
 
 * Bug fixes
     * Fixed a couple bugs in interacting with pastebuffer/clipboard on macOS and Linux
@@ -18,8 +18,10 @@ News
         * These also strongly felt out of place
         * ``load`` and ``_relative_load`` now require a file path
         * ``edit`` and ``save`` now use a temporary file if a file path isn't provided
-    * Load command has better error checking and reporting
-    * Clipboard copy and paste functionality is now handled by the ``pyperclip`` module
+    * ``load`` command has better error checking and reporting
+    * Clipboard copy and paste functionality is now handled by the **pyperclip** module
+        * NOTE: This adds an additional required 3rd-party dependency
+    * Added a lot of unit tests
 
 0.7.3
 -----

tests/scripts/raises_exception.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python
+# coding=utf-8
+"""
+Example demonstrating what happens when a Python script raises an exception
+"""
+1 + 'blue'

tests/scripts/recursive.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python
+# coding=utf-8
+"""
+Example demonstrating that running a Python script recursively inside another Python script isn't allowed
+"""
+cmd('pyscript ../script.py')

tests/test_cmd2.py

@@ -142,6 +142,34 @@ def test_base_run_pyscript(base_app, capsys, request):
     out, err = capsys.readouterr()
     assert out == expected
 
+def test_recursive_pyscript_not_allowed(base_app, capsys, request):
+    test_dir = os.path.dirname(request.module.__file__)
+    python_script = os.path.join(test_dir, 'scripts', 'recursive.py')
+    expected = 'ERROR: Recursively entering interactive Python consoles is not allowed.\n'
+
+    run_cmd(base_app, "pyscript {}".format(python_script))
+    out, err = capsys.readouterr()
+    assert err == expected
+
+def test_pyscript_with_nonexist_file(base_app, capsys):
+    python_script = 'does_not_exist.py'
+    run_cmd(base_app, "pyscript {}".format(python_script))
+    out, err = capsys.readouterr()
+    assert err.startswith('ERROR: [Errno 2] No such file or directory:')
+
+def test_pyscript_with_exception(base_app, capsys, request):
+    test_dir = os.path.dirname(request.module.__file__)
+    python_script = os.path.join(test_dir, 'scripts', 'raises_exception.py')
+    run_cmd(base_app, "pyscript {}".format(python_script))
+    out, err = capsys.readouterr()
+    assert err.startswith('Traceback')
+    assert err.endswith("TypeError: unsupported operand type(s) for +: 'int' and 'str'\n")
+
+def test_pyscript_requires_an_argument(base_app, capsys):
+    run_cmd(base_app, "pyscript")
+    out, err = capsys.readouterr()
+    assert err.startswith('ERROR: pyscript command requires at least 1 argument ...')
+
 
 def test_base_error(base_app):
     out = run_cmd(base_app, 'meow')
@@ -957,3 +985,21 @@ def test_select_uneven_list_of_tuples(select_app):
 
     # And verify the expected output to stdout
     assert out == expected
+
+@pytest.fixture
+def noarglist_app():
+    cmd2.set_use_arg_list(False)
+    app = cmd2.Cmd()
+    app.stdout = StdOut()
+    return app
+
+def test_pyscript_with_noarglist(noarglist_app, capsys, request):
+    test_dir = os.path.dirname(request.module.__file__)
+    python_script = os.path.join(test_dir, '..', 'examples', 'scripts', 'arg_printer.py')
+    expected = """Running Python script 'arg_printer.py' which was called with 2 arguments
+arg 1: 'foo'
+arg 2: 'bar'
+"""
+    run_cmd(noarglist_app, 'pyscript {} foo bar'.format(python_script))
+    out, err = capsys.readouterr()
+    assert out == expected