Added a basic unit test for the save command

1 files changed, 26 insertions, 1 deletions

diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py
index 2005f678..6f169ffc 100644
--- a/tests/test_cmd2.py
+++ b/tests/test_cmd2.py
@@ -5,7 +5,7 @@ Cmd2 unit/functional testing
 Copyright 2016 Federico Ceratto <federico.ceratto@gmail.com>
 Released under MIT license, see LICENSE file
 """
-
+import os
 import mock
 from conftest import run_cmd, _normalize
 from six import StringIO
@@ -200,3 +200,28 @@ def test_base_cmdenvironment(base_app):
     out_params = set(out[2].split("Settable parameters: ")[1].split())
 
     assert settable_params == out_params
+
+
+def test_base_save(base_app, capsys):
+    # TODO: Use a temporary directory for the file
+    filename = 'deleteme.txt'
+    run_cmd(base_app, 'help')
+    run_cmd(base_app, 'help save')
+    run_cmd(base_app, 'save * {}'.format(filename))
+    out, err = capsys.readouterr()
+    assert out == 'Saved to deleteme.txt\n'
+
+    with open(filename) as f:
+        content = [line.strip() for line in f.readlines()]
+    expected = _normalize("""
+help
+
+help save
+
+save * deleteme.txt
+""")
+
+    assert content == expected
+
+    # Delete file that was created
+    os.remove(filename)