File size: 3,575 Bytes
fd2c364 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | import { describe, it, expect, vi } from "vitest";
import { render, screen } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import React from "react";
import { adaptSystemMessage } from "#/utils/system-message-adapter";
import { EventState } from "#/stores/use-event-store";
import { SystemMessageModal } from "#/components/features/conversation-panel/system-message-modal";
import { ToolsContextMenu } from "#/components/features/controls/tools-context-menu";
const v1Event: EventState["events"] = [
{
id: "v1-id",
timestamp: "2025-12-30T12:00:00Z",
source: "agent",
system_prompt: {
type: "text",
text: "v1 prompt",
},
tools: [
{
type: "function",
function: {
name: "bash",
description: "Execute bash",
parameters: {},
},
},
],
},
];
const adaptedResult = adaptSystemMessage(v1Event);
vi.mock("#/hooks/query/use-active-conversation", () => ({
useActiveConversation: () => ({ data: { conversation_version: "V1" } }),
}));
vi.mock("#/hooks/use-user-providers", () => ({
useUserProviders: () => ({ providers: ["test"] }),
}));
describe("SystemMessage UI Rendering", () => {
it("should render the 'Show Agent Tools' button in the context menu", () => {
render(
<ToolsContextMenu
onClose={() => {}}
onShowSkills={() => {}}
onShowPlugins={() => {}}
onShowHooks={() => {}}
onShowAgentTools={() => {}}
/>,
);
expect(screen.getByTestId("show-agent-tools-button")).toBeInTheDocument();
});
it("should display the adapted v1 system prompt content correctly", () => {
render(
<SystemMessageModal
isOpen
onClose={() => {}}
systemMessage={adaptedResult}
/>,
);
const messageElement = screen.getByText("v1 prompt");
expect(messageElement).toBeDefined();
expect(messageElement).toBeVisible();
});
});
describe("ToolsContextMenu - Switch agent profile", () => {
it("hides the Switch agent profile item unless the caller enables it", () => {
// The gate (pre-start only + profiles available) is owned by
// ChatInputActions; without the prop the item — and its provider-backed
// submenu content — must not mount.
render(
<ToolsContextMenu
onClose={() => {}}
onShowSkills={() => {}}
onShowPlugins={() => {}}
onShowHooks={() => {}}
onShowAgentTools={() => {}}
/>,
);
expect(
screen.queryByTestId("switch-agent-profile-button"),
).not.toBeInTheDocument();
});
});
describe("ToolsContextMenu - Show Plugins", () => {
it("renders the Show Plugins item and calls onShowPlugins when clicked", async () => {
const onShowPlugins = vi.fn();
render(
<ToolsContextMenu
onClose={() => {}}
onShowSkills={() => {}}
onShowPlugins={onShowPlugins}
onShowHooks={() => {}}
onShowAgentTools={() => {}}
shouldShowPlugins
/>,
);
await userEvent.click(screen.getByTestId("show-plugins-button"));
expect(onShowPlugins).toHaveBeenCalledTimes(1);
});
it("hides the Show Plugins item when the conversation has no attached plugins", () => {
render(
<ToolsContextMenu
onClose={() => {}}
onShowSkills={() => {}}
onShowPlugins={() => {}}
onShowHooks={() => {}}
onShowAgentTools={() => {}}
/>,
);
expect(screen.queryByTestId("show-plugins-button")).not.toBeInTheDocument();
});
});
|