116 lines
2.8 KiB
Go
116 lines
2.8 KiB
Go
|
package starnet
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"io"
|
|||
|
|
"net"
|
|||
|
|
"net/http"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestRequestTimeoutDoesNotMutateClientTimeout(t *testing.T) {
|
|||
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
time.Sleep(200 * time.Millisecond)
|
|||
|
|
w.WriteHeader(http.StatusOK)
|
|||
|
|
}))
|
|||
|
|
defer s.Close()
|
|||
|
|
|
|||
|
|
client := NewClientNoErr()
|
|||
|
|
baseTimeout := client.HTTPClient().Timeout
|
|||
|
|
|
|||
|
|
_, err := client.Get(s.URL, WithTimeout(80*time.Millisecond))
|
|||
|
|
if err == nil {
|
|||
|
|
t.Fatal("expected request timeout error")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if client.HTTPClient().Timeout != baseTimeout {
|
|||
|
|
t.Fatalf("client timeout mutated: got=%v want=%v", client.HTTPClient().Timeout, baseTimeout)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resp, err := client.Get(s.URL, WithTimeout(400*time.Millisecond))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("second request should succeed with larger timeout: %v", err)
|
|||
|
|
}
|
|||
|
|
defer resp.Close()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestRequestTimeoutNoLingeringConnDeadline(t *testing.T) {
|
|||
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
_, _ = io.Copy(io.Discard, r.Body)
|
|||
|
|
w.WriteHeader(http.StatusOK)
|
|||
|
|
_, _ = w.Write([]byte("ok"))
|
|||
|
|
}))
|
|||
|
|
defer s.Close()
|
|||
|
|
|
|||
|
|
client := NewClientNoErr()
|
|||
|
|
|
|||
|
|
resp1, err := client.Post(s.URL, WithBodyString("first"), WithTimeout(120*time.Millisecond))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("first request should succeed: %v", err)
|
|||
|
|
}
|
|||
|
|
_ = resp1.Close()
|
|||
|
|
|
|||
|
|
// 如果请求超时依赖连接级绝对 deadline，经过该等待后复用连接会出现误超时。
|
|||
|
|
time.Sleep(220 * time.Millisecond)
|
|||
|
|
|
|||
|
|
resp2, err := client.Post(s.URL, WithBodyString("second"), WithTimeout(1*time.Second))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("second request should not be affected by previous timeout window: %v", err)
|
|||
|
|
}
|
|||
|
|
_ = resp2.Close()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestConnReadDeadlineTimeoutAndRecover(t *testing.T) {
|
|||
|
|
ln, err := net.Listen("tcp", "127.0.0.1:0")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("listen error: %v", err)
|
|||
|
|
}
|
|||
|
|
defer ln.Close()
|
|||
|
|
|
|||
|
|
done := make(chan struct{})
|
|||
|
|
go func() {
|
|||
|
|
defer close(done)
|
|||
|
|
conn, err := ln.Accept()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
defer conn.Close()
|
|||
|
|
|
|||
|
|
time.Sleep(180 * time.Millisecond)
|
|||
|
|
_, _ = conn.Write([]byte("x"))
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
c, err := Dial("tcp", ln.Addr().String())
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("dial error: %v", err)
|
|||
|
|
}
|
|||
|
|
defer c.Close()
|
|||
|
|
|
|||
|
|
if err := c.SetReadDeadline(time.Now().Add(60 * time.Millisecond)); err != nil {
|
|||
|
|
t.Fatalf("set read deadline error: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
buf := make([]byte, 1)
|
|||
|
|
_, err = c.Read(buf)
|
|||
|
|
if err == nil {
|
|||
|
|
t.Fatal("expected read timeout error")
|
|||
|
|
}
|
|||
|
|
if ne, ok := err.(net.Error); !ok || !ne.Timeout() {
|
|||
|
|
t.Fatalf("expected net timeout error, got: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := c.SetReadDeadline(time.Time{}); err != nil {
|
|||
|
|
t.Fatalf("clear read deadline error: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if _, err := io.ReadFull(c, buf); err != nil {
|
|||
|
|
t.Fatalf("read after clearing deadline should succeed: %v", err)
|
|||
|
|
}
|
|||
|
|
if string(buf) != "x" {
|
|||
|
|
t.Fatalf("unexpected payload: %q", string(buf))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
<-done
|
|||
|
|
}
|