root/gost_software
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix timeout when testing

Browse Source
This commit is contained in:
ginuerzh 2018-12-23 11:23:24 +08:00
parent 8d16b2d0b5
commit 89584a3a33
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
common_test.go
View File

@ -14,9 +14,9 @@ import (
func init() {
// SetLogger(&LogLogger{})
// Debug = true
DialTimeout = 500 * time.Millisecond
HandshakeTimeout = 500 * time.Millisecond
ConnectTimeout = 500 * time.Millisecond
DialTimeout = 1000 * time.Millisecond
HandshakeTimeout = 1000 * time.Millisecond
ConnectTimeout = 1000 * time.Millisecond
cert, err := GenCertificate()
if err != nil {

7
http_test.go
View File

@ -12,6 +12,7 @@ import (
"net/http/httptest"
"net/url"
"testing"
"time"
)
// proxyConn obtains a connection to the proxy server.
@ -76,14 +77,14 @@ func proxyRoundtrip(client *Client, server *Server, targetURL string, data []byt
return
}
// conn.SetDeadline(time.Now().Add(500 * time.Millisecond))
// defer conn.SetDeadline(time.Time{})
conn, err = client.Connect(conn, u.Host)
if err != nil {
return
}
conn.SetDeadline(time.Now().Add(500 * time.Millisecond))
defer conn.SetDeadline(time.Time{})
return httpRoundtrip(conn, targetURL, data)
}