func (s *RewriteSuite) TestSameHeaderMulti(c *C) { var outURL string handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { outURL = rawURL(req) w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{ "^http://localhost/(foo)/(bar)$", `http://localhost/$1/{{.Request.Header.Get "X-Header"}}/$2/{{.Request.Header.Get "X-Header"}}`, false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, err := testutils.Get(srv.URL+"/foo/bar", testutils.Host("localhost"), testutils.Header("X-Header", "baz")) c.Assert(err, IsNil) c.Assert(re.StatusCode, Equals, http.StatusOK) c.Assert(outURL, Equals, "http://localhost/foo/baz/bar/baz") }
func (s *RewriteSuite) TestRedirect(c *C) { handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{"^http://localhost/(foo)/(bar)", "https://localhost/$2", false, true}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, err := testutils.Get(srv.URL+"/foo/bar", testutils.Host("localhost")) c.Assert(re.StatusCode, Equals, http.StatusFound) c.Assert(re.Header.Get("Location"), Equals, "https://localhost/bar") }
func (s *RewriteSuite) TestRewriteNoMatch(c *C) { var outURL string handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { outURL = rawURL(req) w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{"^http://localhost/foo/(.*)", "http://localhost$1", false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, err := testutils.Get(srv.URL+"/fooo/bar", testutils.Host("localhost")) c.Assert(err, IsNil) c.Assert(re.StatusCode, Equals, http.StatusOK) c.Assert(outURL, Equals, "http://localhost/fooo/bar") }
func (s *RewriteSuite) TestDontRewriteResponseBody(c *C) { handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.Write([]byte(`{"foo": "{{.Request.Header.Get "X-Header"}}"}`)) }) rh, err := newRewriteHandler(handler, &Rewrite{"", "", false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, body, err := testutils.Get(srv.URL, testutils.Host("localhost"), testutils.Header("X-Header", "bar")) c.Assert(re.StatusCode, Equals, http.StatusOK) c.Assert(string(body), Equals, `{"foo": "{{.Request.Header.Get "X-Header"}}"}`) }
func (s *RewriteSuite) TestRewriteInQuery(c *C) { var outURL string handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { outURL = rawURL(req) w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{"^http://localhost/foo\\?(.*)=(.*)", `http://localhost/foo?$1={{.Request.Header.Get "X-Header"}}`, false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, err := testutils.Get(srv.URL+"/foo?a=b", testutils.Host("localhost"), testutils.Header("X-Header", "xxx")) c.Assert(err, IsNil) c.Assert(re.StatusCode, Equals, http.StatusOK) c.Assert(outURL, Equals, "http://localhost/foo?a=xxx") }
func (s *RewriteSuite) TestUnknownVar(c *C) { var outURL string handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { outURL = rawURL(req) w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{"^http://localhost/(foo)/(bar)$", `http://localhost/$1/{{.Bad}}/$2`, false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, err := testutils.Get(srv.URL+"/foo/bar", testutils.Host("localhost")) c.Assert(err, IsNil) c.Assert(re.StatusCode, Equals, http.StatusInternalServerError) }
// TestContentLength makes sure Content-Length is re-calculated if body rewrite is enabled. func (s *RewriteSuite) TestContentLength(c *C) { handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { w.Header().Set("Content-Length", "45") w.WriteHeader(200) w.Write([]byte(`{"foo": "{{.Request.Header.Get "X-Header"}}"}`)) }) rh, err := newRewriteHandler(handler, &Rewrite{"", "", true, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewServer(rh) defer srv.Close() re, _, _ := testutils.Get(srv.URL, testutils.Host("localhost"), testutils.Header("X-Header", "bar")) c.Assert(re.Header.Get("Content-Length"), Equals, "14") }
// What real-world scenario does this test? func (s *RewriteSuite) TestRewriteScheme(c *C) { var outURL *url.URL handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { outURL = req.URL w.Write([]byte("hello")) }) rh, err := newRewriteHandler(handler, &Rewrite{"^https://localhost/(foo)/(bar)$", "http://localhost/$1/$2", false, false}) c.Assert(rh, NotNil) c.Assert(err, IsNil) srv := httptest.NewUnstartedServer(rh) srv.StartTLS() defer srv.Close() re, _, err := testutils.Get(srv.URL+"/foo/bar", testutils.Host("localhost")) c.Assert(err, IsNil) c.Assert(re.StatusCode, Equals, http.StatusOK) c.Assert(outURL.Scheme, Equals, "http") c.Assert(outURL.Path, Equals, "/foo/bar") c.Assert(outURL.Host, Equals, "localhost") }