Made it possible to mock the websocket client side