| import { describe, expect, it } from "vitest"; |
| import { getImageMetadata } from "../../media/image-ops.js"; |
| import { createSolidPngBuffer } from "../../plugin-sdk/test-helpers/image-fixtures.js"; |
| import { projectScreenshotResult } from "./computer-tool-result.js"; |
|
|
| describe("computer screenshot dimensions", () => { |
| it.each([ |
| [1024, 768, 1200, 1024, 768, true], |
| [1080, 1920, 1280, 720, 1280, true], |
| [1200, 1243, 1200, 1158, 1200, true], |
| [1, 3, 1, 1, 1, true], |
| [32, 16, 1200, 32, 16, false], |
| ] as const)( |
| "reports %dx%d at cap %d as the delivered %dx%d image", |
| async (width, height, referenceWidth, deliveredWidth, deliveredHeight, reportsDimensions) => { |
| const { result } = await projectScreenshotResult({ |
| capture: { |
| base64: createSolidPngBuffer(width, height, { r: 70, g: 125, b: 180 }).toString("base64"), |
| mimeType: "image/png", |
| displayFrameId: "display-frame", |
| ...(reportsDimensions ? { width, height } : {}), |
| }, |
| target: { host: "node", nodeId: "desktop-node", screenIndex: 0 }, |
| action: "screenshot", |
| noteLines: [], |
| referenceWidth, |
| }); |
| const image = result.content.find((block) => block.type === "image"); |
| if (!image) { |
| throw new Error("The screenshot image is missing"); |
| } |
| const metadata = await getImageMetadata(Buffer.from(image.data, "base64")); |
| expect(metadata).toMatchObject({ width: deliveredWidth, height: deliveredHeight }); |
| expect(result.details).toMatchObject({ |
| width: deliveredWidth, |
| height: deliveredHeight, |
| refWidth: referenceWidth, |
| }); |
| expect(result.content).toContainEqual({ |
| type: "text", |
| text: expect.stringContaining(`screenshot ${deliveredWidth}x${deliveredHeight}`), |
| }); |
| }, |
| ); |
| }); |
|
|
