Function test_websocket_endpoint_on_disconnect

tests/test_endpoints.py:168–178

(
    test_client_factory: TestClientFactory,
)

def test_websocket_endpoint_on_disconnect(
    test_client_factory: TestClientFactory,
) -> None:
    class WebSocketApp(WebSocketEndpoint):
        async def on_disconnect(self, websocket: WebSocket, close_code: int) -> None:
            assert close_code == 1001
            await websocket.close(code=close_code)

    client = test_client_factory(WebSocketApp)
    with client.websocket_connect(class="st">"/ws") as websocket:
        websocket.close(code=1001)