Update WebSocketTransport to passing integration tests