Function TestIntercepts

elasticsearch_internal_test.go:1610–1732

(t *testing.T)

}

func TestIntercepts(t *testing.T) {
	successTp := func(request *http.Request) (*http.Response, error) {
		h := request.Header.Clone()
		h.Add("X-Elastic-Product", "Elasticsearch")
		h.Add("X-Found-Handling-Cluster", "foo-bar-cluster-id")
		h.Add("X-Found-Handling-Instance", "0123456789")
		return &http.Response{
			StatusCode: http.StatusOK,
			Header:     h,
			Body:       io.NopCloser(strings.NewReader(`{}`)),
		}, nil
	}

	const (
		testInterceptReqHeader  = "X-TEST-INTERCEPT-REQ"
		testInterceptRespHeader = "X-TEST-INTERCEPT-RESP"
	)

	dummyInterceptorFactory := func(num int) elastictransport.InterceptorFunc {
		return func(next elastictransport.RoundTripFunc) elastictransport.RoundTripFunc {
			return func(r *http.Request) (*http.Response, error) {
				r.Header.Add(testInterceptReqHeader, strconv.Itoa(num))
				resp, err := next(r)
				resp.Header.Add(testInterceptRespHeader, strconv.Itoa(num))
				return resp, err
			}
		}
	}

	validateReqRespH := func(expectReq, expectResp string) func(t *testing.T, resp *esapi.Response) {
		return func(t *testing.T, resp *esapi.Response) {
			if strings.Join(resp.Header.Values(testInterceptReqHeader), ",") != expectReq {
				t.Error("expected header value", resp.Header.Values(testInterceptReqHeader))
			}
			if strings.Join(resp.Header.Values(testInterceptRespHeader), ",") != expectResp {
				t.Error("expected header value", resp.Header.Values(testInterceptRespHeader))
			}
		}
	}

	type args struct {
		interceptors []elastictransport.InterceptorFunc
	}
	tests := []struct {
		name             string
		args             args
		wantErr          bool
		validateResponse func(t *testing.T, resp *esapi.Response)
	}{
		{
			name:    "interceptor array nil",
			args:    args{interceptors: nil},
			wantErr: false,
		},
		{
			name:    "interceptor array empty",
			args:    args{interceptors: []elastictransport.InterceptorFunc{}},
			wantErr: false,