78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
// Copyright 2009 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.
|
|
|
|
// Tests for client.go
|
|
|
|
package http_test
|
|
|
|
import (
|
|
"fmt"
|
|
. "http"
|
|
"http/httptest"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
var robotsTxtHandler = HandlerFunc(func(w ResponseWriter, r *Request) {
|
|
w.Header().Set("Last-Modified", "sometime")
|
|
fmt.Fprintf(w, "User-agent: go\nDisallow: /something/")
|
|
})
|
|
|
|
func TestClient(t *testing.T) {
|
|
ts := httptest.NewServer(robotsTxtHandler)
|
|
defer ts.Close()
|
|
|
|
r, _, err := Get(ts.URL)
|
|
var b []byte
|
|
if err == nil {
|
|
b, err = ioutil.ReadAll(r.Body)
|
|
r.Body.Close()
|
|
}
|
|
if err != nil {
|
|
t.Error(err)
|
|
} else if s := string(b); !strings.HasPrefix(s, "User-agent:") {
|
|
t.Errorf("Incorrect page body (did not begin with User-agent): %q", s)
|
|
}
|
|
}
|
|
|
|
func TestClientHead(t *testing.T) {
|
|
ts := httptest.NewServer(robotsTxtHandler)
|
|
defer ts.Close()
|
|
|
|
r, err := Head(ts.URL)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := r.Header["Last-Modified"]; !ok {
|
|
t.Error("Last-Modified header not found.")
|
|
}
|
|
}
|
|
|
|
type recordingTransport struct {
|
|
req *Request
|
|
}
|
|
|
|
func (t *recordingTransport) RoundTrip(req *Request) (resp *Response, err os.Error) {
|
|
t.req = req
|
|
return nil, os.NewError("dummy impl")
|
|
}
|
|
|
|
func TestGetRequestFormat(t *testing.T) {
|
|
tr := &recordingTransport{}
|
|
client := &Client{Transport: tr}
|
|
url := "http://dummy.faketld/"
|
|
client.Get(url) // Note: doesn't hit network
|
|
if tr.req.Method != "GET" {
|
|
t.Errorf("expected method %q; got %q", "GET", tr.req.Method)
|
|
}
|
|
if tr.req.URL.String() != url {
|
|
t.Errorf("expected URL %q; got %q", url, tr.req.URL.String())
|
|
}
|
|
if tr.req.Header == nil {
|
|
t.Errorf("expected non-nil request Header")
|
|
}
|
|
}
|