import { describe, expect, it } from "vitest";
import { gradientImage, noiseImage, solidImage } from "./generate";

describe("solidImage", () => {
	it("заливает весь холст заданным цветом", () => {
		const out = solidImage(2, 2, [10, 20, 30, 255]);
		expect(out.width).toBe(2);
		expect([...out.data]).toEqual(new Array(4).fill([10, 20, 30, 255]).flat());
	});

	it.each([
		[0, 10],
		[10, 0],
		[2.5, 10],
		[-1, 5],
	])("бросает ошибку на размерах %i x %i", (w, h) => {
		expect(() => solidImage(w, h, [0, 0, 0, 255])).toThrow();
	});
});

describe("noiseImage", () => {
	it("детерминирован: одно зерно — одни байты", () => {
		expect([...noiseImage(4, 4, 42).data]).toEqual([
			...noiseImage(4, 4, 42).data,
		]);
	});

	it("разные зерна дают разные данные", () => {
		const a = [...noiseImage(8, 8, 1).data];
		const b = [...noiseImage(8, 8, 2).data];
		expect(a).not.toEqual(b);
	});

	it("альфа всегда непрозрачная", () => {
		const data = noiseImage(3, 3, 7).data;
		for (let i = 3; i < data.length; i += 4) {
			expect(data[i]).toBe(255);
		}
	});
});

describe("gradientImage", () => {
	it("горизонтальный градиент идёт от цвета A к цвету B", () => {
		const out = gradientImage(
			3,
			1,
			[0, 0, 0, 255],
			[255, 255, 255, 255],
			"horizontal",
		);
		const px = (x: number) => [...out.data.slice(x * 4, x * 4 + 4)];
		expect(px(0)).toEqual([0, 0, 0, 255]);
		expect(px(1)).toEqual([128, 128, 128, 255]);
		expect(px(2)).toEqual([255, 255, 255, 255]);
	});

	it("вертикальный градиент меняется по строкам", () => {
		const out = gradientImage(
			1,
			2,
			[0, 0, 0, 255],
			[100, 100, 100, 255],
			"vertical",
		);
		expect(out.data[0]).toBe(0);
		expect(out.data[4]).toBe(100);
	});
});