Fix websocket test helper to use the new websocket_client.

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