Added unit test for cmdloop()

1 files changed, 19 insertions, 1 deletions

diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py
index d1906fc4..d352bd4e 100644
--- a/tests/test_cmd2.py
+++ b/tests/test_cmd2.py
@@ -15,7 +15,7 @@ import six
 from code import InteractiveConsole
 
 import cmd2
-from conftest import run_cmd, normalize, BASE_HELP, HELP_HISTORY, SHORTCUTS_TXT, SHOW_TXT, SHOW_LONG
+from conftest import run_cmd, normalize, BASE_HELP, HELP_HISTORY, SHORTCUTS_TXT, SHOW_TXT, SHOW_LONG, StdOut
 
 
 def test_ver():
@@ -461,3 +461,21 @@ def test_base_py_interactive(base_app):
 
     # Make sure our mock was called once and only once
     m.assert_called_once()
+
+
+def test_base_cmdloop_with_queue(capsys):
+    # Create a cmd2.Cmd() instance and make sure basic settings are like we want for test
+    app = cmd2.Cmd()
+    app.use_rawinput = True
+    app.intro = 'Hello World, this is an intro ...'
+    app.cmdqueue.append('quit\n')
+    app.stdout = StdOut()
+
+    # Need to patch sys.argv so cmd2 doesn't think it was called with arguments
+    testargs = ["prog"]
+    expected = app.intro + '\n'
+    with mock.patch.object(sys, 'argv', testargs):
+        # Run the command loop
+        app.cmdloop()
+    out = app.stdout.buffer
+    assert out == expected