fix tests

This commit is contained in:
ginuerzh 2019-06-21 21:30:48 +08:00
parent e996e7c35b
commit 9ec2bb9b87
8 changed files with 14 additions and 5 deletions

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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