Luke The Dev
95 lines
2.5 KiB
TypeScript
95 lines
2.5 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import type { PendingExecApproval } from "@/features/agents/approvals/types";
|
|
import { resolveExecApprovalFollowUpIntent } from "@/features/agents/approvals/execApprovalLifecycleWorkflow";
|
|
import type { AgentState } from "@/features/agents/state/store";
|
|
|
|
const createAgent = (agentId: string, sessionKey: string): AgentState => ({
|
|
agentId,
|
|
name: agentId,
|
|
sessionKey,
|
|
status: "idle",
|
|
sessionCreated: true,
|
|
awaitingUserInput: false,
|
|
hasUnseenActivity: false,
|
|
outputLines: [],
|
|
lastResult: null,
|
|
lastDiff: null,
|
|
runId: null,
|
|
runStartedAt: null,
|
|
streamText: null,
|
|
thinkingTrace: null,
|
|
latestOverride: null,
|
|
latestOverrideKind: null,
|
|
lastAssistantMessageAt: null,
|
|
lastActivityAt: null,
|
|
latestPreview: null,
|
|
lastUserMessage: null,
|
|
draft: "",
|
|
sessionSettingsSynced: true,
|
|
historyLoadedAt: null,
|
|
historyFetchLimit: null,
|
|
historyFetchedCount: null,
|
|
historyMaybeTruncated: false,
|
|
toolCallingEnabled: true,
|
|
showThinkingTraces: true,
|
|
model: "openai/gpt-5",
|
|
thinkingLevel: "medium",
|
|
avatarSeed: agentId,
|
|
avatarUrl: null,
|
|
});
|
|
|
|
const createApproval = (): PendingExecApproval => ({
|
|
id: "approval-1",
|
|
agentId: null,
|
|
sessionKey: "agent:agent-1:main",
|
|
command: "npm run test",
|
|
cwd: "/repo",
|
|
host: "gateway",
|
|
security: "allowlist",
|
|
ask: "always",
|
|
resolvedPath: "/usr/bin/npm",
|
|
createdAtMs: 1,
|
|
expiresAtMs: 2,
|
|
resolving: false,
|
|
error: null,
|
|
});
|
|
|
|
describe("lifecycleControllerWorkflow integration", () => {
|
|
it("allow-once and allow-always still trigger follow-up message send once", () => {
|
|
const approval = createApproval();
|
|
const agents = [createAgent("agent-1", "agent:agent-1:main")];
|
|
let sendCount = 0;
|
|
|
|
for (const decision of ["allow-once", "allow-always"] as const) {
|
|
const intent = resolveExecApprovalFollowUpIntent({
|
|
decision,
|
|
approval,
|
|
agents,
|
|
followUpMessage: "An exec approval was granted.",
|
|
});
|
|
if (intent.shouldSend) {
|
|
sendCount += 1;
|
|
}
|
|
}
|
|
|
|
expect(sendCount).toBe(2);
|
|
});
|
|
|
|
it("deny decision does not trigger follow-up message send", () => {
|
|
const intent = resolveExecApprovalFollowUpIntent({
|
|
decision: "deny",
|
|
approval: createApproval(),
|
|
agents: [createAgent("agent-1", "agent:agent-1:main")],
|
|
followUpMessage: "An exec approval was granted.",
|
|
});
|
|
|
|
expect(intent).toEqual({
|
|
shouldSend: false,
|
|
agentId: null,
|
|
sessionKey: null,
|
|
message: null,
|
|
});
|
|
});
|
|
});
|