test: cover wire shape, auth, empty input, batchSize, Retry-After date, override, and timeout with signal (CL-9107)

Author: TheGreatAxios

src/embed.test.ts

@@ -111,3 +111,56 @@ test("classifies a 200 that is not JSON instead of leaking a SyntaxError", async
     reason: { category: "protocol_mismatch" },
   });
 });
+
+test("sends only configured fields to {baseURL}/embeddings", async () => {
+  harness = setupHarness();
+  const stream = harness.scenario.createStream();
+  harness.scenario.whenRequestBodyMatches(
+    (body, req) =>
+      req.url === "https://embed.example/v1/embeddings" &&
+      !req.headers.has("authorization") &&
+      body === JSON.stringify({ model: CONFIG.model, input: ["a"] }),
+    stream,
+  );
+  stream.enqueueAll(
+    [new TextEncoder().encode('{"data":[{"index":0,"embedding":[1]}]}')],
+    { startAt: 1 },
+  );
+  const pending = embedTexts(["a"], CONFIG, { deps: harness.deps });
+  expect(await run(harness, pending)).toEqual([[1]]);
+});
+
+test("sends a bearer token only when apiKey is set", async () => {
+  harness = setupHarness();
+  const stream = harness.scenario.createStream();
+  harness.scenario.whenRequestMatches(
+    (req) => req.headers.get("authorization") === "Bearer sk-test",
+    stream,
+  );
+  stream.enqueueAll(
+    [new TextEncoder().encode('{"data":[{"index":0,"embedding":[1]}]}')],
+    { startAt: 1 },
+  );
+  const pending = embedTexts(
+    ["a"],
+    { ...CONFIG, apiKey: "sk-test" },
+    { deps: harness.deps },
+  );
+  expect(await run(harness, pending)).toEqual([[1]]);
+});
+
+test("returns [] for empty input without a request", async () => {
+  harness = setupHarness();
+  const pending = embedTexts([], CONFIG, { deps: harness.deps });
+  expect(await run(harness, pending)).toEqual([]);
+});
+
+test("rejects a non-positive batchSize before any request", async () => {
+  harness = setupHarness();
+  for (const batchSize of [0, -1, 1.5]) {
+    await expect(
+      embedTexts(["a"], { ...CONFIG, batchSize }, { deps: harness.deps }),
+    ).rejects.toThrow();
+  }
+  await harness.run();
+});

tests/retry.test.ts

@@ -53,3 +53,46 @@ test("rejects a 401 without retrying", async () => {
     reason: { category: "credential_failure", statusCode: 401 },
   });
 });
+
+test("waits until an HTTP-date Retry-After", async () => {
+  harness = setupHarness({ enableInferenceTimers: true });
+  const at = new Date(Date.now() + 20_000).toUTCString();
+  reply(harness, "rate limited", 429, { "retry-after": at });
+  reply(harness, JSON.stringify({ data: [{ index: 0, embedding: [1] }] }), 200);
+
+  const pending = embedTexts(["a"], CONFIG, { deps: harness.deps });
+  await harness.run();
+
+  expect(await pending).toEqual([[1]]);
+  expect(harness.clock.now()).toBeGreaterThanOrEqual(19_000);
+});
+
+test("honors a caller-supplied extractRetryAfterMs", async () => {
+  harness = setupHarness({ enableInferenceTimers: true });
+  reply(harness, "rate limited", 429);
+  reply(harness, JSON.stringify({ data: [{ index: 0, embedding: [1] }] }), 200);
+
+  const pending = embedTexts(["a"], CONFIG, {
+    deps: harness.deps,
+    extractRetryAfterMs: () => 45_000,
+  });
+  await harness.run();
+
+  expect(await pending).toEqual([[1]]);
+  expect(harness.clock.now()).toBeGreaterThanOrEqual(45_000);
+});
+
+test("applies timeoutMs even when the caller passes a signal", async () => {
+  harness = setupHarness();
+  const stream = harness.scenario.createStream();
+  harness.scenario.whenRequestMatches(() => true, stream);
+
+  const pending = embedTexts(
+    ["a"],
+    { ...CONFIG, timeoutMs: 50 },
+    { deps: harness.deps, signal: new AbortController().signal },
+  );
+  await harness.run();
+
+  await expect(pending).rejects.toBeInstanceOf(EmbeddingRequestError);
+});