cogito/maxattempts_test.go

package cogito

import "testing"

// TestWithMaxAttemptsClampsBelowOne locks the fix for a silent tool-execution
// no-op: the tool-run loops iterate `for range o.maxAttempts`, so a maxAttempts
// of 0 (a caller forwarding an unset config field into WithMaxAttempts) would
// run the tool zero times and hand the model an empty result with no error.
// WithMaxAttempts must clamp anything below 1 up to 1.
func TestWithMaxAttemptsClampsBelowOne(t *testing.T) {
	cases := []struct {
		in, want int
	}{
		{0, 1},
		{-5, 1},
		{1, 1},
		{3, 3},
	}
	for _, c := range cases {
		o := &Options{}
		WithMaxAttempts(c.in)(o)
		if o.maxAttempts != c.want {
			t.Errorf("WithMaxAttempts(%d): maxAttempts=%d, want %d", c.in, o.maxAttempts, c.want)
		}
	}
}