Improve tests.
Remove MockWebSocket.