Added test to websocket code