update websocket test code