Function TestSelectWithOffsetCommand

parser/parser_test.go:393–424

(t *testing.T)

}

func TestSelectWithOffsetCommand(t *testing.T) {
	input := "SELECT * FROM tableName OFFSET 5;"
	expectedOffsetCommand := ast.OffsetCommand{
		Token: token.Token{Type: token.OFFSET, Literal: "OFFSET"},
		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.HasOffsetCommand() {
		t.Fatalf("select command should have offset command")
	}
	testOffsetCommands(t, expectedOffsetCommand, selectCommand.OffsetCommand)
}

func TestSelectWithLimitAndOffsetCommand(t *testing.T) {
	input := "SELECT * FROM tableName ORDER BY colName1 DESC LIMIT 2 OFFSET 13;"