Function TestDoClientHostHeaderRequestCustom

pkg/http/client_test.go:189–241

TestDoClientHostHeaderRequestCustom should validate that a custom hostname for a target will be obeyed and will not include the port

(t *testing.T)

// TestDoClientHostHeaderRequestCustom should validate that a custom hostname for a target will be obeyed
// and will not include the port
func TestDoClientHostHeaderRequestCustom(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{
				HostHeader: []byte("no-port-host-header"),
				Hostname:   "diff-hostname-123",
				IP:         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"},
		{"X-Host", "no-port-host-header"},
	}
	for _, v := range expected {
		assert.Contains(t, resp.Headers, v)
	}

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

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