From 7db1033705f6da9cee6ed55065a66c911716bb6f Mon Sep 17 00:00:00 2001 From: Naveen Michaud-Agrawal Date: Mon, 21 Sep 2020 09:11:38 -0400 Subject: [PATCH] Add test that reproduces error without fix --- tests/test_connection.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/test_connection.py b/tests/test_connection.py index 2a6da83..1ea4fc2 100644 --- a/tests/test_connection.py +++ b/tests/test_connection.py @@ -732,6 +732,18 @@ async def handler(request): await client_ws.send_message('Hello from client!') +async def test_server_sends_after_close(nursery): + async def handler(request): + server_ws = await request.accept() + with pytest.raises(ConnectionClosed): + await server_ws.send_message("Hello from server") + server = await nursery.start(serve_websocket, handler, HOST, 0, None) + stream = await trio.open_tcp_stream(HOST, server.port) + client_ws = await wrap_client_stream(nursery, stream, HOST, RESOURCE) + async with client_ws: + await stream.aclose() + + async def test_server_does_not_close_handshake(nursery): async def handler(stream): request = await wrap_server_stream(nursery, stream)