89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
#ifndef PARSERTEST_INCLUDED
|
|
#define PARSERTEST_INCLUDED
|
|
/* Copyright (c) 2012, 2015, 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 as published by
|
|
the Free Software Foundation; version 2 of the License.
|
|
|
|
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 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,
|
|
51 Franklin Street, Suite 500, Boston, MA 02110-1335 USA */
|
|
|
|
/**
|
|
@file parsertest.h
|
|
|
|
*/
|
|
|
|
#include "test_utils.h"
|
|
|
|
#include "sql_class.h"
|
|
#include "sql_parse.h"
|
|
|
|
using my_testing::Server_initializer;
|
|
using my_testing::Mock_error_handler;
|
|
|
|
/*
|
|
A class for unit testing the parser.
|
|
*/
|
|
class ParserTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
virtual void SetUp() { initializer.SetUp(); }
|
|
virtual void TearDown() { initializer.TearDown(); }
|
|
|
|
THD *thd() { return initializer.thd(); }
|
|
|
|
Server_initializer initializer;
|
|
|
|
/*
|
|
Parses a query and returns a parse tree. In our parser this is
|
|
called a SELECT_LEX.
|
|
*/
|
|
SELECT_LEX *parse(const char *query, int expected_error_code)
|
|
{
|
|
Parser_state state;
|
|
|
|
size_t length= strlen(query);
|
|
char *mutable_query= const_cast<char*>(query);
|
|
|
|
state.init(thd(), mutable_query, length);
|
|
|
|
/*
|
|
This tricks the server to parse the query and then stop,
|
|
without executing.
|
|
*/
|
|
initializer.set_expected_error(ER_MUST_CHANGE_PASSWORD);
|
|
thd()->security_context()->set_password_expired(true);
|
|
|
|
Mock_error_handler handler(thd(), expected_error_code);
|
|
lex_start(thd());
|
|
|
|
if (thd()->db().str == NULL)
|
|
{
|
|
// The THD DTOR will do my_free() on this.
|
|
char *db= static_cast<char*>(my_malloc(PSI_NOT_INSTRUMENTED, 3, MYF(0)));
|
|
sprintf(db, "db");
|
|
LEX_CSTRING db_lex_cstr= { db, strlen(db) };
|
|
thd()->reset_db(db_lex_cstr);
|
|
}
|
|
|
|
lex_start(thd());
|
|
mysql_reset_thd_for_next_command(thd());
|
|
bool err= parse_sql(thd(), &state, NULL);
|
|
EXPECT_FALSE(err);
|
|
return thd()->lex->current_select();
|
|
}
|
|
|
|
SELECT_LEX *parse(const char *query) { return parse(query, 0); }
|
|
|
|
};
|
|
|
|
|
|
#endif // PARSERTEST_INCLUDED
|