Function TestSelectWithLimitCommand

parser/parser_test.go:360–391

(t *testing.T)

}

func TestSelectWithLimitCommand(t *testing.T) {
	input := "SELECT * FROM tableName LIMIT 5;"
	expectedLimitCommand := ast.LimitCommand{
		Token: token.Token{Type: token.LIMIT, Literal: "LIMIT"},
		Count: 5,
	}
	expectedTableName := "tableName"
	expectedSpaces := []ast.Space{{ColumnName: token.Token{Type: token.ASTERISK, Literal: "*"}}}

	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
	}

	if !selectCommand.HasLimitCommand() {
		t.Fatalf("sequences does not contain where command")
	}

	testLimitCommands(t, expectedLimitCommand, selectCommand.LimitCommand)
}

func TestSelectWithOffsetCommand(t *testing.T) {
	input := "SELECT * FROM tableName OFFSET 5;"