From f4c3013a7215e45d6200f14f6c8164a10693aacd Mon Sep 17 00:00:00 2001
From: Todd Leonhardt <todd.leonhardt@gmail.com>
Date: Sun, 2 Jul 2017 15:06:25 -0400
Subject: Added a bunch of unit tests

Also improved error handling in some exceptional cases.
---
 tests/test_parsing.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'tests/test_parsing.py')

diff --git a/tests/test_parsing.py b/tests/test_parsing.py
index dda29911..1a32a734 100644
--- a/tests/test_parsing.py
+++ b/tests/test_parsing.py
@@ -20,7 +20,6 @@ def hist():
     h = cmd2.History([HistoryItem('first'), HistoryItem('second'), HistoryItem('third'), HistoryItem('fourth')])
     return h
 
-
 @pytest.fixture
 def parser():
     c = cmd2.Cmd()
@@ -38,6 +37,11 @@ def input_parser():
     c = cmd2.Cmd()
     return c.parser_manager.input_source_parser
 
+@pytest.fixture
+def option_parser():
+    op = cmd2.OptionParser()
+    return op
+
 
 def test_remaining_args():
     assert cmd2.remaining_args('-f bar   bar   cow', ['bar', 'cow']) == 'bar   cow'
@@ -265,3 +269,11 @@ def test_parse_input_redirect_from_unicode_filename(input_parser):
     line = '< café'
     results = input_parser.parseString(line)
     assert results.inputFrom == line
+
+
+def test_option_parser_exit_with_msg(option_parser, capsys):
+    msg = 'foo bar'
+    option_parser.exit(msg=msg)
+    out, err = capsys.readouterr()
+    assert out == msg + '\n'
+    assert err == ''
-- 
cgit v1.2.1