internal/util/claude_model_test.go

Rama main · réplica de lectura del repositorio original.

package util

import "testing"

func TestIsClaudeThinkingModel(t *testing.T) {
	tests := []struct {
		name     string
		model    string
		expected bool
	}{
		// Claude thinking models - should return true
		{"claude-sonnet-4-5-thinking", "claude-sonnet-4-5-thinking", true},
		{"claude-opus-4-5-thinking", "claude-opus-4-5-thinking", true},
		{"claude-opus-4-6-thinking", "claude-opus-4-6-thinking", true},
		{"Claude-Sonnet-Thinking uppercase", "Claude-Sonnet-4-5-Thinking", true},
		{"claude thinking mixed case", "Claude-THINKING-Model", true},

		// Non-thinking Claude models - should return false
		{"claude-sonnet-4-5 (no thinking)", "claude-sonnet-4-5", false},
		{"claude-opus-4-5 (no thinking)", "claude-opus-4-5", false},
		{"claude-3-5-sonnet", "claude-3-5-sonnet-20240620", false},

		// Non-Claude models - should return false
		{"gemini-3-pro-preview", "gemini-3-pro-preview", false},
		{"gemini-thinking model", "gemini-3-pro-thinking", false}, // not Claude
		{"gpt-4o", "gpt-4o", false},
		{"empty string", "", false},

		// Edge cases
		{"thinking without claude", "thinking-model", false},
		{"claude without thinking", "claude-model", false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := IsClaudeThinkingModel(tt.model)
			if result != tt.expected {
				t.Errorf("IsClaudeThinkingModel(%q) = %v, expected %v", tt.model, result, tt.expected)
			}
		})
	}
}

func TestIsClaudeModel(t *testing.T) {
	tests := []struct {
		name     string
		model    string
		expected bool
	}{
		{"Claude model", "claude-sonnet-4-6", true},
		{"Claude thinking model", "claude-opus-4-6-thinking", true},
		{"case insensitive", "Claude-Sonnet-4-5", true},
		{"Gemini model", "gemini-3-pro-preview", false},
		{"empty string", "", false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := IsClaudeModel(tt.model); got != tt.expected {
				t.Errorf("IsClaudeModel(%q) = %v, expected %v", tt.model, got, tt.expected)
			}
		})
	}
}

func TestIsClaudeLegacySystemReminderModel(t *testing.T) {
	tests := []struct {
		name     string
		model    string
		expected bool
	}{
		{"listed model", "claude-sonnet-4-5", true},
		{"listed snapshot", "claude-haiku-4-5-20251001", true},
		{"mixed case and whitespace", "  Claude-Opus-4-6 ", true},
		{"provider prefix", "anthropic/claude-sonnet-4-6", true},
		{"supported model", "claude-opus-5-5", false},
		{"unknown future model", "claude-sonnet-9", false},
		{"unlisted suffix", "claude-sonnet-4-5-thinking", false},
		{"empty string", "", false},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := IsClaudeLegacySystemReminderModel(tt.model); got != tt.expected {
				t.Errorf("IsClaudeLegacySystemReminderModel(%q) = %v, expected %v", tt.model, got, tt.expected)
			}
		})
	}
}