@gotgenes/pi-permission-system 9.2.0 → 10.0.0

package/test/permission-ui-prompt.test.ts

@@ -0,0 +1,146 @@
+import { describe, expect, it } from "vitest";
+
+import {
+  buildDirectUiPrompt,
+  buildForwardedUiPrompt,
+  buildRpcUiPrompt,
+} from "#src/permission-ui-prompt";
+
+describe("buildDirectUiPrompt", () => {
+  it("maps a tool_call prompt to the tool surface and command value", () => {
+    expect(
+      buildDirectUiPrompt({
+        requestId: "req-1",
+        source: "tool_call",
+        agentName: "Explore",
+        message: "Allow git push?",
+        toolName: "bash",
+        command: "git push",
+      }),
+    ).toEqual({
+      requestId: "req-1",
+      source: "tool_call",
+      surface: "bash",
+      value: "git push",
+      agentName: "Explore",
+      message: "Allow git push?",
+      forwarding: null,
+    });
+  });
+
+  it("normalizes a skill prompt to the skill surface and skill-name value", () => {
+    expect(
+      buildDirectUiPrompt({
+        requestId: "req-2",
+        source: "skill_input",
+        agentName: null,
+        message: "Allow skill?",
+        skillName: "deploy-helper",
+      }),
+    ).toEqual({
+      requestId: "req-2",
+      source: "skill_input",
+      surface: "skill",
+      value: "deploy-helper",
+      agentName: null,
+      message: "Allow skill?",
+      forwarding: null,
+    });
+  });
+
+  it("derives value with command > path > target > skillName > toolName precedence", () => {
+    expect(
+      buildDirectUiPrompt({
+        requestId: "req-3",
+        source: "tool_call",
+        agentName: null,
+        message: "m",
+        toolName: "read",
+        path: "/etc/hosts",
+        target: "ignored",
+      }).value,
+    ).toBe("/etc/hosts");
+  });
+});
+
+describe("buildRpcUiPrompt", () => {
+  it("maps an RPC prompt to the rpc_prompt source with passthrough surface/value", () => {
+    expect(
+      buildRpcUiPrompt({
+        requestId: "req-rpc",
+        surface: "bash",
+        value: "git push",
+        agentName: "Worker",
+        message: "Allow git push?",
+      }),
+    ).toEqual({
+      requestId: "req-rpc",
+      source: "rpc_prompt",
+      surface: "bash",
+      value: "git push",
+      agentName: "Worker",
+      message: "Allow git push?",
+      forwarding: null,
+    });
+  });
+
+  it("defaults missing surface, value, and agentName to null", () => {
+    expect(
+      buildRpcUiPrompt({ requestId: "req-rpc-2", message: "Allow?" }),
+    ).toEqual({
+      requestId: "req-rpc-2",
+      source: "rpc_prompt",
+      surface: null,
+      value: null,
+      agentName: null,
+      message: "Allow?",
+      forwarding: null,
+    });
+  });
+});
+
+describe("buildForwardedUiPrompt", () => {
+  it("populates forwarding context and carries the original source/surface/value", () => {
+    expect(
+      buildForwardedUiPrompt({
+        requestId: "req-fwd",
+        message: "Subagent 'Explore' requested permission.\n\nAllow git push?",
+        requesterAgentName: "Explore",
+        requesterSessionId: "child-session",
+        source: "tool_call",
+        surface: "bash",
+        value: "git push",
+      }),
+    ).toEqual({
+      requestId: "req-fwd",
+      source: "tool_call",
+      surface: "bash",
+      value: "git push",
+      agentName: "Explore",
+      message: "Subagent 'Explore' requested permission.\n\nAllow git push?",
+      forwarding: {
+        requesterAgentName: "Explore",
+        requesterSessionId: "child-session",
+      },
+    });
+  });
+
+  it("falls back to source tool_call with null surface/value when the request omits them", () => {
+    expect(
+      buildForwardedUiPrompt({
+        requestId: "req-fwd-old",
+        message: "Allow?",
+        requesterAgentName: null,
+        requesterSessionId: null,
+      }),
+    ).toEqual({
+      requestId: "req-fwd-old",
+      source: "tool_call",
+      surface: null,
+      value: null,
+      agentName: null,
+      message: "Allow?",
+      forwarding: { requesterAgentName: null, requesterSessionId: null },
+    });
+  });
+});