73 lines
2.7 KiB
TypeScript
73 lines
2.7 KiB
TypeScript
import { describe, expect, test } from "vitest";
|
|
|
|
import { appManageSchema } from "@homarr/validation/app";
|
|
|
|
import { hasCompleteAssistantToolArguments, hasFailedAssistantToolArguments } from "./assistant-human-tool-status";
|
|
import { getAssistantAppFormValues } from "./assistant-human-tools";
|
|
|
|
describe("assistant human tool forms", () => {
|
|
test("keeps every human tool in its loading state while arguments are streaming", () => {
|
|
expect(hasCompleteAssistantToolArguments({ type: "running" })).toBe(false);
|
|
expect(hasCompleteAssistantToolArguments({ type: "complete" })).toBe(true);
|
|
expect(hasCompleteAssistantToolArguments({ type: "requires-action", reason: "interrupt" })).toBe(true);
|
|
expect(hasCompleteAssistantToolArguments({ type: "incomplete", reason: "error" })).toBe(false);
|
|
expect(hasFailedAssistantToolArguments({ type: "incomplete", reason: "error" })).toBe(true);
|
|
});
|
|
|
|
test("treats cancelled and truncated tool arguments as failed instead of loading forever", () => {
|
|
expect(hasFailedAssistantToolArguments({ type: "incomplete", reason: "cancelled" })).toBe(true);
|
|
expect(hasFailedAssistantToolArguments({ type: "incomplete", reason: "length" })).toBe(true);
|
|
expect(hasFailedAssistantToolArguments({ type: "running" })).toBe(false);
|
|
});
|
|
|
|
test("waits for streamed app arguments before initializing the form", () => {
|
|
expect(
|
|
getAssistantAppFormValues(
|
|
{ name: "Wiki" },
|
|
{
|
|
type: "running",
|
|
},
|
|
),
|
|
).toBeNull();
|
|
});
|
|
|
|
test("initializes the native app form with every completed tool argument", () => {
|
|
const values = getAssistantAppFormValues(
|
|
{
|
|
name: "Wikipedia",
|
|
description: "The Free Encyclopedia",
|
|
iconUrl: "https://cdn.jsdelivr.net/gh/homarr-labs/dashboard-icons/svg/wikipedia.svg",
|
|
href: "https://wikipedia.org",
|
|
pingUrl: "https://wikipedia.org",
|
|
},
|
|
{ type: "complete" },
|
|
);
|
|
|
|
expect(values).toEqual({
|
|
name: "Wikipedia",
|
|
description: "The Free Encyclopedia",
|
|
iconUrl: "https://cdn.jsdelivr.net/gh/homarr-labs/dashboard-icons/svg/wikipedia.svg",
|
|
href: "https://wikipedia.org",
|
|
pingUrl: "https://wikipedia.org",
|
|
});
|
|
expect(appManageSchema.safeParse(values).success).toBe(true);
|
|
});
|
|
|
|
test("drops unusable icon identifiers while preserving the other app fields", () => {
|
|
expect(
|
|
getAssistantAppFormValues(
|
|
{
|
|
name: "Wikipedia",
|
|
iconUrl: "wikipedia",
|
|
href: "https://wikipedia.org",
|
|
},
|
|
{ type: "complete" },
|
|
),
|
|
).toMatchObject({
|
|
name: "Wikipedia",
|
|
iconUrl: "",
|
|
href: "https://wikipedia.org",
|
|
});
|
|
});
|
|
});
|