Merge pull request #6257 from proppy/test-conntimeout

utils: add test for TimeoutConn
This commit is contained in:
Victor Vieux 2014-06-18 16:23:39 -07:00
Родитель 7a0e599142 39320f9393
Коммит 1ca7964cdd
1 изменённых файлов: 33 добавлений и 0 удалений

33
utils/timeoutconn_test.go Normal file
Просмотреть файл

@ -0,0 +1,33 @@
package utils
import (
"bufio"
"fmt"
"net"
"net/http"
"net/http/httptest"
"testing"
"time"
)
func TestTimeoutConnRead(t *testing.T) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, "hello")
}))
defer ts.Close()
conn, err := net.Dial("tcp", ts.URL[7:])
if err != nil {
t.Fatalf("failed to create connection to %q: %v", ts.URL, err)
}
tconn := NewTimeoutConn(conn, 1*time.Second)
if _, err = bufio.NewReader(tconn).ReadString('\n'); err == nil {
t.Fatalf("expected timeout error, got none")
}
if _, err := fmt.Fprintf(tconn, "GET / HTTP/1.0\r\n\r\n"); err != nil {
t.Errorf("unexpected error: %v", err)
}
if _, err = bufio.NewReader(tconn).ReadString('\n'); err != nil {
t.Errorf("unexpected error: %v", err)
}
}