Function TestWindowSpec

pkg/sql/ast/nodes_test.go:331–398

Test WindowSpec node

(t *testing.T)

// Test WindowSpec node
func TestWindowSpec(t *testing.T) {
	tests := []struct {
		name         string
		windowSpec   *WindowSpec
		wantLiteral  string
		wantChildren int
	}{
		{
			name: "simple window spec",
			windowSpec: &WindowSpec{
				Name: "w",
			},
			wantLiteral:  "WINDOW",
			wantChildren: 0,
		},
		{
			name: "with PARTITION BY",
			windowSpec: &WindowSpec{
				Name:        "w",
				PartitionBy: []Expression{&Identifier{Name: "dept"}},
			},
			wantLiteral:  "WINDOW",
			wantChildren: 1,
		},
		{
			name: "with PARTITION BY and ORDER BY",
			windowSpec: &WindowSpec{
				Name:        "w",
				PartitionBy: []Expression{&Identifier{Name: "dept"}},
				OrderBy:     []OrderByExpression{{Expression: &Identifier{Name: "salary"}, Ascending: true}},
			},
			wantLiteral:  "WINDOW",
			wantChildren: 2,
		},
		{
			name: "with frame clause",
			windowSpec: &WindowSpec{
				Name: "w",
				FrameClause: &WindowFrame{
					Type:  "ROWS",
					Start: WindowFrameBound{Type: "CURRENT ROW"},
				},
			},
			wantLiteral:  "WINDOW",
			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.windowSpec.TokenLiteral(); got != tt.wantLiteral {
				t.Errorf("WindowSpec.TokenLiteral() = %v, want %v", got, tt.wantLiteral)
			}

			// Test Children
			children := tt.windowSpec.Children()