diff --git a/Lib/test/test_io.py b/Lib/test/test_io.py index 65fa572d5a85ab..1ca3edac8c8dc9 100644 --- a/Lib/test/test_io.py +++ b/Lib/test/test_io.py @@ -4020,17 +4020,23 @@ def test_issue119506(self): chunk_size = 8192 class MockIO(self.MockRawIO): + written = False def write(self, data): - t.write("efg") + if not self.written: + self.written = True + t.write("middle") return super().write(data) buf = MockIO() t = self.TextIOWrapper(buf) - t.write("a" * (chunk_size - 1)) - t.write("bcd") + t.write("abc") + t.write("def") + # writing data which size >= chunk_size cause flushing buffer before write. + t.write("g" * chunk_size) t.flush() - self.assertEqual([b"a" * (chunk_size - 1), b"efgbcd"], buf._write_stack) + self.assertEqual([b"abcdef", b"middle", b"g"*chunk_size], + buf._write_stack) class PyTextIOWrapperTest(TextIOWrapperTest):