Quic requires an initial message to start the session so we need connect

tunnel/default.go

@@ -144,15 +144,8 @@ func (t *tun) listen(link transport.Socket, listener bool) {
 			return
 		}
 
-		// first check Micro-Tunnel
 		switch msg.Header["Micro-Tunnel"] {
-		case "connect":
+		case "connect", "close":
-			// assuming new connection
-			// TODO: do something with this
-			continue
-		case "close":
-			// assuming connection closed
-			// TODO: do something with this
 			continue
 		}
 
@@ -289,13 +282,11 @@ func (t *tun) connect() error {
 			continue
 		}
 
-		err = c.Send(&transport.Message{
+		if err := c.Send(&transport.Message{
 			Header: map[string]string{
 				"Micro-Tunnel": "connect",
 			},
-		})
+		}); err != nil {
-
-		if err != nil {
 			continue
 		}
 
@@ -399,6 +390,7 @@ func (t *tun) Dial(addr string) (Conn, error) {
 	if !ok {
 		return nil, errors.New("error dialing " + addr)
 	}
+
 	// set remote
 	c.remote = addr
 	// set local

tunnel/tunnel_test.go

@@ -3,6 +3,7 @@ package tunnel
 import (
 	"sync"
 	"testing"
+	"time"
 
 	"github.com/micro/go-micro/transport"
 )
@@ -93,6 +94,8 @@ func TestTwoTunnel(t *testing.T) {
 	}
 	defer tunB.Close()
 
+	time.Sleep(time.Millisecond * 50)
+
 	// start tunA
 	err = tunA.Connect()
 	if err != nil {
@@ -100,14 +103,19 @@ func TestTwoTunnel(t *testing.T) {
 	}
 	defer tunA.Close()
 
+	time.Sleep(time.Millisecond * 50)
+
 	var wg sync.WaitGroup
 
 	// start accepting connections
+	// on tunnel A
 	wg.Add(1)
-	go testAccept(t, tunB, &wg)
+	go testAccept(t, tunA, &wg)
 
-	// send a message
+	time.Sleep(time.Millisecond * 50)
-	testSend(t, tunA)
+
+	// dial and send via B
+	testSend(t, tunB)
 
 	// wait until done
 	wg.Wait()