Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 2 additions & 7 deletions package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

80 changes: 34 additions & 46 deletions src/__tests__/CodexACPAgent/CodexAcpClient.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,17 @@
import {describe, expect, it, vi, beforeEach} from 'vitest';
import type {CodexAuthRequest} from "../../CodexAuthMethod";
import type * as acp from "@agentclientprotocol/sdk";
import {createTestFixture, createCodexMockTestFixture, createTestSessionState, type TestFixture} from "../acp-test-utils";
import {
createTestFixture,
createCodexMockTestFixture,
createTestModel,
createTestSessionState,
createTestThread,
createTestThreadStartResponse,
createTestTurn,
createTestTurnCompletedNotification,
type TestFixture
} from "../acp-test-utils";
import type {ServerNotification} from "../../app-server";
import type {SessionState} from "../../CodexAcpServer";
import {AgentMode} from "../../AgentMode";
Expand Down Expand Up @@ -143,32 +153,13 @@ describe('ACP server test', { timeout: 40_000 }, () => {
const codexAppServerClient = mockFixture.getCodexAppServerClient();

const listSkillsSpy = vi.spyOn(codexAppServerClient, "listSkills").mockResolvedValue({ data: [] });
const threadStartSpy = vi.spyOn(codexAppServerClient, "threadStart").mockResolvedValue({
thread: { id: "thread-id" } as any,
model: "gpt-5",
modelProvider: "openai",
cwd: "/workspace",
approvalPolicy: "on-request",
sandbox: "workspace-write",
reasoningEffort: "medium",
} as any);
const threadStartSpy = vi.spyOn(codexAppServerClient, "threadStart").mockResolvedValue(
createTestThreadStartResponse({
thread: createTestThread({ id: "thread-id" }),
})
);
vi.spyOn(codexAppServerClient, "listModels").mockResolvedValue({
data: [{
id: "gpt-5",
model: "gpt-5",
upgrade: null,
upgradeInfo: null,
availabilityNux: null,
displayName: "gpt-5",
description: "test model",
hidden: false,
supportedReasoningEfforts: [{ reasoningEffort: "medium", description: "balanced" }],
defaultReasoningEffort: "medium",
inputModalities: ["text"],
supportsPersonality: false,
additionalSpeedTiers: [],
isDefault: true
}],
data: [createTestModel({ displayName: "gpt-5" })],
nextCursor: null
});

Expand Down Expand Up @@ -233,24 +224,19 @@ describe('ACP server test', { timeout: 40_000 }, () => {
const codexAppServerClient = mockFixture.getCodexAppServerClient();

vi.spyOn(codexAcpAgent, "checkAuthorization").mockResolvedValue(undefined);
vi.spyOn(codexAppServerClient, "threadStart").mockResolvedValue({
thread: { id: "thread-id" } as any,
model: "gpt-5",
reasoningEffort: "medium",
} as any);
vi.spyOn(codexAppServerClient, "threadStart").mockResolvedValue(
createTestThreadStartResponse({
thread: createTestThread({ id: "thread-id" }),
})
);
vi.spyOn(codexAppServerClient, "listModels").mockResolvedValue({
data: [{
id: "gpt-5",
name: "GPT-5",
inputModalities: ["text"],
supportedReasoningEfforts: [],
}],
hasMore: false,
} as any);
data: [createTestModel({ supportedReasoningEfforts: [] })],
nextCursor: null,
});
vi.spyOn(codexAppServerClient, "accountRead").mockResolvedValue({
requiresOpenaiAuth: false,
account: null,
} as any);
});
vi.spyOn(codexAppServerClient, "listSkills").mockResolvedValue({ data: [] });
const mcpServer = {
name: "broken-mcp",
Expand Down Expand Up @@ -287,12 +273,14 @@ describe('ACP server test', { timeout: 40_000 }, () => {

const listSkillsSpy = vi.spyOn(codexAppServerClient, "listSkills").mockResolvedValue({ data: [] });
const turnStartSpy = vi.spyOn(codexAppServerClient, "turnStart").mockResolvedValue({
turn: { id: "turn-id", items: [], status: "inProgress", error: null }
} as any);
vi.spyOn(codexAppServerClient, "awaitTurnCompleted").mockResolvedValue({
threadId: "session-id",
turn: { id: "turn-id", items: [], status: "completed", error: null }
} as any);
turn: createTestTurn()
});
vi.spyOn(codexAppServerClient, "awaitTurnCompleted").mockResolvedValue(
createTestTurnCompletedNotification({
threadId: "session-id",
turn: createTestTurn({ status: "completed" })
})
);

vi.spyOn(codexAcpAgent, "getSessionState").mockReturnValue(createTestSessionState({
sessionId: "session-id",
Expand Down
109 changes: 104 additions & 5 deletions src/__tests__/acp-test-utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,13 @@ import {startCodexConnection} from "../CodexJsonRpcConnection";
import {CodexAcpServer, type SessionState} from "../CodexAcpServer";
import type {AgentSideConnection, RequestPermissionResponse} from "@agentclientprotocol/sdk";
import type {ServerNotification} from "../app-server";
import type {
Model,
Thread,
ThreadStartResponse,
Turn,
TurnCompletedNotification
} from "../app-server/v2";
import type {MessageConnection} from "vscode-jsonrpc/node";
import path from "node:path";
import fs from "node:fs";
Expand Down Expand Up @@ -315,6 +322,96 @@ export function createTestSessionState(overrides?: Partial<SessionState>): Sessi
};
}

export function createTestModel(overrides?: Partial<Model>): Model {
return {
id: "gpt-5",
model: "gpt-5",
upgrade: null,
upgradeInfo: null,
availabilityNux: null,
displayName: "GPT-5",
description: "test model",
hidden: false,
supportedReasoningEfforts: [{ reasoningEffort: "medium", description: "balanced" }],
defaultReasoningEffort: "medium",
inputModalities: ["text"],
supportsPersonality: false,
additionalSpeedTiers: [],
isDefault: true,
...overrides,
};
}

export function createTestTurn(overrides?: Partial<Turn>): Turn {
return {
id: "turn-id",
items: [],
status: "inProgress",
error: null,
startedAt: null,
completedAt: null,
durationMs: null,
...overrides,
};
}

export function createTestThread(overrides?: Partial<Thread>): Thread {
return {
id: "thread-id",
forkedFromId: null,
preview: "",
ephemeral: false,
modelProvider: "openai",
createdAt: 0,
updatedAt: 0,
status: { type: "idle" },
path: null,
cwd: "/workspace",
cliVersion: "0.0.0-test",
source: "appServer",
agentNickname: null,
agentRole: null,
gitInfo: null,
name: null,
turns: [],
...overrides,
};
}

export function createTestThreadStartResponse(overrides?: Partial<ThreadStartResponse>): ThreadStartResponse {
return {
thread: createTestThread(),
model: "gpt-5",
modelProvider: "openai",
serviceTier: null,
cwd: "/workspace",
instructionSources: [],
approvalPolicy: "on-request",
approvalsReviewer: "user",
sandbox: {
type: "workspaceWrite",
writableRoots: [],
readOnlyAccess: { type: "fullAccess" },
networkAccess: false,
excludeTmpdirEnvVar: false,
excludeSlashTmp: false,
},
permissionProfile: null,
reasoningEffort: "medium",
...overrides,
};
}

export function createTestTurnCompletedNotification(
overrides?: Partial<TurnCompletedNotification>
): TurnCompletedNotification {
return {
threadId: "session-id",
turn: createTestTurn({ status: "completed" }),
...overrides,
};
}

export async function setupPromptAndSendNotifications(
fixture: CodexMockTestFixture,
sessionId: string,
Expand All @@ -323,15 +420,17 @@ export async function setupPromptAndSendNotifications(
): Promise<void> {
const codexAcpAgent = fixture.getCodexAcpAgent();
const codexAppServerClient = fixture.getCodexAppServerClient();
const turn = { id: "turn-id", items: [], status: "inProgress" as const, error: null };
const turn = createTestTurn();

codexAppServerClient.turnStart = vi.fn().mockResolvedValue({
turn,
});
codexAppServerClient.awaitTurnCompleted = vi.fn().mockResolvedValue({
threadId: sessionId,
turn: { ...turn, status: "completed" },
});
codexAppServerClient.awaitTurnCompleted = vi.fn().mockResolvedValue(
createTestTurnCompletedNotification({
threadId: sessionId,
turn: createTestTurn({ ...turn, status: "completed" }),
})
);

vi.spyOn(codexAcpAgent, "getSessionState").mockReturnValue(sessionState);

Expand Down
Loading