Skip to content

Commit d24204b

Browse files
committed
Stabilize Windows transport agent probe
1 parent c7c846d commit d24204b

1 file changed

Lines changed: 12 additions & 3 deletions

File tree

tests/e2e/transport.spec.ts

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ async function observeWsTransport(page: Page): Promise<WsTransportBaseline> {
238238
await invokeRpc(page, 'agent_send', {
239239
workspaceId: workspace.workspaceId,
240240
sessionId,
241-
input: 'transport-agent',
241+
input: buildAgentProbeInput(workspace.target),
242242
appendNewline: true,
243243
});
244244
controlPlaneCommands.push('agent_send');
@@ -548,7 +548,10 @@ async function openWorkspace(page: Page): Promise<WorkspaceHandle> {
548548

549549
async function invokeRpc<T>(page: Page, command: string, payload: Record<string, unknown> = {}) {
550550
const response = await page.request.post(`/api/rpc/${command}`, { data: payload });
551-
expect(response.ok()).toBeTruthy();
551+
if (!response.ok()) {
552+
const body = await response.text();
553+
throw new Error(`RPC ${command} failed with ${response.status()}: ${body}`);
554+
}
552555
const body = await response.json();
553556
expect(body.ok).not.toBe(false);
554557
return body.data as T;
@@ -646,10 +649,16 @@ function buildTerminalProbeInput(target: WorkspaceHandle['target']) {
646649

647650
function buildAgentProbeCommand(target: WorkspaceHandle['target']) {
648651
return isWindowsNativeTarget(target)
649-
? 'powershell -NoLogo -NoProfile -Command "$line = [Console]::In.ReadLine(); if ($null -ne $line) { Write-Output $line }"'
652+
? 'cmd /Q /D /K'
650653
: 'cat';
651654
}
652655

656+
function buildAgentProbeInput(target: WorkspaceHandle['target']) {
657+
return isWindowsNativeTarget(target)
658+
? 'echo transport-agent'
659+
: 'transport-agent';
660+
}
661+
653662
function countTrackedSockets(tracker: TransportTrackerSnapshot, fragment: string) {
654663
return tracker.urls.filter((url) => url.includes(fragment)).length;
655664
}

0 commit comments

Comments
 (0)