Rewrite websocket test