feat: expose websocket connection constructor

transport/websocket.go

@@ -120,12 +120,14 @@ func (wst *WebsocketTransport) HandleConnection(
 	return &WebsocketConnection{socket, wst}, nil
 }
 
-/**
+/*
+*
 Websocket connection do not require any additional processing
 */
 func (wst *WebsocketTransport) Serve(w http.ResponseWriter, r *http.Request) {}
 
-/**
+/*
+*
 Returns websocket connection with default params
 */
 func GetDefaultWebsocketTransport() *WebsocketTransport {
@@ -137,3 +139,7 @@ func GetDefaultWebsocketTransport() *WebsocketTransport {
 		BufferSize:     WsDefaultBufferSize,
 	}
 }
+
+func NewWebsocketConnection(socket *websocket.Conn, transport *WebsocketTransport) *WebsocketConnection {
+	return &WebsocketConnection{socket, transport}
+}