135 lines
3.5 KiB
C++
135 lines
3.5 KiB
C++
|
|
|
|
#include "thirdparty/microtest/microtest.h"
|
|
#include "sql_asserts.h"
|
|
#include "SQLParser.h"
|
|
|
|
using namespace hsql;
|
|
|
|
TEST(SelectTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT * FROM students;",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT_NULL(stmt->whereClause);
|
|
ASSERT_NULL(stmt->groupBy);
|
|
|
|
delete result;
|
|
}
|
|
|
|
TEST(SelectExprTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT a, MAX(b), CUSTOM(c, F(un)) FROM students;",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT_NULL(stmt->whereClause);
|
|
ASSERT_NULL(stmt->groupBy);
|
|
|
|
ASSERT_EQ(stmt->selectList->size(), 3);
|
|
|
|
ASSERT(stmt->selectList->at(0)->isType(kExprColumnRef));
|
|
ASSERT_STREQ(stmt->selectList->at(0)->getName(), "a");
|
|
|
|
ASSERT(stmt->selectList->at(1)->isType(kExprFunctionRef));
|
|
ASSERT_STREQ(stmt->selectList->at(1)->getName(), "MAX");
|
|
ASSERT_NOTNULL(stmt->selectList->at(1)->exprList);
|
|
ASSERT_EQ(stmt->selectList->at(1)->exprList->size(), 1);
|
|
ASSERT(stmt->selectList->at(1)->exprList->at(0)->isType(kExprColumnRef));
|
|
ASSERT_STREQ(stmt->selectList->at(1)->exprList->at(0)->getName(), "b");
|
|
|
|
ASSERT(stmt->selectList->at(2)->isType(kExprFunctionRef));
|
|
ASSERT_STREQ(stmt->selectList->at(2)->getName(), "CUSTOM");
|
|
ASSERT_NOTNULL(stmt->selectList->at(2)->exprList);
|
|
ASSERT_EQ(stmt->selectList->at(2)->exprList->size(), 2);
|
|
ASSERT(stmt->selectList->at(2)->exprList->at(0)->isType(kExprColumnRef));
|
|
ASSERT_STREQ(stmt->selectList->at(2)->exprList->at(0)->getName(), "c");
|
|
|
|
ASSERT(stmt->selectList->at(2)->exprList->at(1)->isType(kExprFunctionRef));
|
|
ASSERT_STREQ(stmt->selectList->at(2)->exprList->at(1)->getName(), "F");
|
|
ASSERT_EQ(stmt->selectList->at(2)->exprList->at(1)->exprList->size(), 1);
|
|
ASSERT(stmt->selectList->at(2)->exprList->at(1)->exprList->at(0)->isType(kExprColumnRef));
|
|
ASSERT_STREQ(stmt->selectList->at(2)->exprList->at(1)->exprList->at(0)->getName(), "un");
|
|
|
|
delete result;
|
|
}
|
|
|
|
|
|
TEST(SelectHavingTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT city, AVG(grade) AS avg_grade FROM students GROUP BY city HAVING AVG(grade) < 2.0",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT_FALSE(stmt->selectDistinct);
|
|
|
|
GroupByDescription* group = stmt->groupBy;
|
|
ASSERT_NOTNULL(group);
|
|
ASSERT_EQ(group->columns->size(), 1);
|
|
ASSERT(group->having->isSimpleOp('<'));
|
|
ASSERT(group->having->expr->isType(kExprFunctionRef));
|
|
ASSERT(group->having->expr2->isType(kExprLiteralFloat));
|
|
|
|
delete result;
|
|
}
|
|
|
|
|
|
TEST(SelectDistinctTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT DISTINCT grade, city FROM students;",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT(stmt->selectDistinct);
|
|
ASSERT_NULL(stmt->whereClause);
|
|
|
|
delete result;
|
|
}
|
|
|
|
TEST(SelectGroupDistinctTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT city, COUNT(name), COUNT(DISTINCT grade) FROM students GROUP BY city;",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT_FALSE(stmt->selectDistinct);
|
|
ASSERT_EQ(stmt->selectList->size(), 3);
|
|
ASSERT(!stmt->selectList->at(1)->distinct);
|
|
ASSERT(stmt->selectList->at(2)->distinct);
|
|
|
|
delete result;
|
|
}
|
|
|
|
|
|
TEST(OrderByTest) {
|
|
TEST_PARSE_SINGLE_SQL(
|
|
"SELECT grade, city FROM students ORDER BY grade, city DESC;",
|
|
kStmtSelect,
|
|
SelectStatement,
|
|
result,
|
|
stmt);
|
|
|
|
ASSERT_NULL(stmt->whereClause);
|
|
ASSERT_NOTNULL(stmt->order);
|
|
|
|
ASSERT_EQ(stmt->order->size(), 2);
|
|
ASSERT_EQ(stmt->order->at(0)->type, kOrderAsc);
|
|
ASSERT_STREQ(stmt->order->at(0)->expr->name, "grade");
|
|
|
|
ASSERT_EQ(stmt->order->at(1)->type, kOrderDesc);
|
|
ASSERT_STREQ(stmt->order->at(1)->expr->name, "city");
|
|
|
|
delete result;
|
|
}
|