robot.api.logger#console flush output

mkorpela · mkorpela · commit 9a18c29d3313 · 2013-01-17T12:39:29.000+02:00
diff --git a/src/robot/api/logger.py b/src/robot/api/logger.py
@@ -102,3 +102,4 @@ def console(msg, newline=True):
     if newline:
         msg += '\n'
     sys.__stdout__.write(msg)
+    sys.__stdout__.flush()
diff --git a/utest/api/test_logging_api.py b/utest/api/test_logging_api.py
@@ -0,0 +1,33 @@
+import unittest
+import sys
+from robot.api.logger import console
+
+class FakeStdout(object):
+
+    def __init__(self):
+        self.flushed = False
+
+    def write(self, text):
+        self.text = text
+
+    def flush(self):
+        self.flushed = True
+
+
+class TestLoggingApi(unittest.TestCase):
+
+    def setUp(self):
+        self._fake_stdout = FakeStdout()
+        sys.__stdout__, self._original = self._fake_stdout, sys.__stdout__
+
+    def tearDown(self):
+        sys.__stdout__ = self._original
+
+    def test_console_flushes(self):
+        console('foo', newline=False)
+        self.assertTrue(self._fake_stdout.flushed)
+        self.assertEqual(self._fake_stdout.text, 'foo')
+
+
+if __name__ == '__main__':
+    unittest.main()
