Function TestCommonTableExpr

pkg/sql/ast/nodes_test.go:97–144

Test CommonTableExpr node

(t *testing.T)

// Test CommonTableExpr node
func TestCommonTableExpr(t *testing.T) {
	tests := []struct {
		name         string
		cte          *CommonTableExpr
		wantLiteral  string
		wantChildren int
	}{
		{
			name: "simple CTE",
			cte: &CommonTableExpr{
				Name:      "my_cte",
				Statement: &SelectStatement{},
			},
			wantLiteral:  "my_cte",
			wantChildren: 1,
		},
		{
			name: "CTE with columns",
			cte: &CommonTableExpr{
				Name:      "cte_with_cols",
				Columns:   []string{"col1", "col2", "col3"},
				Statement: &SelectStatement{},
			},
			wantLiteral:  "cte_with_cols",
			wantChildren: 1,
		},
	}

	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.cte.TokenLiteral(); got != tt.wantLiteral {
				t.Errorf("CommonTableExpr.TokenLiteral() = %v, want %v", got, tt.wantLiteral)
			}

			// Test Children
			children := tt.cte.Children()
			if len(children) != tt.wantChildren {
				t.Errorf("CommonTableExpr.Children() count = %d, want %d", len(children), tt.wantChildren)
			}

			// Test that it implements Statement interface
			var _ Statement = tt.cte
			tt.cte.statementNode()
		})
	}
}

// Test SetOperation node
func TestSetOperation(t *testing.T) {