Carlos Florêncio
86 lines
2.7 KiB
TypeScript
86 lines
2.7 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import type { UtilityAgent } from "@/lib/api/domains/utility-api";
|
|
import { USE_DEFAULT } from "./utility-sections";
|
|
import {
|
|
isUtilityAgentDirty,
|
|
mergeRefreshedUtilityAgents,
|
|
replaceCustomUtilityAgents,
|
|
updateBuiltinProfileDraft,
|
|
} from "./utility-agents-section";
|
|
|
|
const DRAFT_MODEL = "draft-model";
|
|
|
|
function agent(id: string, builtin: boolean, model: string): UtilityAgent {
|
|
return {
|
|
id,
|
|
name: id,
|
|
description: "",
|
|
builtin,
|
|
enabled: true,
|
|
agent_id: "agent-1",
|
|
model,
|
|
prompt: "",
|
|
created_at: "",
|
|
updated_at: "",
|
|
};
|
|
}
|
|
|
|
describe("utility agent helpers", () => {
|
|
it("compares only draftable built-in fields", () => {
|
|
const baseline = agent("commit", true, "saved-model");
|
|
|
|
expect(isUtilityAgentDirty({ ...baseline, description: "refreshed" }, baseline)).toBe(false);
|
|
expect(isUtilityAgentDirty({ ...baseline, enabled: false }, baseline)).toBe(true);
|
|
expect(isUtilityAgentDirty({ ...baseline, model: DRAFT_MODEL }, baseline)).toBe(true);
|
|
});
|
|
|
|
it("refreshes immediate custom resources without replacing built-in drafts", () => {
|
|
const builtinDraft = agent("commit", true, DRAFT_MODEL);
|
|
const refreshedCustom = agent("custom-new", false, "saved-model");
|
|
|
|
expect(
|
|
replaceCustomUtilityAgents(
|
|
[builtinDraft, agent("custom-old", false, "old-model")],
|
|
[refreshedCustom],
|
|
),
|
|
).toEqual([builtinDraft, refreshedCustom]);
|
|
});
|
|
|
|
it("refreshes saved builtins while preserving unsaved model overrides", () => {
|
|
const baseline = agent("commit", true, "saved-model");
|
|
const draft = { ...baseline, model: DRAFT_MODEL };
|
|
const refreshed = { ...baseline, prompt: "updated in dialog", model: "dialog-model" };
|
|
|
|
expect(mergeRefreshedUtilityAgents([draft], [baseline], [refreshed])).toEqual([
|
|
{ ...refreshed, model: DRAFT_MODEL },
|
|
]);
|
|
});
|
|
});
|
|
|
|
describe("updateBuiltinProfileDraft", () => {
|
|
it("treats the picker fallback value as an inherited binding", () => {
|
|
const base = {
|
|
...agent("commit", true, ""),
|
|
enabled: false,
|
|
agent_profile_id: "deleted-profile",
|
|
profile_binding_state: "unconfigured",
|
|
};
|
|
|
|
for (const pickerValue of ["", USE_DEFAULT]) {
|
|
const draft = updateBuiltinProfileDraft(base, pickerValue);
|
|
|
|
expect(draft.agent_profile_id).toBe("");
|
|
expect(draft.profile_binding_state).toBe("inherit");
|
|
expect(draft.enabled).toBe(true);
|
|
}
|
|
});
|
|
|
|
it("keeps a concrete profile selection explicit", () => {
|
|
const draft = updateBuiltinProfileDraft(agent("commit", true, ""), "profile-1");
|
|
|
|
expect(draft.agent_profile_id).toBe("profile-1");
|
|
expect(draft.profile_binding_state).toBe("explicit");
|
|
expect(draft.enabled).toBe(true);
|
|
});
|
|
});
|