0da32b916c
- Convert unquoted idents to lowercase. - Recognize quoted idents. - Allow all unicode whitespace characters - Added UnexpectedSymbol token for unexpected input (otherwise it is just ignored) - Handle mixed case keywords in the lexer file instead of filtering the stream
124 lines
3.3 KiB
C++
124 lines
3.3 KiB
C++
#include <gtest/gtest.h>
|
|
#include <gmock/gmock-matchers.h>
|
|
#include "PrintTo_Qt.h"
|
|
#include "sqlparser/Parser.h"
|
|
|
|
using namespace testing;
|
|
using namespace sqlast;
|
|
|
|
TEST(NewSqlLexer, Select)
|
|
{
|
|
std::string source = "SELECT";
|
|
antlr4::ANTLRInputStream input(source);
|
|
PgsqlLexer lexer(&input);
|
|
|
|
auto token = lexer.nextToken();
|
|
ASSERT_EQ(PgsqlLexer::Select, token->getType());
|
|
}
|
|
|
|
TEST(NewSqlLexer, Ident)
|
|
{
|
|
std::string source = "Abc";
|
|
antlr4::ANTLRInputStream input(source);
|
|
PgsqlLexer lexer(&input);
|
|
|
|
auto token = lexer.nextToken();
|
|
ASSERT_EQ(PgsqlLexer::Ident, token->getType());
|
|
ASSERT_EQ("abc", token->getText());
|
|
}
|
|
|
|
TEST(NewSqlLexer, QuotedIdent)
|
|
{
|
|
std::string source = "\"Abc\"";
|
|
antlr4::ANTLRInputStream input(source);
|
|
PgsqlLexer lexer(&input);
|
|
|
|
auto token = lexer.nextToken();
|
|
ASSERT_EQ(PgsqlLexer::Ident, token->getType());
|
|
ASSERT_EQ("Abc", token->getText());
|
|
}
|
|
|
|
|
|
TEST(NewSqlParser, statementList)
|
|
{
|
|
std::string input_string = "SEleCT 1; Select 2;";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_TRUE(program != nullptr);
|
|
ASSERT_EQ(2, program->Count());
|
|
ASSERT_EQ(0, parser.errorCount());
|
|
}
|
|
|
|
TEST(NewSqlParser, missingSemi)
|
|
{
|
|
std::string input_string = "1";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_EQ(1, parser.errorCount());
|
|
}
|
|
|
|
TEST(NewSqlParser, selectList)
|
|
{
|
|
std::string input_string = "SEleCT 1, 'Tekst'";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_TRUE(program != nullptr);
|
|
ASSERT_EQ(1, program->Count());
|
|
ASSERT_EQ(0, parser.errorCount());
|
|
|
|
SelectStatement &s = dynamic_cast<SelectStatement&>(program->Get(0));
|
|
SelectList* sl = s.GetSelectList();
|
|
ASSERT_TRUE(sl != nullptr);
|
|
ASSERT_EQ(2, sl->Count());
|
|
|
|
SelectItem& si = sl->Get(1);
|
|
StringLiteral& string_literal = dynamic_cast<StringLiteral&>(si.GetExpression());
|
|
ASSERT_EQ("Tekst", string_literal.GetValue());
|
|
}
|
|
|
|
TEST(NewSqlParser, selectAliasWithoutAs)
|
|
{
|
|
std::string input_string = "SELECT 1 a";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_TRUE(program != nullptr);
|
|
ASSERT_EQ(1, program->Count());
|
|
ASSERT_EQ(0, parser.errorCount());
|
|
|
|
SelectStatement &s = dynamic_cast<SelectStatement&>(program->Get(0));
|
|
SelectList* sl = s.GetSelectList();
|
|
SelectItem& si = sl->Get(0);
|
|
ASSERT_EQ("a", si.GetAlias());
|
|
}
|
|
|
|
|
|
TEST(NewSqlParser, selectAliasWithAs)
|
|
{
|
|
std::string input_string = "SELECT 1 AS b";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_TRUE(program != nullptr);
|
|
ASSERT_EQ(1, program->Count());
|
|
ASSERT_EQ(0, parser.errorCount());
|
|
|
|
SelectStatement &s = dynamic_cast<SelectStatement&>(program->Get(0));
|
|
SelectList* sl = s.GetSelectList();
|
|
SelectItem& si = sl->Get(0);
|
|
ASSERT_EQ("b", si.GetAlias());
|
|
}
|
|
|
|
TEST(NewSqlParser, selectFrom)
|
|
{
|
|
std::string input_string = "SELECT 1 FROM a";
|
|
Parser parser(input_string);
|
|
std::unique_ptr<sqlast::StatementList> program = parser.Parse();
|
|
|
|
ASSERT_TRUE(program != nullptr);
|
|
ASSERT_EQ(1, program->Count());
|
|
ASSERT_EQ(0, parser.errorCount());
|
|
}
|