recovery_orchestrator_test.go

package claudesdk

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func TestStripMediaContent(t *testing.T) {
	t.Parallel()

	content := Blocks(
		TextInput("keep"),
		ImageInput("image/png", "AQID"),
		PDFInput("JVBERi0xLjQK"),
		PathInput("/tmp/file.txt"),
	)

	stripped, changed := StripMediaContent(content)
	require.True(t, changed)

	blocks := stripped.Blocks()
	require.Len(t, blocks, 2)
	require.IsType(t, &TextBlock{}, blocks[0])
	require.IsType(t, &TextBlock{}, blocks[1])
}

func TestRecoveryOrchestratorPlanResult_StripsMedia(t *testing.T) {
	t.Parallel()

	orchestrator := NewRecoveryOrchestrator(RecoveryOrchestratorOptions{
		EnableMediaStrip: true,
	})

	content := Blocks(TextInput("describe"), ImageInput("image/png", "AQID"))
	result := &ResultMessage{
		IsError: true,
		Errors:  []string{"image too large for request"},
	}

	plan := orchestrator.PlanResult(result, content, &ClaudeAgentOptions{}, 1)
	require.True(t, plan.Retry)
	require.True(t, plan.ContentChanged)
	require.Equal(t, RecoveryActionStripMedia, plan.Action)
	require.Len(t, plan.Content.Blocks(), 1)
}

func TestRecoveryOrchestratorPlanResult_SwitchesToFallbackModel(t *testing.T) {
	t.Parallel()

	orchestrator := NewRecoveryOrchestrator(RecoveryOrchestratorOptions{
		EnableFallbackSwitch: true,
	})

	result := &ResultMessage{
		IsError: true,
		Errors:  []string{"invalid model requested"},
	}

	plan := orchestrator.PlanResult(result, Text("hello"), &ClaudeAgentOptions{
		Model:         "claude-opus-4-6",
		FallbackModel: "claude-opus-4-1",
	}, 1)
	require.True(t, plan.Retry)
	require.NotNil(t, plan.OverrideModel)
	require.Equal(t, "claude-opus-4-1", *plan.OverrideModel)
}

func TestRecoveryOrchestratorPlanResult_UsesRetryDelay(t *testing.T) {
	t.Parallel()

	orchestrator := NewRecoveryOrchestrator(RecoveryOrchestratorOptions{
		RetryPolicy: RetryPolicy{
			MaxRetries:                    3,
			BaseDelay:                     time.Second,
			MaxDelay:                      time.Second,
			JitterRatio:                   0,
			MaxConsecutiveOverloadRetries: 3,
		},
	})

	result := &ResultMessage{
		IsError: true,
		Errors:  []string{"dial tcp 1.2.3.4:443: connection reset by peer"},
	}

	plan := orchestrator.PlanResult(result, Text("hello"), &ClaudeAgentOptions{}, 1)
	require.True(t, plan.Retry)
	require.Equal(t, time.Second, plan.Delay)
}