Function TestDoClientRedirectRequest

pkg/http/client_test.go:288–360

(t *testing.T)

}

func TestDoClientRedirectRequest(t *testing.T) {
	simpleServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("x-custom-header", "key")
		w.Header().Set("x-request-url", r.URL.String())
		w.Header().Set("x-host", r.Host)

		if r.URL.Path == "/before" {
			w.Header().Set("Location", "/after")
			w.WriteHeader(302)
		} else if r.URL.Path == "/after" {
			w.WriteHeader(201)
		}

		body, err := httputil.DumpRequest(r, true)
		assert.Nil(t, err)
		w.Write(body)
	}))

	var (
		path     = "/before"
		client   = NewHTTPClient(simpleServer.Listener.Addr().String(), false)
		hostname = strings.SplitN(simpleServer.Listener.Addr().String(), ":", 2)[0]
		port, _  = strconv.Atoi(strings.SplitN(simpleServer.Listener.Addr().String(), ":", 2)[1])
		req      = Request{Route: &Route{Path: []byte(path)}, Target: &Target{Hostname: hostname, Port: port}}
		resp     = Response{}
		config   = Config{
			Timeout:      1 * time.Second,
			ReadHeaders:  true,
			ReadBody:     true,
			MaxRedirects: 2,
		}
	)
	req.Target.ParseHostHeader()
	resp, err := DoClient(client, req, &config)
	assert.Nil(t, err)

	// First Request
	resp = resp
	// we expect the request uri to be empty
	assert.Equal(t, "", string(resp.URI))
	// it should be a redirect
	assert.Equal(t, 302, resp.StatusCode)
	// we also expect our headers and body to come back as expected
	expected := []*Header{
		{"X-Custom-Header", "key"},
		{"X-Request-Url", "/before"},
	}
	for _, v := range expected {
		assert.Contains(t, resp.Headers, v)
	}
	assert.NotEqual(t, "", string(resp.Body))

	// redirected request
	nresp := resp.Next
	// url := fmt.Sprintf("%s/after", simpleServer.URL)
	// we expect the URI to contain the location header
	assert.Equal(t, "/after", string(nresp.URI))
	// it should be a redirect