Add missing tests for the websocket service to ensure proper coverage