1import assert from "node:assert/strict";
2import test from "node:test";
3 
4import { collect_generated_image_urls_from_response_parts } from "./client.ts";
5 
6test("response part fallback finds generated images when legacy generated markers are absent", () => {
7  const generatedUrl = "https://lh3.googleusercontent.com/gg-dl/example-generated-image";
8  const initialCandidate = ["rcid-1", ["image generated successfully"]];
9  const imageCandidate = [
10    "rcid-1",
11    ["image generated successfully"],
12    { nestedPayload: [{ media: generatedUrl }] },
13  ];
14  const responseJson = [
15    ["wrb.fr", null, JSON.stringify([null, [], null, null, [initialCandidate]])],
16    ["wrb.fr", null, JSON.stringify([null, [], null, null, [imageCandidate]])],
17  ];
18 
19  assert.equal(initialCandidate[12], undefined);
20  assert.equal(
21    /http:\/\/googleusercontent\.com\/image_generation_content\/\d+/.test(String(initialCandidate[1]?.[0])),
22    false,
23  );
24  assert.deepEqual(collect_generated_image_urls_from_response_parts(responseJson, 0, 0), [generatedUrl]);
25});
26