Added a WebSocket test