Add tests for the websocket service