import { describe, expect, it } from "vitest";
import { registerSingleProviderPlugin } from "../../test/helpers/plugins/plugin-registration.js";
import plugin from "./index.js";
import { DOUBAO_CODING_MODEL_CATALOG, DOUBAO_MODEL_CATALOG } from "./models.js";

describe("volcengine plugin", () => {
  it("augments the catalog with bundled standard and plan models", async () => {
    const provider = await registerSingleProviderPlugin(plugin);
    const entries = await provider.augmentModelCatalog?.({
      env: process.env,
      entries: [],
    } as never);

    expect(entries).toContainEqual(
      expect.objectContaining({
        provider: "volcengine",
        id: DOUBAO_MODEL_CATALOG[0].id,
        name: DOUBAO_MODEL_CATALOG[0].name,
        reasoning: DOUBAO_MODEL_CATALOG[0].reasoning,
        input: [...DOUBAO_MODEL_CATALOG[0].input],
        contextWindow: DOUBAO_MODEL_CATALOG[0].contextWindow,
      }),
    );
    expect(entries).toContainEqual(
      expect.objectContaining({
        provider: "volcengine-plan",
        id: DOUBAO_CODING_MODEL_CATALOG[0].id,
        name: DOUBAO_CODING_MODEL_CATALOG[0].name,
        reasoning: DOUBAO_CODING_MODEL_CATALOG[0].reasoning,
        input: [...DOUBAO_CODING_MODEL_CATALOG[0].input],
        contextWindow: DOUBAO_CODING_MODEL_CATALOG[0].contextWindow,
      }),
    );
  });
});