asyncio: Fix test_stdin_broken_pipe(), drain() can also raise

ConnectionResetError
2014-07-17 14:01:14 +02:00 · 2014-07-17 14:01:14 +02:00 · ddc8c8db1c
parent d55b54d5c0
commit ddc8c8db1c
1 changed files with 2 additions and 2 deletions
--- a/Lib/test/test_asyncio/test_subprocess.py
+++ b/Lib/test/test_asyncio/test_subprocess.py
@ -130,9 +130,9 @@ class SubprocessMixin:
    def test_stdin_broken_pipe(self):
        proc, large_data = self.prepare_broken_pipe_test()

-        # drain() must raise BrokenPipeError
+        # drain() must raise BrokenPipeError or ConnectionResetError
        proc.stdin.write(large_data)
-        self.assertRaises(BrokenPipeError,
+        self.assertRaises((BrokenPipeError, ConnectionResetError),
                          self.loop.run_until_complete, proc.stdin.drain())
        self.loop.run_until_complete(proc.wait())