Merge pull request #211 from kotfu/fix/feedback_to_output

.pfeedback() now honors feedback_to_output setting

Author: tleonhardt

cmd2.py

@@ -627,7 +627,7 @@ def pfeedback(self, msg):
             if self.feedback_to_output:
                 self.poutput(msg)
             else:
-                print(msg)
+                sys.stderr.write("{}\n".format(msg))
 
     def colorize(self, val, color):
         """Given a string (``val``), returns that string wrapped in UNIX-style

tests/test_cmd2.py

@@ -7,6 +7,7 @@
 """
 import os
 import sys
+import tempfile
 
 import mock
 import pytest
@@ -504,25 +505,63 @@ def test_save_invalid_path(base_app, capsys):
 
 
 def test_output_redirection(base_app):
-    # TODO: Use a temporary directory/file for this file
-    filename = 'out.txt'
-
-    # Verify that writing to a file works
-    run_cmd(base_app, 'help > {}'.format(filename))
-    expected = normalize(BASE_HELP)
-    with open(filename) as f:
-        content = normalize(f.read())
-    assert content == expected
-
-    # Verify that appending to a file also works
-    run_cmd(base_app, 'help history >> {}'.format(filename))
-    expected = normalize(BASE_HELP + '\n' + HELP_HISTORY)
-    with open(filename) as f:
-        content = normalize(f.read())
-    assert content == expected
-
-    # Delete file that was created
-    os.remove(filename)
+    fd, filename = tempfile.mkstemp(prefix='cmd2_test', suffix='.txt')
+    os.close(fd)
+    
+    try:
+        # Verify that writing to a file works
+        run_cmd(base_app, 'help > {}'.format(filename))
+        expected = normalize(BASE_HELP)
+        with open(filename) as f:
+            content = normalize(f.read())
+        assert content == expected
+
+        # Verify that appending to a file also works
+        run_cmd(base_app, 'help history >> {}'.format(filename))
+        expected = normalize(BASE_HELP + '\n' + HELP_HISTORY)
+        with open(filename) as f:
+            content = normalize(f.read())
+        assert content == expected
+    except:
+        raise
+    finally:
+        os.remove(filename)
+
+
+def test_feedback_to_output_true(base_app):
+    base_app.feedback_to_output = True
+    base_app.timing = True
+    f, filename = tempfile.mkstemp(prefix='cmd2_test', suffix='.txt')
+    os.close(f)
+    
+    try:
+        run_cmd(base_app, 'help > {}'.format(filename))
+        with open(filename) as f:
+            content = f.readlines()
+        assert content[-1].startswith('Elapsed: ')
+    except:
+        raise
+    finally:
+        os.remove(filename)
+
+
+def test_feedback_to_output_false(base_app, capsys):
+    base_app.feedback_to_output = False
+    base_app.timing = True
+    f, filename = tempfile.mkstemp(prefix='feedback_to_output', suffix='.txt')
+    os.close(f)
+    
+    try:
+        run_cmd(base_app, 'help > {}'.format(filename))
+        out, err = capsys.readouterr()
+        with open(filename) as f:
+            content = f.readlines()
+        assert not content[-1].startswith('Elapsed: ')
+        assert err.startswith('Elapsed')
+    except:
+        raise
+    finally:
+        os.remove(filename)
 
 
 def test_allow_redirection(base_app):
@@ -619,9 +658,9 @@ def test_base_timing(base_app, capsys):
     assert out == expected
     out, err = capsys.readouterr()
     if sys.platform == 'win32':
-        assert out.startswith('Elapsed: 0:00:00')
+        assert err.startswith('Elapsed: 0:00:00')
     else:
-        assert out.startswith('Elapsed: 0:00:00.0')
+        assert err.startswith('Elapsed: 0:00:00.0')
 
 
 def test_base_debug(base_app, capsys):