wehub-resource-sync
79 lines
3.1 KiB
TypeScript
79 lines
3.1 KiB
TypeScript
import { afterEach, beforeAll, describe, expect, it } from "bun:test";
|
|
import {
|
|
buildLargeTextLines,
|
|
buildSampleImage,
|
|
encodeRgbPng,
|
|
ProtocolProbeComponent,
|
|
} from "@oh-my-pi/pi-coding-agent/debug/protocol-probe";
|
|
import { initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import { getImageDimensions, ImageBudget, ImageProtocol, TERMINAL } from "@oh-my-pi/pi-tui";
|
|
|
|
beforeAll(async () => {
|
|
// buildLargeTextLines styles the OSC 66 span through the global theme singleton.
|
|
await initTheme();
|
|
});
|
|
|
|
type MutableTerminalInfo = { imageProtocol: ImageProtocol | null };
|
|
const terminal = TERMINAL as unknown as MutableTerminalInfo;
|
|
const originalImageProtocol = TERMINAL.imageProtocol;
|
|
|
|
afterEach(() => {
|
|
terminal.imageProtocol = originalImageProtocol;
|
|
});
|
|
|
|
describe("encodeRgbPng / buildSampleImage", () => {
|
|
it("emits a PNG whose IHDR dimensions decode back to the requested size", () => {
|
|
// The graphics test relies on producing a *valid* PNG: Kitty/iTerm2 hand
|
|
// the bytes to the terminal and Sixel decodes them natively, so a malformed
|
|
// header (wrong CRC, bad chunk framing) silently breaks the whole panel.
|
|
const sample = buildSampleImage(48, 24);
|
|
expect(sample.mimeType).toBe("image/png");
|
|
expect(sample.dimensions).toEqual({ widthPx: 48, heightPx: 24 });
|
|
|
|
const decoded = getImageDimensions(sample.base64, "image/png");
|
|
expect(decoded).toEqual({ widthPx: 48, heightPx: 24 });
|
|
});
|
|
|
|
it("begins with the PNG signature and round-trips arbitrary pixel sizes", () => {
|
|
const png = encodeRgbPng(3, 2, new Uint8Array(3 * 2 * 3));
|
|
expect(Array.from(png.subarray(0, 8))).toEqual([0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a]);
|
|
|
|
const decoded = getImageDimensions(Buffer.from(png).toString("base64"), "image/png");
|
|
expect(decoded).toEqual({ widthPx: 3, heightPx: 2 });
|
|
});
|
|
});
|
|
|
|
it("uses independent graphics ids for repeated probe panels", () => {
|
|
terminal.imageProtocol = ImageProtocol.Kitty;
|
|
const budget = new ImageBudget(8, () => {});
|
|
const image = buildSampleImage(8, 8);
|
|
const first = new ProtocolProbeComponent({ image, imageBudget: budget, notificationSuppressed: true });
|
|
const second = new ProtocolProbeComponent({ image, imageBudget: budget, notificationSuppressed: true });
|
|
|
|
budget.beginPass();
|
|
const firstBytes = first.render(80).join("\n");
|
|
const secondBytes = second.render(80).join("\n");
|
|
budget.endPass();
|
|
|
|
const firstMatch = firstBytes.match(/i=(\d+)/);
|
|
const secondMatch = secondBytes.match(/i=(\d+)/);
|
|
expect(firstMatch).not.toBeNull();
|
|
expect(secondMatch).not.toBeNull();
|
|
const firstId = Number(firstMatch![1]);
|
|
const secondId = Number(secondMatch![1]);
|
|
expect(secondId).toBe((firstId + 1) & 0xffffff || 1);
|
|
});
|
|
|
|
describe("buildLargeTextLines", () => {
|
|
it("encodes each scale as an OSC 66 span and reserves scale-1 blank rows below it", () => {
|
|
const lines = buildLargeTextLines([2, 3]);
|
|
// scale 2 → 1 reserved row, scale 3 → 2 reserved rows.
|
|
expect(lines).toHaveLength(5);
|
|
expect(lines[0]).toContain("\x1b]66;s=2;");
|
|
expect(lines[1]).toBe("");
|
|
expect(lines[2]).toContain("\x1b]66;s=3;");
|
|
expect(lines[3]).toBe("");
|
|
expect(lines[4]).toBe("");
|
|
});
|
|
});
|