You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
341 lines
12 KiB
341 lines
12 KiB
/* Copyright (c) 2011, 2018, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License, version 2.0,
|
|
as published by the Free Software Foundation.
|
|
|
|
This program is also distributed with certain software (including
|
|
but not limited to OpenSSL) that is licensed under separate terms,
|
|
as designated in a particular file or component or in included license
|
|
documentation. The authors of MySQL hereby grant you an additional
|
|
permission to link the program and your derivative works with the
|
|
separately licensed software that they have included with MySQL.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License, version 2.0, for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
// First include (the generated) my_config.h, to get correct platform defines.
|
|
#include "my_config.h"
|
|
|
|
#include <assert.h>
|
|
#include <gtest/gtest.h>
|
|
#include <stddef.h>
|
|
#include <string>
|
|
|
|
#include "sql/sql_lex.h"
|
|
#include "thr_lock.h"
|
|
#include "unittest/gunit/parsertest.h"
|
|
#include "unittest/gunit/test_utils.h"
|
|
|
|
namespace union_syntax_unittest {
|
|
|
|
using my_testing::Mock_error_handler;
|
|
using my_testing::Server_initializer;
|
|
|
|
class UnionSyntaxTest : public ParserTest {
|
|
protected:
|
|
void test_union_syntax(const char *query, int num_terms,
|
|
bool expect_syntax_error = false) {
|
|
SELECT_LEX *term1 = parse(query, expect_syntax_error ? ER_PARSE_ERROR : 0);
|
|
EXPECT_EQ(nullptr, term1->first_inner_unit());
|
|
EXPECT_EQ(nullptr, term1->next_select_in_list());
|
|
EXPECT_EQ(1, term1->get_item_list()->head()->val_int());
|
|
|
|
SELECT_LEX_UNIT *top_union = term1->master_unit();
|
|
EXPECT_EQ(nullptr, top_union->outer_select());
|
|
|
|
if (num_terms > 1) {
|
|
SELECT_LEX *term2 = term1->next_select();
|
|
ASSERT_FALSE(term2 == NULL);
|
|
|
|
EXPECT_EQ(nullptr, term2->first_inner_unit());
|
|
EXPECT_EQ(term1, term2->next_select_in_list());
|
|
EXPECT_EQ(2, term2->get_item_list()->head()->val_int());
|
|
|
|
if (num_terms > 2) {
|
|
} else
|
|
EXPECT_EQ(nullptr, term2->next_select());
|
|
|
|
EXPECT_EQ(top_union, term2->master_unit());
|
|
}
|
|
}
|
|
|
|
SELECT_LEX *test_ignored_trailing_limit_on_query_term(const char *query) {
|
|
SELECT_LEX *term1 = parse(query);
|
|
|
|
EXPECT_EQ(nullptr, term1->select_limit);
|
|
|
|
SELECT_LEX_UNIT *top_union = term1->master_unit();
|
|
|
|
EXPECT_EQ(4, top_union->fake_select_lex->select_limit->val_int());
|
|
|
|
EXPECT_EQ(nullptr, top_union->outer_select());
|
|
|
|
SELECT_LEX *term2 = term1->next_select();
|
|
|
|
EXPECT_EQ(nullptr, term2->first_inner_unit());
|
|
EXPECT_EQ(term1, term2->next_select_in_list());
|
|
EXPECT_EQ(2, term2->get_item_list()->head()->val_int());
|
|
|
|
return term1;
|
|
}
|
|
|
|
void test_ignored_trailing_limit_on_query_term(const char *query,
|
|
int expected_limit) {
|
|
SELECT_LEX *term1 = test_ignored_trailing_limit_on_query_term(query);
|
|
EXPECT_EQ(expected_limit, term1->next_select()->select_limit->val_int())
|
|
<< " Outer limit clause should be ignored.";
|
|
}
|
|
};
|
|
|
|
TEST_F(UnionSyntaxTest, First) {
|
|
SELECT_LEX *top = parse("SELECT 1");
|
|
SELECT_LEX_UNIT *top_union = top->master_unit();
|
|
|
|
EXPECT_EQ(nullptr, top->outer_select());
|
|
EXPECT_EQ(top, top_union->first_select());
|
|
}
|
|
|
|
using std::string;
|
|
|
|
string parenthesize(string query, int n) {
|
|
string new_query(n, '(');
|
|
new_query += query;
|
|
new_query.append(n, ')');
|
|
return new_query;
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, QueryExpParensDualOrder) {
|
|
SELECT_LEX *term = parse("( SELECT 1 FROM t1 ORDER BY 2 ) ORDER BY 3");
|
|
ASSERT_FALSE(term == NULL);
|
|
EXPECT_EQ(nullptr, term->next_select());
|
|
EXPECT_EQ(nullptr, term->first_inner_unit());
|
|
|
|
// The query expression.
|
|
SELECT_LEX_UNIT *exp = term->master_unit();
|
|
ASSERT_FALSE(exp == NULL);
|
|
EXPECT_EQ(term, exp->first_select());
|
|
|
|
// Table expression
|
|
EXPECT_EQ(1U, term->table_list.elements);
|
|
TABLE_LIST *table_list = term->table_list.first;
|
|
EXPECT_STREQ("t1", table_list->alias);
|
|
|
|
// Inner order clause, outer is ignored
|
|
EXPECT_EQ(1U, term->order_list.elements);
|
|
ORDER *inner_order = term->order_list.first;
|
|
Item *order_exp = *inner_order->item;
|
|
EXPECT_EQ(2, order_exp->val_int());
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, QueryExpParensOrder) {
|
|
SELECT_LEX *term = parse("(SELECT 1 FROM t1) ORDER BY 3");
|
|
ASSERT_FALSE(term == NULL);
|
|
EXPECT_EQ(nullptr, term->next_select());
|
|
EXPECT_EQ(nullptr, term->first_inner_unit());
|
|
|
|
// The query expression.
|
|
SELECT_LEX_UNIT *exp = term->master_unit();
|
|
ASSERT_FALSE(exp == NULL);
|
|
EXPECT_EQ(term, exp->first_select());
|
|
|
|
// Table expression
|
|
EXPECT_EQ(1U, term->table_list.elements);
|
|
TABLE_LIST *table_list = term->table_list.first;
|
|
EXPECT_STREQ("t1", table_list->alias);
|
|
|
|
// Inner order clause, outer is used when inner is missing
|
|
EXPECT_EQ(1U, term->order_list.elements);
|
|
ORDER *inner_order = term->order_list.first;
|
|
ASSERT_FALSE(inner_order == NULL);
|
|
Item *order_exp = *inner_order->item;
|
|
EXPECT_EQ(3, order_exp->val_int());
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, QueryExpParensLimit) {
|
|
SELECT_LEX *term = parse("(SELECT 1 FROM t1 LIMIT 2) LIMIT 3");
|
|
ASSERT_FALSE(term == NULL);
|
|
EXPECT_EQ(nullptr, term->next_select());
|
|
EXPECT_EQ(nullptr, term->first_inner_unit());
|
|
|
|
// The query expression.
|
|
SELECT_LEX_UNIT *exp = term->master_unit();
|
|
ASSERT_FALSE(exp == NULL);
|
|
EXPECT_EQ(term, exp->first_select());
|
|
|
|
// Table expression
|
|
EXPECT_EQ(1U, term->table_list.elements);
|
|
TABLE_LIST *table_list = term->table_list.first;
|
|
EXPECT_STREQ("t1", table_list->alias);
|
|
|
|
EXPECT_EQ(3, term->select_limit->val_int())
|
|
<< " Outer limit clause overrides the inner one.";
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, Simple) {
|
|
SELECT_LEX *block1 = parse("SELECT 1 UNION SELECT 2 UNION SELECT 3");
|
|
SELECT_LEX *block2 = block1->next_select();
|
|
SELECT_LEX *block3 = block2->next_select();
|
|
|
|
EXPECT_EQ(1, block1->get_item_list()->head()->val_int());
|
|
EXPECT_EQ(2, block2->get_item_list()->head()->val_int());
|
|
EXPECT_EQ(3, block3->get_item_list()->head()->val_int());
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, ThreeWay) {
|
|
const int max_parens = 2;
|
|
|
|
for (int top_parens = 0; top_parens <= max_parens; ++top_parens)
|
|
for (int u1parens = 0; u1parens <= max_parens; ++u1parens)
|
|
for (int u2parens = 0; u2parens <= max_parens; ++u2parens)
|
|
for (int u3parens = 0; u3parens <= max_parens; ++u3parens) {
|
|
string query = parenthesize("SELECT 1", u1parens) + " UNION " +
|
|
parenthesize("SELECT 2", u2parens) + " UNION " +
|
|
parenthesize("SELECT 3", u3parens);
|
|
|
|
query = parenthesize(query, top_parens);
|
|
test_union_syntax(query.c_str(), 3);
|
|
}
|
|
}
|
|
|
|
const char *get_order_by_column_name(SELECT_LEX *query_block, int index = 0) {
|
|
EXPECT_EQ(nullptr, *query_block->order_list.next);
|
|
ORDER *current = query_block->order_list.first;
|
|
for (int i = 0; i < index; ++i) current = current->next;
|
|
Item *order_expression = current->item[0];
|
|
EXPECT_EQ(Item::FIELD_ITEM, order_expression->type());
|
|
// ASSERT_EQ(Item::FIELD_ITEM, order_expression->type());
|
|
assert(Item::FIELD_ITEM == order_expression->type());
|
|
return order_expression->item_name.ptr();
|
|
}
|
|
|
|
const char *get_order_by_column_name(SELECT_LEX_UNIT *query_expression,
|
|
int index = 0) {
|
|
SELECT_LEX *fake = query_expression->fake_select_lex;
|
|
// Why can't I use ASSERT_FALSE here?!
|
|
EXPECT_FALSE(fake == NULL);
|
|
return get_order_by_column_name(fake, index);
|
|
}
|
|
|
|
int get_limit(SELECT_LEX *query_block) {
|
|
return query_block->select_limit->val_int();
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, UnionOrderLimit) {
|
|
SELECT_LEX *block1 =
|
|
parse("SELECT 1 UNION SELECT 2 FROM t1 ORDER BY a LIMIT 123");
|
|
SELECT_LEX *block2 = block1->next_select();
|
|
|
|
EXPECT_EQ(nullptr, block1->select_limit)
|
|
<< "First query block should not have a limit clause.";
|
|
|
|
EXPECT_EQ(nullptr, block2->select_limit)
|
|
<< "Second query block should not have a limit clause.";
|
|
|
|
EXPECT_EQ(0U, block1->order_list.elements)
|
|
<< "First query block should not have an order by clause.";
|
|
|
|
EXPECT_EQ(0U, block2->order_list.elements)
|
|
<< "Second query block should not have an order by clause.";
|
|
|
|
SELECT_LEX_UNIT *query_expression = block1->master_unit();
|
|
|
|
// The limit and order by clauses should belong to the whole query
|
|
// expression, i.e. the "fake select lex".
|
|
SELECT_LEX *fake = query_expression->fake_select_lex;
|
|
EXPECT_EQ(123, get_limit(fake));
|
|
EXPECT_STREQ("a", get_order_by_column_name(fake));
|
|
|
|
// EXPECT_STREQ("union",
|
|
// ((Item_field*)fake->order_list.first->item[0])->context->first_name_resolution_table->alias);
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, UnionNestedQueryBlock) {
|
|
SELECT_LEX *block1 =
|
|
parse("SELECT 1 UNION (SELECT 2 FROM t1 ORDER BY a LIMIT 123)");
|
|
SELECT_LEX *block2 = block1->next_select();
|
|
|
|
EXPECT_EQ(nullptr, block1->select_limit)
|
|
<< "First query block should not have a limit clause.";
|
|
|
|
EXPECT_EQ(0U, block1->order_list.elements)
|
|
<< "First query block should not have an order by clause.";
|
|
|
|
EXPECT_EQ(123, get_limit(block2));
|
|
EXPECT_STREQ("a", get_order_by_column_name(block2));
|
|
|
|
SELECT_LEX_UNIT *query_expression = block1->master_unit();
|
|
|
|
// The limit and order by clauses should belong to the whole query
|
|
// expression, i.e. the "fake select lex".
|
|
SELECT_LEX *fake = query_expression->fake_select_lex;
|
|
// EXPECT_EQ(nullptr, fake->select_limit);
|
|
|
|
EXPECT_EQ(0U, fake->order_list.elements)
|
|
<< "The union should not have an order by clause.";
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, NestedQueryExpWithLimit) {
|
|
SELECT_LEX *block1 =
|
|
parse("(SELECT 1 ORDER BY a LIMIT 5) UNION SELECT 2 ORDER BY b LIMIT 8");
|
|
|
|
SELECT_LEX_UNIT *query_expression = block1->master_unit();
|
|
|
|
EXPECT_STREQ("a", get_order_by_column_name(block1, 0));
|
|
EXPECT_EQ(5, block1->select_limit->val_int());
|
|
|
|
SELECT_LEX *fake = query_expression->fake_select_lex;
|
|
EXPECT_STREQ("b", get_order_by_column_name(fake, 0));
|
|
EXPECT_EQ(8, fake->select_limit->val_int());
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, IgnoredTrailingLimitOnQueryTerm) {
|
|
SELECT_LEX *block1 = parse("(SELECT 1) UNION SELECT 2 LIMIT 123");
|
|
SELECT_LEX *block2 = block1->next_select();
|
|
EXPECT_EQ(2, block2->get_item_list()->head()->val_int());
|
|
|
|
// Neither query block should have a limit clause.
|
|
EXPECT_EQ(nullptr, block1->select_limit);
|
|
EXPECT_EQ(nullptr, block2->select_limit);
|
|
|
|
SELECT_LEX_UNIT *query_expression = block1->master_unit();
|
|
|
|
// The limit clause should belong to the whole query expression, i.e. the
|
|
// "fake select lex".
|
|
EXPECT_EQ(123, query_expression->fake_select_lex->select_limit->val_int());
|
|
|
|
return;
|
|
test_ignored_trailing_limit_on_query_term(
|
|
"(SELECT 1) UNION (SELECT 2 LIMIT 3) LIMIT 4", 3);
|
|
|
|
test_ignored_trailing_limit_on_query_term(
|
|
"(SELECT 1) UNION ((SELECT 2 LIMIT 3)) LIMIT 4", 3);
|
|
|
|
test_ignored_trailing_limit_on_query_term(
|
|
"(SELECT 1) UNION (SELECT 2) LIMIT 4");
|
|
|
|
test_ignored_trailing_limit_on_query_term(
|
|
"(SELECT 1) UNION ((SELECT 2)) LIMIT 4");
|
|
}
|
|
|
|
TEST_F(UnionSyntaxTest, InnerVsOuterOrder) {
|
|
SELECT_LEX *query_block =
|
|
parse("(SELECT b, a FROM t1 ORDER by b, a LIMIT 3) ORDER by a, b");
|
|
|
|
EXPECT_STREQ("b", get_order_by_column_name(query_block, 0));
|
|
EXPECT_STREQ("a", get_order_by_column_name(query_block, 1));
|
|
EXPECT_EQ(3, get_limit(query_block));
|
|
|
|
SELECT_LEX_UNIT *query_expression = query_block->master_unit();
|
|
EXPECT_STREQ("a", get_order_by_column_name(query_expression, 0));
|
|
EXPECT_STREQ("b", get_order_by_column_name(query_expression, 1));
|
|
// EXPECT_EQ(2, get_limit(query_expression->fake_select_lex));
|
|
}
|
|
|
|
} // namespace union_syntax_unittest
|
|
|