test(tools,utils): add ToolRegistry unit tests and fix Truncate panic on negative maxLen (#517)

Add comprehensive unit tests for the ToolRegistry covering registration, lookup, execution, context injection, async callbacks, schema generation, provider definition conversion, and concurrent access. Fix a defensive edge case in Truncate where a negative maxLen would cause a slice bounds panic, and add table-driven tests covering boundary conditions, zero/negative lengths, and Unicode handling. Co-authored-by: Cursor <cursoragent@cursor.com>
2026-06-12 18:08:54 +00:00 · 2026-02-22 18:40:59 +08:00
parent 6b55fb5f1d
commit cb0c8703fb
3 changed files with 459 additions and 0 deletions
@@ -0,0 +1,106 @@
+package utils
+
+import "testing"
+
+func TestTruncate(t *testing.T) {
+	tests := []struct {
+		name   string
+		input  string
+		maxLen int
+		want   string
+	}{
+		{
+			name:   "short string unchanged",
+			input:  "hi",
+			maxLen: 10,
+			want:   "hi",
+		},
+		{
+			name:   "exact length unchanged",
+			input:  "hello",
+			maxLen: 5,
+			want:   "hello",
+		},
+		{
+			name:   "long string truncated with ellipsis",
+			input:  "hello world",
+			maxLen: 8,
+			want:   "hello...",
+		},
+		{
+			name:   "maxLen equals 4 leaves 1 char plus ellipsis",
+			input:  "abcdef",
+			maxLen: 4,
+			want:   "a...",
+		},
+		{
+			name:   "maxLen 3 returns first 3 chars without ellipsis",
+			input:  "abcdef",
+			maxLen: 3,
+			want:   "abc",
+		},
+		{
+			name:   "maxLen 2 returns first 2 chars",
+			input:  "abcdef",
+			maxLen: 2,
+			want:   "ab",
+		},
+		{
+			name:   "maxLen 1 returns first char",
+			input:  "abcdef",
+			maxLen: 1,
+			want:   "a",
+		},
+		{
+			name:   "maxLen 0 returns empty",
+			input:  "hello",
+			maxLen: 0,
+			want:   "",
+		},
+		{
+			name:   "negative maxLen returns empty",
+			input:  "hello",
+			maxLen: -1,
+			want:   "",
+		},
+		{
+			name:   "empty string unchanged",
+			input:  "",
+			maxLen: 5,
+			want:   "",
+		},
+		{
+			name:   "empty string with zero maxLen",
+			input:  "",
+			maxLen: 0,
+			want:   "",
+		},
+		{
+			name:   "unicode truncated correctly",
+			input:  "\U0001f600\U0001f601\U0001f602\U0001f603\U0001f604",
+			maxLen: 4,
+			want:   "\U0001f600...",
+		},
+		{
+			name:   "unicode short enough",
+			input:  "\u00e9\u00e8",
+			maxLen: 5,
+			want:   "\u00e9\u00e8",
+		},
+		{
+			name:   "mixed ascii and unicode",
+			input:  "Go\U0001f680\U0001f525\U0001f4a5\U0001f30d",
+			maxLen: 5,
+			want:   "Go...",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := Truncate(tt.input, tt.maxLen)
+			if got != tt.want {
+				t.Errorf("Truncate(%q, %d) = %q, want %q", tt.input, tt.maxLen, got, tt.want)
+			}
+		})
+	}
+}