tools/chrome_proxy/testserver/server_test.go

   1 package server
   2
   3 import (
   4         "encoding/base64"
   5         "encoding/json"
   6         "net/http"
   7         "net/http/httptest"
   8         "net/url"
   9         "reflect"
  10         "strconv"
  11         "testing"
  12 )
  13
  14 func composeQuery(path string, code int, headers http.Header, body []byte) (string, error) {
  15         u, err := url.Parse(path)
  16         if err != nil {
  17                 return "", err
  18         }
  19         q := u.Query()
  20         if code > 0 {
  21                 q.Set("respStatus", strconv.Itoa(code))
  22         }
  23         if headers != nil {
  24                 h, err := json.Marshal(headers)
  25                 if err != nil {
  26                         return "", err
  27                 }
  28                 q.Set("respHeader", base64.URLEncoding.EncodeToString(h))
  29         }
  30         if len(body) > 0 {
  31                 q.Set("respBody", base64.URLEncoding.EncodeToString(body))
  32         }
  33         u.RawQuery = q.Encode()
  34         return u.String(), nil
  35 }
  36
  37 func TestResponseOverride(t *testing.T) {
  38         tests := []struct {
  39                 name    string
  40                 code    int
  41                 headers http.Header
  42                 body    []byte
  43         }{
  44                 {name: "code", code: 204},
  45                 {name: "body", body: []byte("new body")},
  46                 {
  47                         name: "headers",
  48                         headers: http.Header{
  49                                 "Via":          []string{"Via1", "Via2"},
  50                                 "Content-Type": []string{"random content"},
  51                         },
  52                 },
  53                 {
  54                         name: "everything",
  55                         code: 204,
  56                         body: []byte("new body"),
  57                         headers: http.Header{
  58                                 "Via":          []string{"Via1", "Via2"},
  59                                 "Content-Type": []string{"random content"},
  60                         },
  61                 },
  62         }
  63
  64         for _, test := range tests {
  65                 u, err := composeQuery("http://test.com/override", test.code, test.headers, test.body)
  66                 if err != nil {
  67                         t.Errorf("%s: composeQuery: %v", test.name, err)
  68                         return
  69                 }
  70                 req, err := http.NewRequest("GET", u, nil)
  71                 if err != nil {
  72                         t.Errorf("%s: http.NewRequest: %v", test.name, err)
  73                         return
  74                 }
  75                 w := httptest.NewRecorder()
  76                 defaultResponse(w, req)
  77                 if test.code > 0 {
  78                         if got, want := w.Code, test.code; got != want {
  79                                 t.Errorf("%s: response code: got %d want %d", test.name, got, want)
  80                                 return
  81                         }
  82                 }
  83                 if test.headers != nil {
  84                         for k, want := range test.headers {
  85                                 got, ok := w.HeaderMap[k]
  86                                 if !ok || !reflect.DeepEqual(got, want) {
  87                                         t.Errorf("%s: header %s: code: got %v want %v", test.name, k, got, want)
  88                                         return
  89                                 }
  90                         }
  91                 }
  92                 if test.body != nil {
  93                         if got, want := string(w.Body.Bytes()), string(test.body); got != want {
  94                                 t.Errorf("%s: body: got %s want %s", test.name, got, want)
  95                                 return
  96                         }
  97                 }
  98         }
  99 }