Fix error result from Transport.Connect errors

websocket_transport.go

@@ -23,11 +23,12 @@ func (t *WebsocketTransport) Connect() error {
 	defer cancel()
 
 	wsConn, _, err := websocket.Dial(ctx, t.Config.Address, nil)
-	if err == nil {
-		t.wsConn = wsConn
-		t.netConn = websocket.NetConn(t.ctx, t.wsConn, websocket.MessageText)
+	if err != nil {
+		return NewConnError(err, true)
 	}
-	return err
+	t.wsConn = wsConn
+	t.netConn = websocket.NetConn(t.ctx, t.wsConn, websocket.MessageText)
+	return nil
 }
 
 func (t WebsocketTransport) StartTLS(domain string) error {

xmpp_transport.go

@@ -19,7 +19,10 @@ func (t *XMPPTransport) Connect() error {
 	var err error
 
 	t.conn, err = net.DialTimeout("tcp", t.Config.Address, time.Duration(t.Config.ConnectTimeout)*time.Second)
-	return err
+	if err != nil {
+		return NewConnError(err, true)
+	}
+	return nil
 }
 
 func (t XMPPTransport) DoesStartTLS() bool {