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

describe("byteplus 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: "byteplus",
        id: BYTEPLUS_MODEL_CATALOG[0].id,
        name: BYTEPLUS_MODEL_CATALOG[0].name,
        reasoning: BYTEPLUS_MODEL_CATALOG[0].reasoning,
        input: [...BYTEPLUS_MODEL_CATALOG[0].input],
        contextWindow: BYTEPLUS_MODEL_CATALOG[0].contextWindow,
      }),
    );
    expect(entries).toContainEqual(
      expect.objectContaining({
        provider: "byteplus-plan",
        id: BYTEPLUS_CODING_MODEL_CATALOG[0].id,
        name: BYTEPLUS_CODING_MODEL_CATALOG[0].name,
        reasoning: BYTEPLUS_CODING_MODEL_CATALOG[0].reasoning,
        input: [...BYTEPLUS_CODING_MODEL_CATALOG[0].input],
        contextWindow: BYTEPLUS_CODING_MODEL_CATALOG[0].contextWindow,
      }),
    );
  });
});