Extract executor retry decision rules

src/message-agent-executor-rules.test.ts

@@ -0,0 +1,115 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+vi.mock('./agent-error-detection.js', () => ({
+  classifyRotationTrigger: vi.fn(() => ({
+    shouldRetry: false,
+    reason: '',
+  })),
+}));
+
+vi.mock('./codex-token-rotation.js', () => ({
+  detectCodexRotationTrigger: vi.fn(() => ({
+    shouldRotate: false,
+    reason: '',
+  })),
+}));
+
+import { classifyRotationTrigger } from './agent-error-detection.js';
+import { detectCodexRotationTrigger } from './codex-token-rotation.js';
+import {
+  isRetryableClaudeSessionFailureAttempt,
+  resolveClaudeRetryTrigger,
+  resolveCodexRetryTrigger,
+  resolvePairedFollowUpQueueAction,
+} from './message-agent-executor-rules.js';
+
+describe('message-agent-executor-rules', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('resolves a Claude retry trigger from streamed trigger state', () => {
+    expect(
+      resolveClaudeRetryTrigger({
+        canRetryClaudeCredentials: true,
+        provider: 'claude',
+        attempt: {
+          sawOutput: false,
+          streamedTriggerReason: { reason: 'auth-expired', retryAfterMs: 5000 },
+        },
+      }),
+    ).toEqual({
+      reason: 'auth-expired',
+      retryAfterMs: 5000,
+    });
+  });
+
+  it('returns null for Claude retry when output was already visible', () => {
+    expect(
+      resolveClaudeRetryTrigger({
+        canRetryClaudeCredentials: true,
+        provider: 'claude',
+        attempt: { sawOutput: true },
+        fallbackMessage: '429 rate limit',
+      }),
+    ).toBeNull();
+    expect(classifyRotationTrigger).not.toHaveBeenCalled();
+  });
+
+  it('resolves a Codex retry trigger from the codex detector', () => {
+    vi.mocked(detectCodexRotationTrigger).mockReturnValue({
+      shouldRotate: true,
+      reason: 'auth-expired',
+    });
+
+    expect(
+      resolveCodexRetryTrigger({
+        canRetryCodex: true,
+        attempt: {},
+        rotationMessage: '401 oauth token has expired',
+      }),
+    ).toEqual({ reason: 'auth-expired' });
+  });
+
+  it('detects retryable Claude session failures from either flag or classifier', () => {
+    expect(
+      isRetryableClaudeSessionFailureAttempt({
+        attempt: {
+          sawOutput: false,
+          retryableSessionFailureDetected: true,
+        },
+        isClaudeCodeAgent: true,
+        provider: 'claude',
+        shouldRetryFreshSessionOnAgentFailure: vi.fn(() => false),
+      }),
+    ).toBe(true);
+
+    const shouldRetryFreshSessionOnAgentFailure = vi.fn(() => true);
+    expect(
+      isRetryableClaudeSessionFailureAttempt({
+        attempt: {
+          sawOutput: false,
+          error: new Error('stale session'),
+        },
+        isClaudeCodeAgent: true,
+        provider: 'claude',
+        shouldRetryFreshSessionOnAgentFailure,
+      }),
+    ).toBe(true);
+    expect(shouldRetryFreshSessionOnAgentFailure).toHaveBeenCalledWith({
+      result: null,
+      error: 'stale session',
+    });
+  });
+
+  it('resolves pending reviewer follow-up requeue from task state', () => {
+    expect(
+      resolvePairedFollowUpQueueAction({
+        completedRole: 'reviewer',
+        executionStatus: 'failed',
+        sawOutput: false,
+        taskStatus: 'review_ready',
+      }),
+    ).toBe('pending');
+  });
+});