Function TestRetry

modules/frontend/pipeline/sync_handler_retry_test.go:16–124

(t *testing.T)

)

func TestRetry(t *testing.T) {
	var try atomic.Int32

	for _, tc := range []struct {
		name          string
		handler       RoundTripper
		maxRetries    int
		expectedTries int32
		expectedRes   *http.Response
		expectedErr   error
	}{
		{
			name: "retry errors until success",
			handler: RoundTripperFunc(func(_ Request) (*http.Response, error) {
				if try.Inc() == 5 {
					return &http.Response{StatusCode: 200}, nil
				}
				return nil, errors.New("this request failed")
			}),
			maxRetries:    5,
			expectedTries: 5,
			expectedRes:   &http.Response{StatusCode: 200},
			expectedErr:   nil,
		},
		{
			name: "don't retry 400's",
			handler: RoundTripperFunc(func(_ Request) (*http.Response, error) {
				try.Inc()
				return &http.Response{StatusCode: 400}, nil
			}),
			maxRetries:    5,
			expectedTries: 1,
			expectedRes:   &http.Response{StatusCode: 400},
			expectedErr:   nil,
		},
		{
			name: "don't retry GRPC request with HTTP 400's",
			handler: RoundTripperFunc(func(_ Request) (*http.Response, error) {
				try.Inc()
				return nil, httpgrpc.ErrorFromHTTPResponse(&httpgrpc.HTTPResponse{Code: 400})
			}),
			maxRetries:    5,
			expectedTries: 1,
			expectedRes:   nil,
			expectedErr:   httpgrpc.ErrorFromHTTPResponse(&httpgrpc.HTTPResponse{Code: 400}),
		},
		{
			name: "retry 500s",
			handler: RoundTripperFunc(func(_ Request) (*http.Response, error) {
				try.Inc()
				return &http.Response{StatusCode: 503}, nil
			}),
			maxRetries:    5,
			expectedTries: 5,
			expectedRes:   &http.Response{StatusCode: 503},
			expectedErr:   nil,
		},
		{