Method on_connect

tests/test_endpoints.py:69–71

(self, websocket: WebSocket)

def test_websocket_endpoint_on_connect(test_client_factory: TestClientFactory) -> None:
    class WebSocketApp(WebSocketEndpoint):
        async def on_connect(self, websocket: WebSocket) -> None:
            assert websocket[class="st">"subprotocols"] == [class="st">"soap", class="st">"wamp"]
            await websocket.accept(subprotocol=class="st">"wamp")

    client = test_client_factory(WebSocketApp)
    with client.websocket_connect(class="st">"/ws", subprotocols=[class="st">"soap", class="st">"wamp"]) as websocket: