Added websocket tests