2010-12-01 21:00:19 -07:00
|
|
|
// Copyright 2010 The Go Authors. All rights reserved.
|
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
// End-to-end serving tests
|
|
|
|
|
|
|
|
package http
|
|
|
|
|
|
|
|
import (
|
2011-01-13 15:10:02 -07:00
|
|
|
"bufio"
|
2010-12-01 21:00:19 -07:00
|
|
|
"bytes"
|
2011-02-10 15:36:22 -07:00
|
|
|
"fmt"
|
2011-01-13 15:10:02 -07:00
|
|
|
"io"
|
2011-02-10 15:36:22 -07:00
|
|
|
"io/ioutil"
|
2010-12-01 21:00:19 -07:00
|
|
|
"os"
|
|
|
|
"net"
|
|
|
|
"testing"
|
2011-02-10 15:36:22 -07:00
|
|
|
"time"
|
2010-12-01 21:00:19 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
type dummyAddr string
|
|
|
|
type oneConnListener struct {
|
|
|
|
conn net.Conn
|
|
|
|
}
|
|
|
|
|
|
|
|
func (l *oneConnListener) Accept() (c net.Conn, err os.Error) {
|
|
|
|
c = l.conn
|
|
|
|
if c == nil {
|
|
|
|
err = os.EOF
|
|
|
|
return
|
|
|
|
}
|
|
|
|
err = nil
|
|
|
|
l.conn = nil
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
func (l *oneConnListener) Close() os.Error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (l *oneConnListener) Addr() net.Addr {
|
|
|
|
return dummyAddr("test-address")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (a dummyAddr) Network() string {
|
|
|
|
return string(a)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (a dummyAddr) String() string {
|
|
|
|
return string(a)
|
|
|
|
}
|
|
|
|
|
|
|
|
type testConn struct {
|
|
|
|
readBuf bytes.Buffer
|
|
|
|
writeBuf bytes.Buffer
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) Read(b []byte) (int, os.Error) {
|
|
|
|
return c.readBuf.Read(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) Write(b []byte) (int, os.Error) {
|
|
|
|
return c.writeBuf.Write(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) Close() os.Error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) LocalAddr() net.Addr {
|
|
|
|
return dummyAddr("local-addr")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) RemoteAddr() net.Addr {
|
|
|
|
return dummyAddr("remote-addr")
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) SetTimeout(nsec int64) os.Error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) SetReadTimeout(nsec int64) os.Error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *testConn) SetWriteTimeout(nsec int64) os.Error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestConsumingBodyOnNextConn(t *testing.T) {
|
|
|
|
conn := new(testConn)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
|
|
conn.readBuf.Write([]byte(
|
|
|
|
"POST / HTTP/1.1\r\n" +
|
|
|
|
"Host: test\r\n" +
|
|
|
|
"Content-Length: 11\r\n" +
|
|
|
|
"\r\n" +
|
|
|
|
"foo=1&bar=1"))
|
|
|
|
}
|
|
|
|
|
|
|
|
reqNum := 0
|
|
|
|
ch := make(chan *Request)
|
|
|
|
servech := make(chan os.Error)
|
|
|
|
listener := &oneConnListener{conn}
|
|
|
|
handler := func(res ResponseWriter, req *Request) {
|
|
|
|
reqNum++
|
|
|
|
t.Logf("Got request #%d: %v", reqNum, req)
|
|
|
|
ch <- req
|
|
|
|
}
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
servech <- Serve(listener, HandlerFunc(handler))
|
|
|
|
}()
|
|
|
|
|
|
|
|
var req *Request
|
|
|
|
t.Log("Waiting for first request.")
|
|
|
|
req = <-ch
|
|
|
|
if req == nil {
|
|
|
|
t.Fatal("Got nil first request.")
|
|
|
|
}
|
|
|
|
if req.Method != "POST" {
|
|
|
|
t.Errorf("For request #1's method, got %q; expected %q",
|
|
|
|
req.Method, "POST")
|
|
|
|
}
|
|
|
|
|
|
|
|
t.Log("Waiting for second request.")
|
|
|
|
req = <-ch
|
|
|
|
if req == nil {
|
|
|
|
t.Fatal("Got nil first request.")
|
|
|
|
}
|
|
|
|
if req.Method != "POST" {
|
|
|
|
t.Errorf("For request #2's method, got %q; expected %q",
|
|
|
|
req.Method, "POST")
|
|
|
|
}
|
|
|
|
|
|
|
|
t.Log("Waiting for EOF.")
|
|
|
|
if serveerr := <-servech; serveerr != os.EOF {
|
|
|
|
t.Errorf("Serve returned %q; expected EOF", serveerr)
|
|
|
|
}
|
|
|
|
}
|
2011-01-13 15:10:02 -07:00
|
|
|
|
2011-02-01 11:58:59 -07:00
|
|
|
type stringHandler string
|
|
|
|
|
|
|
|
func (s stringHandler) ServeHTTP(w ResponseWriter, r *Request) {
|
|
|
|
w.SetHeader("Result", string(s))
|
|
|
|
}
|
|
|
|
|
|
|
|
var handlers = []struct {
|
|
|
|
pattern string
|
|
|
|
msg string
|
|
|
|
}{
|
|
|
|
{"/", "Default"},
|
|
|
|
{"/someDir/", "someDir"},
|
|
|
|
{"someHost.com/someDir/", "someHost.com/someDir"},
|
|
|
|
}
|
|
|
|
|
|
|
|
var vtests = []struct {
|
|
|
|
url string
|
|
|
|
expected string
|
|
|
|
}{
|
|
|
|
{"http://localhost/someDir/apage", "someDir"},
|
|
|
|
{"http://localhost/otherDir/apage", "Default"},
|
|
|
|
{"http://someHost.com/someDir/apage", "someHost.com/someDir"},
|
|
|
|
{"http://otherHost.com/someDir/apage", "someDir"},
|
|
|
|
{"http://otherHost.com/aDir/apage", "Default"},
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestHostHandlers(t *testing.T) {
|
|
|
|
for _, h := range handlers {
|
|
|
|
Handle(h.pattern, stringHandler(h.msg))
|
|
|
|
}
|
|
|
|
l, err := net.Listen("tcp", "127.0.0.1:0") // any port
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
defer l.Close()
|
|
|
|
go Serve(l, nil)
|
|
|
|
conn, err := net.Dial("tcp", "", l.Addr().String())
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
defer conn.Close()
|
|
|
|
cc := NewClientConn(conn, nil)
|
|
|
|
for _, vt := range vtests {
|
|
|
|
var r *Response
|
|
|
|
var req Request
|
|
|
|
if req.URL, err = ParseURL(vt.url); err != nil {
|
|
|
|
t.Errorf("cannot parse url: %v", err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
if err := cc.Write(&req); err != nil {
|
|
|
|
t.Errorf("writing request: %v", err)
|
|
|
|
continue
|
|
|
|
}
|
2011-02-11 13:05:47 -07:00
|
|
|
r, err := cc.Read(&req)
|
2011-02-01 11:58:59 -07:00
|
|
|
if err != nil {
|
|
|
|
t.Errorf("reading response: %v", err)
|
|
|
|
continue
|
|
|
|
}
|
2011-02-22 22:39:25 -07:00
|
|
|
s := r.Header.Get("Result")
|
2011-02-01 11:58:59 -07:00
|
|
|
if s != vt.expected {
|
|
|
|
t.Errorf("Get(%q) = %q, want %q", vt.url, s, vt.expected)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-01-13 15:10:02 -07:00
|
|
|
type responseWriterMethodCall struct {
|
|
|
|
method string
|
|
|
|
headerKey, headerValue string // if method == "SetHeader"
|
|
|
|
bytesWritten []byte // if method == "Write"
|
|
|
|
responseCode int // if method == "WriteHeader"
|
|
|
|
}
|
|
|
|
|
|
|
|
type recordingResponseWriter struct {
|
|
|
|
log []*responseWriterMethodCall
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) RemoteAddr() string {
|
|
|
|
return "1.2.3.4"
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) UsingTLS() bool {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) SetHeader(k, v string) {
|
|
|
|
rw.log = append(rw.log, &responseWriterMethodCall{method: "SetHeader", headerKey: k, headerValue: v})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) Write(buf []byte) (int, os.Error) {
|
|
|
|
rw.log = append(rw.log, &responseWriterMethodCall{method: "Write", bytesWritten: buf})
|
|
|
|
return len(buf), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) WriteHeader(code int) {
|
|
|
|
rw.log = append(rw.log, &responseWriterMethodCall{method: "WriteHeader", responseCode: code})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) Flush() {
|
|
|
|
rw.log = append(rw.log, &responseWriterMethodCall{method: "Flush"})
|
|
|
|
}
|
|
|
|
|
|
|
|
func (rw *recordingResponseWriter) Hijack() (io.ReadWriteCloser, *bufio.ReadWriter, os.Error) {
|
|
|
|
panic("Not supported")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Tests for http://code.google.com/p/go/issues/detail?id=900
|
|
|
|
func TestMuxRedirectLeadingSlashes(t *testing.T) {
|
|
|
|
paths := []string{"//foo.txt", "///foo.txt", "/../../foo.txt"}
|
|
|
|
for _, path := range paths {
|
|
|
|
req, err := ReadRequest(bufio.NewReader(bytes.NewBufferString("GET " + path + " HTTP/1.1\r\nHost: test\r\n\r\n")))
|
|
|
|
if err != nil {
|
|
|
|
t.Errorf("%s", err)
|
|
|
|
}
|
|
|
|
mux := NewServeMux()
|
|
|
|
resp := new(recordingResponseWriter)
|
|
|
|
resp.log = make([]*responseWriterMethodCall, 0)
|
|
|
|
|
|
|
|
mux.ServeHTTP(resp, req)
|
|
|
|
|
|
|
|
dumpLog := func() {
|
|
|
|
t.Logf("For path %q:", path)
|
|
|
|
for _, call := range resp.log {
|
|
|
|
t.Logf("Got call: %s, header=%s, value=%s, buf=%q, code=%d", call.method,
|
|
|
|
call.headerKey, call.headerValue, call.bytesWritten, call.responseCode)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(resp.log) != 2 {
|
|
|
|
dumpLog()
|
|
|
|
t.Errorf("expected 2 calls to response writer; got %d", len(resp.log))
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if resp.log[0].method != "SetHeader" ||
|
|
|
|
resp.log[0].headerKey != "Location" || resp.log[0].headerValue != "/foo.txt" {
|
|
|
|
dumpLog()
|
|
|
|
t.Errorf("Expected SetHeader of Location to /foo.txt")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if resp.log[1].method != "WriteHeader" || resp.log[1].responseCode != StatusMovedPermanently {
|
|
|
|
dumpLog()
|
|
|
|
t.Errorf("Expected WriteHeader of StatusMovedPermanently")
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2011-02-10 15:36:22 -07:00
|
|
|
|
|
|
|
func TestServerTimeouts(t *testing.T) {
|
|
|
|
l, err := net.ListenTCP("tcp", &net.TCPAddr{Port: 0})
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("listen error: %v", err)
|
|
|
|
}
|
|
|
|
addr, _ := l.Addr().(*net.TCPAddr)
|
|
|
|
|
|
|
|
reqNum := 0
|
|
|
|
handler := HandlerFunc(func(res ResponseWriter, req *Request) {
|
|
|
|
reqNum++
|
|
|
|
fmt.Fprintf(res, "req=%d", reqNum)
|
|
|
|
})
|
|
|
|
|
|
|
|
const second = 1000000000 /* nanos */
|
|
|
|
server := &Server{Handler: handler, ReadTimeout: 0.25 * second, WriteTimeout: 0.25 * second}
|
|
|
|
go server.Serve(l)
|
|
|
|
|
|
|
|
url := fmt.Sprintf("http://localhost:%d/", addr.Port)
|
|
|
|
|
|
|
|
// Hit the HTTP server successfully.
|
|
|
|
r, _, err := Get(url)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("http Get #1: %v", err)
|
|
|
|
}
|
|
|
|
got, _ := ioutil.ReadAll(r.Body)
|
|
|
|
expected := "req=1"
|
|
|
|
if string(got) != expected {
|
|
|
|
t.Errorf("Unexpected response for request #1; got %q; expected %q",
|
|
|
|
string(got), expected)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Slow client that should timeout.
|
|
|
|
t1 := time.Nanoseconds()
|
|
|
|
conn, err := net.Dial("tcp", "", fmt.Sprintf("localhost:%d", addr.Port))
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("Dial: %v", err)
|
|
|
|
}
|
|
|
|
buf := make([]byte, 1)
|
|
|
|
n, err := conn.Read(buf)
|
|
|
|
latency := time.Nanoseconds() - t1
|
|
|
|
if n != 0 || err != os.EOF {
|
|
|
|
t.Errorf("Read = %v, %v, wanted %v, %v", n, err, 0, os.EOF)
|
|
|
|
}
|
|
|
|
if latency < second*0.20 /* fudge from 0.25 above */ {
|
|
|
|
t.Errorf("got EOF after %d ns, want >= %d", latency, second*0.20)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Hit the HTTP server successfully again, verifying that the
|
|
|
|
// previous slow connection didn't run our handler. (that we
|
|
|
|
// get "req=2", not "req=3")
|
|
|
|
r, _, err = Get(url)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("http Get #2: %v", err)
|
|
|
|
}
|
|
|
|
got, _ = ioutil.ReadAll(r.Body)
|
|
|
|
expected = "req=2"
|
|
|
|
if string(got) != expected {
|
|
|
|
t.Errorf("Get #2 got %q, want %q", string(got), expected)
|
|
|
|
}
|
|
|
|
|
|
|
|
l.Close()
|
|
|
|
}
|