}
}
-// Test that the special cased "/route" redirect
-// implicitly created by a registered "/route/"
-// properly sets the query string in the redirect URL.
-// See Issue 17841.
-func TestServeWithSlashRedirectKeepsQueryString(t *testing.T) {
- setParallel(t)
- defer afterTest(t)
-
- writeBackQuery := func(w ResponseWriter, r *Request) {
- fmt.Fprintf(w, "%s", r.URL.RawQuery)
- }
-
- mux := NewServeMux()
- mux.HandleFunc("/testOne", writeBackQuery)
- mux.HandleFunc("/testTwo/", writeBackQuery)
-
- ts := httptest.NewServer(mux)
- defer ts.Close()
-
- tests := [...]struct {
- path string
- want string
- }{
- 0: {"/testOne?this=that", "this=that"},
- 1: {"/testTwo?foo=bar", "foo=bar"},
- 2: {"/testTwo?a=1&b=2&a=3", "a=1&b=2&a=3"},
- 3: {"/testTwo?", ""},
- }
-
- for i, tt := range tests {
- res, err := ts.Client().Get(ts.URL + tt.path)
- if err != nil {
- continue
- }
- slurp, _ := ioutil.ReadAll(res.Body)
- res.Body.Close()
- if got, want := string(slurp), tt.want; got != want {
- t.Errorf("#%d: got = %q; want = %q", i, got, want)
- }
- }
-}
-
func BenchmarkResponseStatusLine(b *testing.B) {
b.ReportAllocs()
b.RunParallel(func(pb *testing.PB) {
// The provided code should be in the 3xx range and is usually
// StatusMovedPermanently, StatusFound or StatusSeeOther.
func Redirect(w ResponseWriter, r *Request, urlStr string, code int) {
- queryAlreadySet := false
if u, err := url.Parse(urlStr); err == nil {
// If url was relative, make absolute by
// combining with request path.
urlStr += "/"
}
urlStr += query
- queryAlreadySet = len(query) != 0
}
}
- // We should make sure not to lose the query string of
- // the original request when doing a redirect, if not already set.
- // See Issue 17841.
- if !queryAlreadySet && len(r.URL.RawQuery) != 0 {
- urlStr += "?" + r.URL.RawQuery
- }
-
w.Header().Set("Location", hexEscapeNonASCII(urlStr))
w.WriteHeader(code)