ds2api/internal/util/thinking_test.go

package util

import "testing"

func TestResolveThinkingEnabledPriority(t *testing.T) {
	req := map[string]any{
		"thinking": map[string]any{"type": "disabled"},
		"extra_body": map[string]any{
			"thinking": map[string]any{"type": "enabled"},
		},
		"reasoning_effort": "high",
	}
	if got := ResolveThinkingEnabled(req, true); got {
		t.Fatalf("expected top-level thinking to win, got enabled=%v", got)
	}
}

func TestResolveThinkingEnabledUsesExtraBodyFallback(t *testing.T) {
	req := map[string]any{
		"extra_body": map[string]any{
			"thinking": map[string]any{"type": "disabled"},
		},
	}
	if got := ResolveThinkingEnabled(req, true); got {
		t.Fatalf("expected extra_body thinking to disable, got enabled=%v", got)
	}
}

func TestResolveThinkingEnabledMapsReasoningEffortToEnabled(t *testing.T) {
	for _, effort := range []string{"minimal", "low", "medium", "high", "xhigh"} {
		if got := ResolveThinkingEnabled(map[string]any{"reasoning_effort": effort}, false); !got {
			t.Fatalf("expected reasoning_effort=%s to enable thinking", effort)
		}
	}
}

func TestResolveThinkingEnabledMapsReasoningObject(t *testing.T) {
	req := map[string]any{"reasoning": map[string]any{"effort": "none"}}
	if got := ResolveThinkingEnabled(req, true); got {
		t.Fatalf("expected reasoning.effort=none to disable thinking")
	}
	req = map[string]any{"reasoning": map[string]any{"effort": "medium"}}
	if got := ResolveThinkingEnabled(req, false); !got {
		t.Fatalf("expected reasoning.effort=medium to enable thinking")
	}
}

func TestResolveThinkingEnabledDefaultsWhenUnset(t *testing.T) {
	if !ResolveThinkingEnabled(nil, true) {
		t.Fatal("expected default thinking=true when unset")
	}
	if ResolveThinkingEnabled(nil, false) {
		t.Fatal("expected default thinking=false when unset")
	}
}