test(ui-automation): Stabilize key sequence snapshot polling

Inject deterministic post-action snapshot timing into the key sequence executor test. This keeps the exact polling assertion stable under CI scheduler jitter.

Co-Authored-By: OpenAI Codex <noreply@openai.com>

Author: cameroncooke

src/mcp/tools/ui-automation/__tests__/key_sequence.test.ts

@@ -12,6 +12,17 @@ import {
   simulatorId,
 } from './ui-action-test-helpers.ts';
 
+function createImmediatePostActionTiming() {
+  let nowMs = 0;
+
+  return {
+    now: () => nowMs,
+    sleep: async (durationMs: number) => {
+      nowMs += durationMs;
+    },
+  };
+}
+
 describe('Key Sequence Tool', () => {
   beforeEach(() => {
     sessionStore.clear();
@@ -194,7 +205,12 @@ describe('Key Sequence Tool', () => {
   describe('Handler Behavior (Complete Literal Returns)', () => {
     it('captures a fresh runtime snapshot after a successful key sequence', async () => {
       const { calls, executor } = createTrackingExecutor();
-      const executeKeySequence = createKeySequenceExecutor(executor, createMockAxeHelpers());
+      const executeKeySequence = createKeySequenceExecutor(
+        executor,
+        createMockAxeHelpers(),
+        undefined,
+        createImmediatePostActionTiming(),
+      );
 
       const result = await executeKeySequence({ simulatorId, keyCodes: [40, 42, 44] });
 

src/mcp/tools/ui-automation/key_sequence.ts

@@ -22,7 +22,10 @@ import {
   clearRuntimeSnapshot,
   withSimulatorUiAutomationTransaction,
 } from './shared/snapshot-ui-state.ts';
-import { captureRuntimeSnapshotAfterActionSafely } from './shared/post-action-snapshot.ts';
+import {
+  captureRuntimeSnapshotAfterActionSafely,
+  type PostActionSnapshotTiming,
+} from './shared/post-action-snapshot.ts';
 import type { AxeHelpers } from './shared/axe-command.ts';
 import type { NonStreamingExecutor } from '../../../types/tool-execution.ts';
 import type { UiActionResultDomainResult } from '../../../types/domain-results.ts';
@@ -57,6 +60,7 @@ export function createKeySequenceExecutor(
   executor: CommandExecutor,
   axeHelpers: AxeHelpers = defaultAxeHelpers,
   debuggerManager: DebuggerManager = getDefaultDebuggerManager(),
+  postActionSnapshotTiming?: PostActionSnapshotTiming,
 ): NonStreamingExecutor<KeySequenceParams, KeySequenceResult> {
   return async (params) =>
     withSimulatorUiAutomationTransaction(params.simulatorId, async () => {
@@ -91,6 +95,7 @@ export function createKeySequenceExecutor(
           simulatorId,
           executor,
           axeHelpers,
+          timing: postActionSnapshotTiming,
         });
         return createUiActionSuccessResult(
           action,