diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2022-10-18 15:21:58 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-18 15:21:58 (GMT) |
commit | 0bba980c5f67e72e8fe4803957ac82748d3b483e (patch) | |
tree | eb4ba02d5f6cc73237e7d4597c2ca101ae29ee5e /Lib/test/test_asyncio | |
parent | d798b595ffed4560f1de068206cae8c2bf567085 (diff) | |
download | cpython-0bba980c5f67e72e8fe4803957ac82748d3b483e.zip cpython-0bba980c5f67e72e8fe4803957ac82748d3b483e.tar.gz cpython-0bba980c5f67e72e8fe4803957ac82748d3b483e.tar.bz2 |
[3.11] gh-98174: Handle EPROTOTYPE under macOS in test_sendfile_fallback_close_peer_in_the_middle_of_receiving (GH-98316) (#98357)
Co-authored-by: Nikita Sobolev <mail@sobolevn.me>
(cherry picked from commit 3e82ad05b18d004e4d01fdb643344d6a2bf11900)
Co-authored-by: fancidev <fancidev@gmail.com>
Diffstat (limited to 'Lib/test/test_asyncio')
-rw-r--r-- | Lib/test/test_asyncio/test_sendfile.py | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/Lib/test/test_asyncio/test_sendfile.py b/Lib/test/test_asyncio/test_sendfile.py index a10504b..0198da2 100644 --- a/Lib/test/test_asyncio/test_sendfile.py +++ b/Lib/test/test_asyncio/test_sendfile.py @@ -1,6 +1,7 @@ """Tests for sendfile functionality.""" import asyncio +import errno import os import socket import sys @@ -484,8 +485,17 @@ class SendfileMixin(SendfileBase): srv_proto, cli_proto = self.prepare_sendfile(close_after=1024) with self.assertRaises(ConnectionError): - self.run_loop( - self.loop.sendfile(cli_proto.transport, self.file)) + try: + self.run_loop( + self.loop.sendfile(cli_proto.transport, self.file)) + except OSError as e: + # macOS may raise OSError of EPROTOTYPE when writing to a + # socket that is in the process of closing down. + if e.errno == errno.EPROTOTYPE and sys.platform == "darwin": + raise ConnectionError + else: + raise + self.run_loop(srv_proto.done) self.assertTrue(1024 <= srv_proto.nbytes < len(self.DATA), |