Function TestDoClientSimpleRequest

pkg/http/client_test.go:243–286

(t *testing.T)

}

func TestDoClientSimpleRequest(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)

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

	var (
		path     = "/foo"
		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,
		}
	)
	req.Target.ParseHostHeader()
	resp, err := DoClient(client, req, &config)
	assert.Nil(t, err)

	// we expect the request uri for the first request to be empty
	_ = fmt.Sprintf("%s%s", simpleServer.URL, path)
	assert.Equal(t, "", string(resp.URI))

	// we also expect our headers and body to come back as expected
	expected := []*Header{
		{"X-Custom-Header", "key"},
		{"X-Request-Url", "/foo"},
	}
	for _, v := range expected {
		assert.Contains(t, resp.Headers, v)
	}

	assert.NotEqual(t, "", string(resp.Body))
}

func TestDoClientRedirectRequest(t *testing.T) {
	simpleServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {