Function testLoopback

proxy/loopback_test.go:55–167

(
	t *testing.T,
	routes string,
	params Params,
	input *url.URL,
	expectedStatus int,
	expectedHeader http.Header,
)

}

func testLoopback(
	t *testing.T,
	routes string,
	params Params,
	input *url.URL,
	expectedStatus int,
	expectedHeader http.Header,
) {
	var backend *httptest.Server
	backend = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if params.Flags.PreserveOriginal() {
			if r.Header.Get("X-Test-Preserved") != "test-value" {
				t.Error("failed to preserve original request")
				return
			}

			w.Header().Set("X-Test", "test-value")
		}

		if params.Flags.PreserveHost() && r.Host != "www.example.org" {
			t.Error("failed to preserve host")
		} else if !params.Flags.PreserveHost() {
			u, err := url.Parse(backend.URL)
			if err != nil {
				t.Error(err)
				return
			}

			if r.Host != u.Host {
				t.Error("failed to set host")
				return
			}
		}

		w.Header().Set("X-Backend-Done", "true")
	}))
	defer backend.Close()

	routes = strings.ReplaceAll(routes, "$backend", backend.URL)

	fr := builtin.MakeRegistry()
	fr.Register(&preserveOriginalSpec{})
	fr.Register(&returnPathParam{})
	fr.Register(&setState{})
	fr.Register(&returnState{})
	fr.Register(builtin.NewLoopbackIfStatus())

	p, err := newTestProxyWithFiltersAndParams(fr, routes, params, nil)
	if err != nil {
		t.Error(err)
		return
	}

	defer p.close()

	var u *url.URL
	if input == nil {
		u, err = url.ParseRequestURI("https://www.example.org/test/path")