Function TestSelect

pkg/sql/ast/dml_test.go:20–131

Test Select node (DML)

(t *testing.T)

// Test Select node (DML)
func TestSelect(t *testing.T) {
	tests := []struct {
		name        string
		sel         *Select
		wantLiteral string
		minChildren int
	}{
		{
			name: "simple SELECT with columns",
			sel: &Select{
				Columns: []Expression{
					&Identifier{Name: "id"},
					&Identifier{Name: "name"},
				},
			},
			wantLiteral: "SELECT",
			minChildren: 2,
		},
		{
			name: "SELECT with FROM",
			sel: &Select{
				Columns: []Expression{&Identifier{Name: "*"}},
				From: []TableReference{
					{Name: "users"},
				},
			},
			wantLiteral: "SELECT",
			minChildren: 2,
		},
		{
			name: "SELECT with WHERE",
			sel: &Select{
				Columns: []Expression{&Identifier{Name: "id"}},
				From: []TableReference{
					{Name: "users"},
				},
				Where: &BinaryExpression{
					Left:     &Identifier{Name: "id"},
					Operator: "=",
					Right:    &LiteralValue{Value: "1"},
				},
			},
			wantLiteral: "SELECT",
			minChildren: 3,
		},
		{
			name: "SELECT with GROUP BY and HAVING",
			sel: &Select{
				Columns: []Expression{
					&Identifier{Name: "dept"},
					&FunctionCall{Name: "COUNT", Arguments: []Expression{&Identifier{Name: "*"}}},
				},
				From: []TableReference{
					{Name: "employees"},
				},
				GroupBy: []Expression{&Identifier{Name: "dept"}},
				Having: &BinaryExpression{
					Left:     &FunctionCall{Name: "COUNT", Arguments: []Expression{&Identifier{Name: "*"}}},