import { describe, expect, it } from "vitest";
import { buildQaImageGenerationConfigPatch } from "./image-generation.js";

describe("QA provider image generation config", () => {
  it("uses the selected mock provider for mock-openai image generation", () => {
    const patch = buildQaImageGenerationConfigPatch({
      providerMode: "mock-openai",
      providerBaseUrl: "http://127.0.0.1:44080/v1",
      requiredPluginIds: ["qa-channel"],
    });

    expect(patch.plugins.allow).toEqual(["memory-core", "qa-channel"]);
    expect(patch.agents.defaults.imageGenerationModel.primary).toBe("mock-openai/gpt-image-1");
    expect(patch.models?.providers["mock-openai"]?.baseUrl).toBe("http://127.0.0.1:44080/v1");
  });

  it("uses the selected mock provider for AIMock image generation", () => {
    const patch = buildQaImageGenerationConfigPatch({
      providerMode: "aimock",
      providerBaseUrl: "http://127.0.0.1:45080/v1",
      requiredPluginIds: [],
    });

    expect(patch.agents.defaults.imageGenerationModel.primary).toBe("aimock/gpt-image-1");
    expect(patch.models?.providers.aimock?.baseUrl).toBe("http://127.0.0.1:45080/v1");
    expect(patch.models?.providers["mock-openai"]).toBeUndefined();
  });

  it("enables the live image provider plugin without replacing live model config", () => {
    const patch = buildQaImageGenerationConfigPatch({
      providerMode: "live-frontier",
      requiredPluginIds: ["qa-channel"],
    });

    expect(patch.plugins).toEqual({
      allow: ["memory-core", "openai", "qa-channel"],
      entries: {
        openai: {
          enabled: true,
        },
      },
    });
    expect(patch.agents.defaults.imageGenerationModel.primary).toBe("openai/gpt-image-1");
    expect(patch).not.toHaveProperty("models");
  });
});