import { describe, expect, it } from "vitest";
import type { KernelAttachment } from "../src/core/kernel/index.js";
import { imageBlocksFromAttachments } from "../src/core/tools/ipython.js";

describe("imageBlocksFromAttachments", () => {
	it("returns no blocks when there are no attachments", () => {
		expect(imageBlocksFromAttachments(undefined)).toEqual([]);
		expect(imageBlocksFromAttachments([])).toEqual([]);
	});

	it("converts supported image attachments to ImageContent blocks", () => {
		const attachments: KernelAttachment[] = [
			{ mimeType: "image/png", data: "AAAA", path: "/tmp/a.png" },
			{ mimeType: "image/jpeg", data: "BBBB" },
		];
		expect(imageBlocksFromAttachments(attachments)).toEqual([
			{ type: "image", data: "AAAA", mimeType: "image/png" },
			{ type: "image", data: "BBBB", mimeType: "image/jpeg" },
		]);
	});

	it("drops non-image attachments", () => {
		const attachments: KernelAttachment[] = [
			{ mimeType: "application/pdf", data: "CCCC" },
			{ mimeType: "image/webp", data: "DDDD" },
		];
		expect(imageBlocksFromAttachments(attachments)).toEqual([
			{ type: "image", data: "DDDD", mimeType: "image/webp" },
		]);
	});
});