Function TestRetryTransport_WithCustomConfig

internal/http/retry_test.go:365–427

(t *testing.T)

}

func TestRetryTransport_WithCustomConfig(t *testing.T) {
	// Set custom retry configuration
	customConfig := RetryConfig{
		MaxWait:  200 * time.Millisecond, // Much shorter for testing
		MaxRetry: 2,
		Duration: 25 * time.Millisecond, // Much shorter for testing
		Factor:   2.0,
		Jitter:   0.1,
	}
	SetRetryConfig(customConfig)
	defer func() {
		// Restore default configuration
		SetRetryConfig(RetryConfig{
			MaxWait:  3 * time.Second,
			MaxRetry: 3,
			Duration: 1 * time.Second,
			Factor:   2.0,
			Jitter:   0.1,
		})
	}()

	callCount := 0
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		callCount++
		if callCount <= 2 {
			w.WriteHeader(http.StatusTooManyRequests)
			_, err := w.Write([]byte(`{"error": "Too Many Requests"}`))
			if err != nil {
				t.Errorf("Failed to write response: %v", err)
			}
			return
		}
		w.WriteHeader(http.StatusOK)
		_, err := w.Write([]byte(`{"success": true}`))
		if err != nil {
			t.Errorf("Failed to write response: %v", err)
		}
	}))
	defer server.Close()

	// Create a transport that will retry on 429 with custom configuration
	baseTransport := &http.Transport{}
	retryTransport := NewRetryTransport(baseTransport)

	// Create a request
	req, err := http.NewRequest("GET", server.URL, nil)
	require.NoError(t, err)

	// Execute the request
	start := time.Now()
	resp, err := retryTransport.RoundTrip(req)
	duration := time.Since(start)

	// Verify the response
	require.NoError(t, err)
	assert.Equal(t, http.StatusOK, resp.StatusCode)

	// Verify that we retried (should have called the server 3 times)