Function TestSelectWithAggregateFunctions

parser/parser_test.go:662–709

(t *testing.T)

}

func TestSelectWithAggregateFunctions(t *testing.T) {
	input := "SELECT MIN(colOne), MAX(colOne), COUNT(*), COUNT(colOne), SUM(colOne), AVG(colOne) FROM tbl;"

	expectedTableName := "tbl"
	expectedSpaces := []ast.Space{
		{
			ColumnName:    token.Token{Type: token.IDENT, Literal: "colOne"},
			AggregateFunc: &token.Token{Type: token.MIN, Literal: "MIN"},
		},
		{
			ColumnName:    token.Token{Type: token.ASTERISK, Literal: "colOne"},
			AggregateFunc: &token.Token{Type: token.MAX, Literal: "MAX"},
		},
		{
			ColumnName:    token.Token{Type: token.IDENT, Literal: "*"},
			AggregateFunc: &token.Token{Type: token.COUNT, Literal: "COUNT"},
		},
		{
			ColumnName:    token.Token{Type: token.IDENT, Literal: "colOne"},
			AggregateFunc: &token.Token{Type: token.COUNT, Literal: "COUNT"},
		},
		{
			ColumnName:    token.Token{Type: token.IDENT, Literal: "colOne"},
			AggregateFunc: &token.Token{Type: token.SUM, Literal: "SUM"},
		},
		{
			ColumnName:    token.Token{Type: token.IDENT, Literal: "colOne"},
			AggregateFunc: &token.Token{Type: token.AVG, Literal: "AVG"},
		},
	}

	lexer := lexer.RunLexer(input)
	parserInstance := New(lexer)
	sequences, err := parserInstance.ParseSequence()
	if err != nil {
		t.Fatalf("Got error from parser: %s", err)
	}

	if len(sequences.Commands) != 1 {
		t.Fatalf("sequences does not contain 1 statements. got=%d", len(sequences.Commands))
	}

	selectCommand := sequences.Commands[0].(*ast.SelectCommand)

	if !testSelectStatement(t, selectCommand, expectedTableName, expectedSpaces, false) {
		return
	}
}

func TestParseUpdateCommand(t *testing.T) {
	tests := []struct {