mirror of https://github.com/python/cpython
asyncio: Fix test_stdin_broken_pipe(), drain() can also raise
ConnectionResetError
This commit is contained in:
parent
d55b54d5c0
commit
ddc8c8db1c
|
@ -130,9 +130,9 @@ class SubprocessMixin:
|
||||||
def test_stdin_broken_pipe(self):
|
def test_stdin_broken_pipe(self):
|
||||||
proc, large_data = self.prepare_broken_pipe_test()
|
proc, large_data = self.prepare_broken_pipe_test()
|
||||||
|
|
||||||
# drain() must raise BrokenPipeError
|
# drain() must raise BrokenPipeError or ConnectionResetError
|
||||||
proc.stdin.write(large_data)
|
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.stdin.drain())
|
||||||
self.loop.run_until_complete(proc.wait())
|
self.loop.run_until_complete(proc.wait())
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue