openclaw/src/model-catalog/authority.test.ts

import { describe, expect, it } from "vitest";
import { mergeModelCatalogRowsByAuthority } from "./index.js";
import type { ModelCatalogSource, NormalizedModelCatalogRow } from "./index.js";

function row(source: ModelCatalogSource, name: string): NormalizedModelCatalogRow {
  return {
    provider: "moonshot",
    id: "kimi-k2.6",
    ref: "moonshot/kimi-k2.6",
    mergeKey: "moonshot::kimi-k2.6",
    name,
    source,
    input: ["text"],
    reasoning: false,
    status: source === "provider-index" ? "preview" : "available",
  };
}

function expectSelectedRow(
  rows: NormalizedModelCatalogRow[],
  expected: { name: string; source: ModelCatalogSource },
) {
  expect(rows).toHaveLength(1);
  const [selected] = rows;
  if (!selected) {
    throw new Error("missing selected catalog row");
  }
  expect(selected.name).toBe(expected.name);
  expect(selected.source).toBe(expected.source);
}

describe("model catalog authority", () => {
  it("keeps user config above manifest, cache, and provider-index preview rows", () => {
    expectSelectedRow(
      mergeModelCatalogRowsByAuthority([
        row("provider-index", "Preview"),
        row("cache", "Cached"),
        row("manifest", "Manifest"),
        row("config", "Configured"),
      ]),
      { name: "Configured", source: "config" },
    );
  });

  it("keeps installed manifest rows above cache and provider-index preview rows", () => {
    expectSelectedRow(
      mergeModelCatalogRowsByAuthority([
        row("provider-index", "Preview"),
        row("runtime-refresh", "Refreshed"),
        row("cache", "Cached"),
        row("manifest", "Manifest"),
      ]),
      { name: "Manifest", source: "manifest" },
    );
  });

  it("uses cache rows above provider-index preview rows", () => {
    expectSelectedRow(
      mergeModelCatalogRowsByAuthority([row("provider-index", "Preview"), row("cache", "Cached")]),
      { name: "Cached", source: "cache" },
    );
  });

  it("uses provider-index preview rows when no higher-authority row exists", () => {
    expectSelectedRow(mergeModelCatalogRowsByAuthority([row("provider-index", "Preview")]), {
      name: "Preview",
      source: "provider-index",
    });
  });
});