Function TestDefaultToolSelector_SelectTools

pkg/tools/constraints_test.go:137–203

(t *testing.T)

}

func TestDefaultToolSelector_SelectTools(t *testing.T) {
	selector := &DefaultToolSelector{}
	ctx := context.Background()

	// 创建模拟工具
	allTools := []Tool{
		&mockTool{name: "Read"},
		&mockTool{name: "Write"},
		&mockTool{name: "Bash"},
	}

	tests := []struct {
		name          string
		constraints   ToolConstraints
		expectedCount int
		expectedNames []string
	}{
		{
			name:          "no constraints",
			constraints:   &NoConstraints{},
			expectedCount: 3,
		},
		{
			name:          "whitelist",
			constraints:   NewWhitelistConstraints([]string{"Read", "Write"}),
			expectedCount: 2,
			expectedNames: []string{"Read", "Write"},
		},
		{
			name:          "blacklist",
			constraints:   NewBlacklistConstraints([]string{"Bash"}),
			expectedCount: 2,
			expectedNames: []string{"Read", "Write"},
		},
		{
			name:          "required",
			constraints:   NewRequiredToolConstraints("Read"),
			expectedCount: 1,
			expectedNames: []string{"Read"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			selected, err := selector.SelectTools(ctx, allTools, tt.constraints)
			if err != nil {
				t.Fatalf("SelectTools failed: %v", err)
			}

			if len(selected) != tt.expectedCount {
				t.Errorf("Expected %d tools, got %d", tt.expectedCount, len(selected))
			}

			if tt.expectedNames != nil {
				names := make(map[string]bool)
				for _, tool := range selected {
					names[tool.Name()] = true
				}