Add websocket client for testing purposes
Add a couple of tests