Fix test websocket helper client to use the new websocket_client hex.

The new client makes a callback if the websocket endpoint returns an error. This MR adds an Agent to handle such conditions.