Finished up websocket tests with mocked websocket.