import { describe, expect, it } from "vitest";
import { resolveImageGenerationMaxInputImages } from "./capabilities.js";
import type { ImageGenerationProvider } from "./types.js";

function createProvider(): ImageGenerationProvider {
  return {
    id: "test",
    capabilities: {
      generate: {},
      edit: {
        enabled: true,
        maxInputImages: 1,
        maxInputImagesByModel: {
          "family/pro/edit": 12,
        },
        maxInputImagesByModelPrefix: {
          family: 5,
          "family/pro": 10,
        },
      },
    },
    async generateImage() {
      throw new Error("not used");
    },
  };
}

describe("resolveImageGenerationMaxInputImages", () => {
  it("prefers exact limits, then the longest prefix, then the provider default", () => {
    const provider = createProvider();

    expect(resolveImageGenerationMaxInputImages({ provider, model: "family/pro/edit" })).toBe(12);
    expect(resolveImageGenerationMaxInputImages({ provider, model: "family/pro/v2" })).toBe(10);
    expect(resolveImageGenerationMaxInputImages({ provider, model: "family/basic" })).toBe(5);
    expect(resolveImageGenerationMaxInputImages({ provider, model: "other" })).toBe(1);
  });
});