|
7 | 7 | import os |
8 | 8 | import signal |
9 | 9 | import socket |
| 10 | +import struct |
10 | 11 | import sys |
11 | 12 | import tempfile |
12 | 13 | import time |
@@ -222,6 +223,48 @@ def connection_lost(self, exc): |
222 | 223 | self.assertEqual(received, b"response-before-eof") |
223 | 224 | self.assertEqual(events, ["data", "eof", "lost"]) |
224 | 225 |
|
| 226 | + @unittest.skipUnless(sys.platform == "win32", "requires Winsock") |
| 227 | + def test_write_reports_reset_while_reading_is_paused(self) -> None: |
| 228 | + async def main() -> BaseException | None: |
| 229 | + loop = asyncio.get_running_loop() |
| 230 | + connection_lost = loop.create_future() |
| 231 | + |
| 232 | + class ClientProtocol(asyncio.Protocol): |
| 233 | + def connection_made(self, transport): |
| 234 | + transport.set_write_buffer_limits(0) |
| 235 | + transport.pause_reading() |
| 236 | + |
| 237 | + def connection_lost(self, exc): |
| 238 | + if not connection_lost.done(): |
| 239 | + connection_lost.set_result(exc) |
| 240 | + |
| 241 | + server_sock = socket.create_server(("127.0.0.1", 0)) |
| 242 | + try: |
| 243 | + transport, _ = await loop.create_connection( |
| 244 | + ClientProtocol, *server_sock.getsockname()[:2] |
| 245 | + ) |
| 246 | + peer_sock, _ = server_sock.accept() |
| 247 | + peer_sock.setsockopt( |
| 248 | + socket.SOL_SOCKET, |
| 249 | + socket.SO_LINGER, |
| 250 | + struct.pack("ii", 1, 0), |
| 251 | + ) |
| 252 | + peer_sock.close() |
| 253 | + try: |
| 254 | + for _ in range(1000): |
| 255 | + await asyncio.sleep(0) |
| 256 | + transport.write(b"foo") |
| 257 | + if connection_lost.done(): |
| 258 | + break |
| 259 | + |
| 260 | + return await asyncio.wait_for(connection_lost, 3.0) |
| 261 | + finally: |
| 262 | + transport.close() |
| 263 | + finally: |
| 264 | + server_sock.close() |
| 265 | + |
| 266 | + self.assertIsNotNone(rsloop.run(main())) |
| 267 | + |
225 | 268 | def test_readexactly_larger_than_flow_control_window(self) -> None: |
226 | 269 | # Regression test: a readexactly() waiting for more than 2 * limit |
227 | 270 | # bytes (default limit 64 KiB) used to deadlock because the reader |
|
0 commit comments