Function TestParseRetryAfter

aibridge/keypool/headers_test.go:13–110

(t *testing.T)

)

func TestParseRetryAfter(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name        string
		headers     map[string]string
		nilResponse bool
		expected    time.Duration
	}{
		// nil response.
		{
			name:        "nil_response",
			nilResponse: true,
			expected:    0,
		},
		// No headers set.
		{
			name:     "no_headers",
			headers:  nil,
			expected: 0,
		},
		// retry-after-ms (OpenAI, preferred).
		{
			name:     "openai_retry_after_ms",
			headers:  map[string]string{"retry-after-ms": "2500"},
			expected: 2500 * time.Millisecond,
		},
		{
			name:     "whitespace_trimmed_ms",
			headers:  map[string]string{"retry-after-ms": " 1500 "},
			expected: 1500 * time.Millisecond,
		},
		{
			name:     "negative_ms_returns_zero",
			headers:  map[string]string{"retry-after-ms": "-100"},
			expected: 0,
		},
		// Retry-After (standard, seconds).
		{
			name:     "standard_retry_after_seconds",
			headers:  map[string]string{"Retry-After": "60"},
			expected: 60 * time.Second,
		},
		{
			name:     "whitespace_trimmed_seconds",
			headers:  map[string]string{"Retry-After": " 30 "},
			expected: 30 * time.Second,
		},
		{
			name:     "zero_seconds_returns_zero",
			headers:  map[string]string{"Retry-After": "0"},
			expected: 0,
		},
		{
			name:     "negative_seconds_returns_zero",
			headers:  map[string]string{"Retry-After": "-5"},
			expected: 0,
		},