Function TestSelectStatementSQL

pkg/sql/ast/sql_test.go:21–112

(t *testing.T)

)

func TestSelectStatementSQL(t *testing.T) {
	tests := []struct {
		name string
		stmt *SelectStatement
		want string
	}{
		{
			name: "simple select",
			stmt: &SelectStatement{
				Columns: []Expression{&Identifier{Name: "*"}},
				From:    []TableReference{{Name: "users"}},
			},
			want: "SELECT * FROM users",
		},
		{
			name: "select with where",
			stmt: &SelectStatement{
				Columns: []Expression{&Identifier{Name: "id"}, &Identifier{Name: "name"}},
				From:    []TableReference{{Name: "users"}},
				Where: &BinaryExpression{
					Left: &Identifier{Name: "active"}, Operator: "=",
					Right: &LiteralValue{Value: true, Type: "BOOLEAN"},
				},
			},
			want: "SELECT id, name FROM users WHERE active = TRUE",
		},
		{
			name: "select distinct",
			stmt: &SelectStatement{
				Distinct: true,
				Columns:  []Expression{&Identifier{Name: "status"}},
				From:     []TableReference{{Name: "orders"}},
			},
			want: "SELECT DISTINCT status FROM orders",
		},
		{
			name: "select with alias",
			stmt: &SelectStatement{
				Columns: []Expression{
					&AliasedExpression{
						Expr:  &FunctionCall{Name: "COUNT", Arguments: []Expression{&Identifier{Name: "*"}}},
						Alias: "total",
					},
				},
				From: []TableReference{{Name: "users"}},
			},
			want: "SELECT COUNT(*) AS total FROM users",
		},
		{
			name: "select with join",
			stmt: &SelectStatement{
				Columns: []Expression{&Identifier{Name: "u.name"}, &Identifier{Name: "o.total"}},
				From:    []TableReference{{Name: "users", Alias: "u"}},
				Joins: []JoinClause{{
					Type: "LEFT", Right: TableReference{Name: "orders", Alias: "o"},
					Condition: &BinaryExpression{Left: &Identifier{Name: "u.id"}, Operator: "=", Right: &Identifier{Name: "o.user_id"}},
				}},
			},