1 // Copyright 2011 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
5 // Reverse proxy tests.
17 func TestReverseProxy(t *testing.T) {
18 const backendResponse = "I am the backend"
19 const backendStatus = 404
20 backend := httptest.NewServer(HandlerFunc(func(w ResponseWriter, r *Request) {
21 if len(r.TransferEncoding) > 0 {
22 t.Errorf("backend got unexpected TransferEncoding: %v", r.TransferEncoding)
24 if r.Header.Get("X-Forwarded-For") == "" {
25 t.Errorf("didn't get X-Forwarded-For header")
27 if c := r.Header.Get("Connection"); c != "" {
28 t.Errorf("handler got Connection header value %q", c)
30 if g, e := r.Host, "some-name"; g != e {
31 t.Errorf("backend got Host header %q, want %q", g, e)
33 w.Header().Set("X-Foo", "bar")
34 SetCookie(w, &Cookie{Name: "flavor", Value: "chocolateChip"})
35 w.WriteHeader(backendStatus)
36 w.Write([]byte(backendResponse))
39 backendURL, err := url.Parse(backend.URL)
43 proxyHandler := NewSingleHostReverseProxy(backendURL)
44 frontend := httptest.NewServer(proxyHandler)
45 defer frontend.Close()
47 getReq, _ := NewRequest("GET", frontend.URL, nil)
48 getReq.Host = "some-name"
49 getReq.Header.Set("Connection", "close")
51 res, err := DefaultClient.Do(getReq)
53 t.Fatalf("Get: %v", err)
55 if g, e := res.StatusCode, backendStatus; g != e {
56 t.Errorf("got res.StatusCode %d; expected %d", g, e)
58 if g, e := res.Header.Get("X-Foo"), "bar"; g != e {
59 t.Errorf("got X-Foo %q; expected %q", g, e)
61 if g, e := len(res.Header["Set-Cookie"]), 1; g != e {
62 t.Fatalf("got %d SetCookies, want %d", g, e)
64 if cookie := res.Cookies()[0]; cookie.Name != "flavor" {
65 t.Errorf("unexpected cookie %q", cookie.Name)
67 bodyBytes, _ := ioutil.ReadAll(res.Body)
68 if g, e := string(bodyBytes), backendResponse; g != e {
69 t.Errorf("got body %q; expected %q", g, e)