48 lines
1.8 KiB
TypeScript
48 lines
1.8 KiB
TypeScript
import { describe, expect, test } from "bun:test"
|
|
import { SessionRetry } from "../../src/session/retry"
|
|
import { MessageV2 } from "../../src/session/message-v2"
|
|
|
|
function apiError(headers?: Record<string, string>): MessageV2.APIError {
|
|
return new MessageV2.APIError({
|
|
message: "boom",
|
|
isRetryable: true,
|
|
responseHeaders: headers,
|
|
}).toObject() as MessageV2.APIError
|
|
}
|
|
|
|
describe("session.retry.getRetryDelayInMs", () => {
|
|
test("doubles delay on each attempt when headers missing", () => {
|
|
const error = apiError()
|
|
const delays = Array.from({ length: 7 }, (_, index) => SessionRetry.getRetryDelayInMs(error, index + 1))
|
|
expect(delays).toStrictEqual([2000, 4000, 8000, 16000, 32000, 64000, 128000])
|
|
})
|
|
|
|
test("prefers retry-after-ms when shorter than exponential", () => {
|
|
const error = apiError({ "retry-after-ms": "1500" })
|
|
expect(SessionRetry.getRetryDelayInMs(error, 4)).toBe(1500)
|
|
})
|
|
|
|
test("uses retry-after seconds when reasonable", () => {
|
|
const error = apiError({ "retry-after": "30" })
|
|
expect(SessionRetry.getRetryDelayInMs(error, 3)).toBe(30000)
|
|
})
|
|
|
|
test("falls back to exponential when server delay is long", () => {
|
|
const error = apiError({ "retry-after": "120" })
|
|
expect(SessionRetry.getRetryDelayInMs(error, 2)).toBe(4000)
|
|
})
|
|
|
|
test("accepts http-date retry-after values", () => {
|
|
const date = new Date(Date.now() + 20000).toUTCString()
|
|
const error = apiError({ "retry-after": date })
|
|
const delay = SessionRetry.getRetryDelayInMs(error, 1)
|
|
expect(delay).toBeGreaterThanOrEqual(19000)
|
|
expect(delay).toBeLessThanOrEqual(20000)
|
|
})
|
|
|
|
test("ignores invalid retry hints", () => {
|
|
const error = apiError({ "retry-after": "not-a-number" })
|
|
expect(SessionRetry.getRetryDelayInMs(error, 1)).toBe(2000)
|
|
})
|
|
})
|