Function TestOverflowKindOf

pkg/modelerrors/modelerrors_test.go:322–370

(t *testing.T)

}

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

	t.Run("returns stored Kind on wrapped error", func(t *testing.T) {
		t.Parallel()
		err := &ContextOverflowError{Underlying: errors.New("anything"), Kind: OverflowKindWire}
		assert.Equal(t, OverflowKindWire, OverflowKindOf(err))
	})

	t.Run("classifies underlying when wrap has no Kind", func(t *testing.T) {
		t.Parallel()
		// Legacy wrap: Kind left empty, Underlying carries the signal.
		err := &ContextOverflowError{Underlying: errors.New("prompt is too long")}
		assert.Equal(t, OverflowKindTokens, OverflowKindOf(err))
	})

	t.Run("falls back to tokens on legacy wrap with no signal", func(t *testing.T) {
		t.Parallel()
		err := &ContextOverflowError{Underlying: errors.New("opaque")}
		assert.Equal(t, OverflowKindTokens, OverflowKindOf(err))
	})

	t.Run("returns empty on non-overflow error", func(t *testing.T) {
		t.Parallel()
		assert.Equal(t, OverflowKind(""), OverflowKindOf(errors.New("rate limited")))
		assert.Equal(t, OverflowKind(""), OverflowKindOf(nil))
	})

	t.Run("NewContextOverflowError sets Kind from underlying", func(t *testing.T) {
		t.Parallel()
		// Anthropic 413 with structured body → wire
		under := &StatusError{StatusCode: 413, Err: errors.New(
			`413 Payload Too Large {"type":"error","error":{"type":"request_too_large","message":"too big"}}`)}
		wrapped := NewContextOverflowError(under)
		assert.Equal(t, OverflowKindWire, wrapped.Kind)

		// Token-overflow prose → tokens
		wrapped = NewContextOverflowError(errors.New("prompt is too long"))
		assert.Equal(t, OverflowKindTokens, wrapped.Kind)

		// Image rejection → media
		wrapped = NewContextOverflowError(errors.New("image exceeds 5 MB maximum"))
		assert.Equal(t, OverflowKindMedia, wrapped.Kind)

		// Unclassifiable underlying → tokens (safe historical default)
		wrapped = NewContextOverflowError(errors.New("opaque"))
		assert.Equal(t, OverflowKindTokens, wrapped.Kind)
	})
}

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