[lldb][test] TestFileHandle: flush the output after write

llvm · Mar 5, 2020 · 7a11cc0 · 7a11cc0
1 parent cb06571
commit 7a11cc0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lldb/test/API/python_api/file_handle/TestFileHandle.py b/lldb/test/API/python_api/file_handle/TestFileHandle.py
@@ -129,6 +129,7 @@ def test_legacy_file_out_script(self):
             # even with collect_result=True.
             self.handleCmd('script 1+1')
             self.dbg.GetOutputFileHandle().write('FOO\n')
+            self.dbg.GetOutputFileHandle().flush()
         with open(self.out_filename, 'r') as f:
             self.assertEqual(readStrippedLines(f), ['2', 'FOO'])
 
@@ -246,6 +247,7 @@ def test_fileno_out(self):
             self.assertTrue(status.Success())
             self.handleCmd('script 1+2')
             self.dbg.GetOutputFile().Write(b'quux')
+            self.dbg.GetOutputFile().Flush()
 
         with open(self.out_filename, 'r') as f:
             self.assertEqual(readStrippedLines(f), ['3', 'quux'])