Function TestGetAnnotations

pkg/tools/annotations_test.go:200–235

(t *testing.T)

func (m *mockToolWithoutAnnotations) Prompt() string { return "mock prompt" }

func TestGetAnnotations(t *testing.T) {
	tests := []struct {
		name         string
		tool         Tool
		expectedSafe bool
		expectedRisk int
	}{
		{
			name:         "tool with annotations",
			tool:         &mockToolWithAnnotations{name: "test", annotations: AnnotationsSafeReadOnly},
			expectedSafe: true,
			expectedRisk: RiskLevelSafe,
		},
		{
			name:         "tool without annotations",
			tool:         &mockToolWithoutAnnotations{name: "test"},
			expectedSafe: false,           // 默认不安全
			expectedRisk: RiskLevelMedium, // 未知工具保守处理
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			ann := GetAnnotations(tt.tool)
			if ann == nil {
				t.Fatal("GetAnnotations returned nil")
			}
			if ann.IsSafeForAutoApproval() != tt.expectedSafe {
				t.Errorf("IsSafeForAutoApproval() = %v, want %v", ann.IsSafeForAutoApproval(), tt.expectedSafe)
			}
			if ann.RiskLevel != tt.expectedRisk {
				t.Errorf("RiskLevel = %d, want %d", ann.RiskLevel, tt.expectedRisk)
			}
		})
	}
}

func TestIsToolSafeForAutoApproval(t *testing.T) {
	safeTool := &mockToolWithAnnotations{name: "safe", annotations: AnnotationsSafeReadOnly}