/* * Copyright (c) 2019 TAOS Data, Inc. * * This program is free software: you can use, redistribute, and/or modify * it under the terms of the GNU Affero General Public License, version 3 * or later ("AGPL"), as published by the Free Software Foundation. * * 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. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . */ #ifndef PARSER_TEST_UTIL_H #define PARSER_TEST_UTIL_H #include #define ALLOW_FORBID_FUNC #include "cmdnodes.h" #include "querynodes.h" #include "taoserror.h" #include "tglobal.h" #include "ttime.h" namespace ParserTest { class ParserTestBaseImpl; enum ParserStage { PARSER_STAGE_PARSE = 1, PARSER_STAGE_TRANSLATE, PARSER_STAGE_CALC_CONST }; class ParserTestBase : public testing::Test { public: ParserTestBase(); virtual ~ParserTestBase(); void login(const std::string& user); void useDb(const std::string& acctId, const std::string& db); void run(const std::string& sql, int32_t expect = TSDB_CODE_SUCCESS, ParserStage checkStage = PARSER_STAGE_TRANSLATE); virtual void checkDdl(const SQuery* pQuery, ParserStage stage); private: std::unique_ptr impl_; }; class ParserDdlTest : public ParserTestBase { public: void setCheckDdlFunc(const std::function& func) { checkDdl_ = func; } virtual void checkDdl(const SQuery* pQuery, ParserStage stage) { ASSERT_NE(pQuery, nullptr); ASSERT_NE(pQuery->pRoot, nullptr); if (nullptr != checkDdl_) { checkDdl_(pQuery, stage); } } private: std::function checkDdl_; }; extern bool g_dump; extern void setAsyncFlag(const char* pArg); extern void setLogLevel(const char* pArg); extern int32_t getLogLevel(); extern void setSkipSqlNum(const char* pArg); extern void setLimitSqlNum(const char* pArg); } // namespace ParserTest #endif // PARSER_TEST_UTIL_H