Added unit test

1 files changed, 20 insertions, 11 deletions

diff --git a/tests/test_utils.py b/tests/test_utils.py
index 4574e579..c0b16990 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -126,23 +126,12 @@ def stdout_sim():
     stdsim = cu.StdSim(sys.stdout, echo=True)
     return stdsim
 
-@pytest.fixture
-def stringio_sim():
-    import io
-    stdsim = cu.StdSim(io.StringIO(), echo=True)
-    return stdsim
-
 
 def test_stdsim_write_str(stdout_sim):
     my_str = 'Hello World'
     stdout_sim.write(my_str)
     assert stdout_sim.getvalue() == my_str
 
-def test_stdsim_write_str_inner_no_buffer(stringio_sim):
-    my_str = 'Hello World'
-    stringio_sim.write(my_str)
-    assert stringio_sim.getvalue() == my_str
-
 def test_stdsim_write_bytes(stdout_sim):
     b_str = b'Hello World'
     with pytest.raises(TypeError):
@@ -218,6 +207,26 @@ def test_stdsim_pause_storage(stdout_sim):
     stdout_sim.buffer.write(b_str)
     assert stdout_sim.getbytes() == b''
 
+def test_stdsim_line_buffering(base_app):
+    # This exercises the case of writing binary data that contains new lines/carriage returns to a StdSim
+    # when line buffering is on. The output should immediately be flushed to the underlying stream.
+    import os
+    import tempfile
+    file = tempfile.NamedTemporaryFile(mode='wt')
+    file.line_buffering = True
+
+    stdsim = cu.StdSim(file, echo=True)
+    saved_size = os.path.getsize(file.name)
+
+    bytes_to_write = b'hello\n'
+    stdsim.buffer.write(bytes_to_write)
+    assert os.path.getsize(file.name) == saved_size + len(bytes_to_write)
+    saved_size = os.path.getsize(file.name)
+
+    bytes_to_write = b'hello\r'
+    stdsim.buffer.write(bytes_to_write)
+    assert os.path.getsize(file.name) == saved_size + len(bytes_to_write)
+
 
 @pytest.fixture
 def pr_none():