Function TestFunctionCall

pkg/sql/ast/nodes_test.go:853–921

Test FunctionCall node

(t *testing.T)

// Test FunctionCall node
func TestFunctionCall(t *testing.T) {
	tests := []struct {
		name         string
		funcCall     *FunctionCall
		wantLiteral  string
		wantChildren int
	}{
		{
			name: "COUNT(*) function",
			funcCall: &FunctionCall{
				Name:      "COUNT",
				Arguments: []Expression{&Identifier{Name: "*"}},
			},
			wantLiteral:  "COUNT",
			wantChildren: 1,
		},
		{
			name: "SUM function with column",
			funcCall: &FunctionCall{
				Name:      "SUM",
				Arguments: []Expression{&Identifier{Name: "amount"}},
			},
			wantLiteral:  "SUM",
			wantChildren: 1,
		},
		{
			name: "window function with OVER clause",
			funcCall: &FunctionCall{
				Name:      "ROW_NUMBER",
				Arguments: []Expression{},
				Over:      &WindowSpec{Name: "w"},
			},
			wantLiteral:  "ROW_NUMBER",
			wantChildren: 1,
		},
		{
			name: "function with multiple arguments",
			funcCall: &FunctionCall{
				Name: "COALESCE",
				Arguments: []Expression{
					&Identifier{Name: "email"},
					&LiteralValue{Value: "N/A"},
				},
			},
			wantLiteral:  "COALESCE",
			wantChildren: 2,
		},
	}

	for _, tt := range tests {
		tt := tt // G601: Create local copy to avoid memory aliasing
		t.Run(tt.name, func(t *testing.T) {
			// Test TokenLiteral
			if got := tt.funcCall.TokenLiteral(); got != tt.wantLiteral {
				t.Errorf("FunctionCall.TokenLiteral() = %v, want %v", got, tt.wantLiteral)
			}

			// Test Children