From daeefd2e049b74340307481112a39f77de0f4769 Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Sun, 22 Oct 2017 11:40:31 +0200 Subject: bpo-28326: Fix multiprocessing.Process when stdout and/or stderr is closed or None. (#4073) --- Lib/test/_test_multiprocessing.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'Lib/test/_test_multiprocessing.py') diff --git a/Lib/test/_test_multiprocessing.py b/Lib/test/_test_multiprocessing.py index 8e004e21a4..69c0bd892c 100644 --- a/Lib/test/_test_multiprocessing.py +++ b/Lib/test/_test_multiprocessing.py @@ -582,6 +582,27 @@ class _TestProcess(BaseTestCase): proc.join() self.assertTrue(evt.is_set()) + @classmethod + def _test_error_on_stdio_flush(self, evt): + evt.set() + + def test_error_on_stdio_flush(self): + streams = [io.StringIO(), None] + streams[0].close() + for stream_name in ('stdout', 'stderr'): + for stream in streams: + old_stream = getattr(sys, stream_name) + setattr(sys, stream_name, stream) + try: + evt = self.Event() + proc = self.Process(target=self._test_error_on_stdio_flush, + args=(evt,)) + proc.start() + proc.join() + self.assertTrue(evt.is_set()) + finally: + setattr(sys, stream_name, old_stream) + # # -- cgit v1.2.1