File size: 3,080 Bytes
e249c6d | 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 | import { expect, it, vi } from "vitest";
import { resolveVisibleActiveSessionRunState } from "../../gateway/server-methods/session-active-runs.js";
import { registerAgentRunCapacityWait } from "../../infra/agent-run-capacity-wait.js";
import {
clearAgentRunContext,
getAgentRunLifecycleGeneration,
} from "../../infra/agent-run-registry.js";
import { prepareEmbeddedSessionState } from "./session-preparation.js";
vi.mock("../embedded-agent-runner/runs.js", () => ({
resolveEmbeddedAgentSessionProgressState: () => undefined,
}));
vi.mock("../subagents/registry/subagent-registry-read.js", () => ({
getLatestLiveSubagentRunByChildSessionKey: () => undefined,
isSubagentRunQueued: () => false,
}));
vi.mock("../../sessions/session-state-events.js", () => ({
recordSessionHumanDirectMessage: vi.fn(),
}));
vi.mock("../../skills/discovery/agent-filter.js", () => ({
resolveEffectiveAgentSkillFilter: () => undefined,
}));
vi.mock("./attempt-execution.shared.js", () => ({ persistAgentSession: vi.fn() }));
vi.mock("./run-context.js", () => ({ resolveAgentRunContext: () => ({}) }));
vi.mock("./runtime-loaders.js", () => ({
loadSkillsRuntime: async () => ({
getRemoteSkillEligibility: () => ({}),
resolveReusableWorkspaceSkillSnapshot: () => ({ snapshot: undefined, shouldRefresh: false }),
}),
loadExecDefaultsRuntime: async () => ({
resolveNodeExecEligibility: () => ({ canExec: false }),
}),
}));
it.each([false, true])(
"projects command startup and capacity waits without a chat controller (internal=%s)",
async (internal) => {
const runId = "command-activity";
const sessionKey = "agent:main:command-activity";
const sessionId = "command-activity-session";
const lifecycleGeneration = getAgentRunLifecycleGeneration();
const state = () =>
resolveVisibleActiveSessionRunState({
context: {},
requestedKey: sessionKey,
canonicalKey: sessionKey,
sessionId,
agentId: "main",
});
let releaseWait: (() => void) | undefined;
try {
await prepareEmbeddedSessionState({
cfg: {},
opts: { message: "hello" },
sessionKey,
sessionId,
storePath: "/unused/command.sqlite",
sessionAgentId: "main",
lifecycleGeneration,
runId,
workspaceDir: "/workspace",
executionWorkspaceDir: "/workspace",
watchSkills: false,
isNewSession: false,
isSubagentLaneTurn: false,
suppressVisibleSessionEffects: internal,
sessionStateActor: { actorType: "human" },
});
const active = internal ? { active: false, runIds: [] } : { active: true };
expect(state()).toEqual(active);
releaseWait = registerAgentRunCapacityWait(runId, lifecycleGeneration);
expect(state()).toEqual(internal ? active : { active: true, status: "queued" });
releaseWait?.();
expect(state()).toEqual(active);
} finally {
releaseWait?.();
clearAgentRunContext(runId);
}
expect(state()).toEqual({ active: false, runIds: [] });
},
);
|