fix tests

Author: ginuerzh

http2.go

@@ -135,7 +135,6 @@ func (tr *http2Transporter) Dial(addr string, options ...DialOption) (net.Conn,
 		// So we should try to connect the server.
 		conn, err := opts.Chain.Dial(addr)
 		if err != nil {
-
 			return nil, err
 		}
 		conn.Close()

http2_test.go

@@ -508,6 +508,7 @@ func h2ForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()
@@ -905,6 +906,7 @@ func h2cForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()
@@ -1046,16 +1048,15 @@ func TestHTTP2ProxyWithHostProbeResist(t *testing.T) {
 
 	req := &http.Request{
 		Method:        http.MethodConnect,
-		URL:           &url.URL{Scheme: "https", Host: ln.Addr().String()},
+		URL:           &url.URL{Scheme: "https", Host: cc.addr},
 		Header:        make(http.Header),
 		Proto:         "HTTP/2.0",
 		ProtoMajor:    2,
 		ProtoMinor:    0,
 		Body:          ioutil.NopCloser(bytes.NewReader(sendData)),
-		Host:          cc.addr,
-		ContentLength: -1,
+		Host:          "github.com:443",
+		ContentLength: int64(len(sendData)),
 	}
-	req.Header.Set("Gost-Target", "github.com:443")
 
 	resp, err := cc.client.Do(req)
 	if err != nil {

kcp_test.go

@@ -386,6 +386,7 @@ func kcpForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()

quic_test.go

@@ -387,6 +387,7 @@ func quicForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()

ssh_test.go

@@ -559,6 +559,7 @@ func sshForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()

tls_test.go

@@ -388,6 +388,7 @@ func tlsForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()
@@ -787,6 +788,7 @@ func mtlsForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()

ws_test.go

@@ -386,6 +386,7 @@ func wsForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()
@@ -785,6 +786,7 @@ func mwsForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()

wss_test.go

@@ -387,6 +387,7 @@ func wssForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()
@@ -786,6 +787,7 @@ func mwssForwardTunnelRoundtrip(targetURL string, data []byte) error {
 		Listener: ln,
 		Handler:  TCPDirectForwardHandler(u.Host),
 	}
+	server.Handler.Init()
 
 	go server.Run()
 	defer server.Close()