Function TestGetRoundtripForDynamicBackend

proxy/proxy_test.go:554–620

(t *testing.T)

}

func TestGetRoundtripForDynamicBackend(t *testing.T) {
	payload := []byte("Hello World!")

	s := startTestServer(payload, 0, func(r *http.Request) {
		if th, ok := r.Header["X-Test-Header"]; !ok || th[0] != "test value" {
			t.Error("wrong request header")
		}
	})

	defer s.Close()

	fr := make(filters.Registry)
	fr.Register(builtin.NewSetDynamicBackendHost())
	fr.Register(builtin.NewSetDynamicBackendScheme())
	fr.Register(builtin.NewSetDynamicBackendUrl())

	w := httptest.NewRecorder()

	bu, _ := url.ParseRequestURI(s.URL)
	doc := fmt.Sprintf(
		`dynamic: Method("GET") -> setDynamicBackendScheme(%q) ->setDynamicBackendHost(%q) -> <dynamic>;`+
			`dynamic2: Method("POST") -> setDynamicBackendUrl(%q) -> <dynamic>;`+
			`dynamic3: Path("/defaults") -> <dynamic>;`, bu.Scheme, bu.Host, s.URL)

	tp, err := newTestProxyWithFilters(fr, doc, FlagsNone)
	if err != nil {
		t.Error(err)
		return
	}

	defer tp.close()

	u1, _ := url.ParseRequestURI("https://example1.com")
	r1 := &http.Request{
		URL:    u1,
		Method: "GET",
		Header: http.Header{"X-Test-Header": []string{"test value"}}}
	tp.proxy.ServeHTTP(w, r1)

	if w.Code != http.StatusOK {
		t.Error("wrong status", w.Code)
	}

	u2, _ := url.ParseRequestURI("https://example2.com")
	r2 := &http.Request{
		URL:    u2,
		Method: "POST",
		Header: http.Header{"X-Test-Header": []string{"test value"}}}
	tp.proxy.ServeHTTP(w, r2)

	if w.Code != http.StatusOK {
		t.Error("wrong status", w.Code)
	}

	u3 := &url.URL{Path: "/defaults"}
	r3 := &http.Request{
		URL:    u3,
		Method: "HEAD",