feat: retry parts (#3369)

packages/opencode/test/session/retry.test.ts

@@ -0,0 +1,47 @@
+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)
+  })
+})