Fix connection reference in testing

[stam]

Because we create the connection instance inside the websocket listener, we have no clean way to access it in the tests.

We currently set the websocket.connection, but this is only possible if the websocket is mocked. Therefore the Connection.init contains test only code which is nasty.