mux test

lib/mux/mux_test.go

@@ -26,22 +26,25 @@ func TestNewMux(t *testing.T) {
 	time.Sleep(time.Second * 3)
 	go func() {
 		m2 := NewMux(conn2, "tcp")
+		connCh := make(chan bool, 1)
 		for {
 			c, err := m2.Accept()
 			if err != nil {
 				log.Fatalln(err)
 			}
-			go func(c net.Conn) {
+			connCh <- true
+			go func(c net.Conn, ch chan bool) {
 				c2, err := net.Dial("tcp", "127.0.0.1:80")
 				if err != nil {
 					log.Fatalln(err)
 				}
 				go common.CopyBuffer(c2, c)
 				common.CopyBuffer(c, c2)
-				c.Close()
-				//logs.Warn("close from out npc ")
 				c2.Close()
-			}(c)
+				c.Close()
+				logs.Warn("close npc")
+				<-ch
+			}(c, connCh)
 		}
 	}()
 
@@ -51,12 +54,14 @@ func TestNewMux(t *testing.T) {
 		if err != nil {
 			log.Fatalln(err)
 		}
+		connCh := make(chan bool, 1)
 		for {
 			conn, err := l.Accept()
 			if err != nil {
 				log.Fatalln(err)
 			}
-			go func(conn net.Conn) {
+			connCh <- true
+			go func(conn net.Conn, ch chan bool) {
 				tmpCpnn, err := m1.NewConn()
 				if err != nil {
 					log.Fatalln(err)
@@ -64,9 +69,10 @@ func TestNewMux(t *testing.T) {
 				go common.CopyBuffer(tmpCpnn, conn)
 				common.CopyBuffer(conn, tmpCpnn)
 				conn.Close()
-				tmpCpnn.Close()
+				//tmpCpnn.Close()
 				logs.Warn("close from out nps ", tmpCpnn.connId)
-			}(conn)
+				<-ch
+			}(conn, connCh)
 		}
 	}()