121 lines
3.9 KiB
C++
121 lines
3.9 KiB
C++
/**
|
|
* Copyright (c) 2021 OceanBase
|
|
* OceanBase CE is licensed under Mulan PubL v2.
|
|
* You can use this software according to the terms and conditions of the Mulan PubL v2.
|
|
* You may obtain a copy of Mulan PubL v2 at:
|
|
* http://license.coscl.org.cn/MulanPubL-2.0
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
|
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
|
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
|
* See the Mulan PubL v2 for more details.
|
|
*/
|
|
|
|
#define USING_LOG_PREFIX SQL_OPTIMIZER
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "lib/json/ob_json.h"
|
|
#include "lib/allocator/ob_mod_define.h"
|
|
#include "lib/allocator/page_arena.h"
|
|
#include "lib/oblog/ob_log.h"
|
|
#include "sql/resolver/expr/ob_raw_expr_util.h"
|
|
#include "sql/engine/expr/ob_expr_operator.h"
|
|
|
|
using namespace oceanbase::sql;
|
|
using namespace oceanbase::common;
|
|
namespace test {
|
|
|
|
#define MAKE_RAW_EXPR_FROM_STR(str, expr) \
|
|
({ \
|
|
ret = ObRawExprUtils::make_raw_expr_from_str( \
|
|
str, strlen(str), ctx, expr, columns, sys_vars, &sub_query_info, aggr_exprs, win_exprs); \
|
|
ret; \
|
|
})
|
|
|
|
class TestRawExprToStr : public ::testing::Test {
|
|
public:
|
|
TestRawExprToStr()
|
|
{}
|
|
virtual ~TestRawExprToStr()
|
|
{}
|
|
virtual void SetUp()
|
|
{}
|
|
virtual void TearDown()
|
|
{}
|
|
|
|
private:
|
|
// disallow copy and assign
|
|
TestRawExprToStr(const TestRawExprToStr& other);
|
|
TestRawExprToStr& operator=(const TestRawExprToStr& ohter);
|
|
};
|
|
|
|
TEST_F(TestRawExprToStr, basic)
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
// mock params
|
|
// stmts
|
|
ObArenaAllocator allocator(ObModIds::TEST);
|
|
ObRawExprFactory expr_factory(allocator);
|
|
ObArray<ObQualifiedName> columns;
|
|
ObArray<ObVarInfo> sys_vars;
|
|
ObArray<ObSubQueryInfo> sub_query_info;
|
|
ObArray<ObAggFunRawExpr*> aggr_exprs;
|
|
ObArray<ObWinFunRawExpr*> win_exprs;
|
|
ObTimeZoneInfo tz_info;
|
|
ObNameCaseMode case_mode = OB_NAME_CASE_INVALID;
|
|
ObExprResolveContext ctx(expr_factory, &tz_info, case_mode);
|
|
ctx.connection_charset_ = ObCharset::get_default_charset();
|
|
ctx.dest_collation_ = ObCharset::get_default_collation(ctx.connection_charset_);
|
|
ctx.is_extract_param_type_ = false;
|
|
ObSQLSessionInfo session;
|
|
session.set_use_static_typing_engine(false);
|
|
ctx.session_info_ = &session;
|
|
|
|
const int64_t buf_len = 1024;
|
|
int64_t pos = 0;
|
|
char buf[buf_len];
|
|
|
|
ObRawExpr* expr = NULL;
|
|
// const char* inner_offset = "1+c1 > ? and 'abc' || c2 = 'def'";
|
|
const char* expr1 = "1+c1 > ? and SUM(1) OR 2 >= 1";
|
|
const char* expr2 = "CASE WHEN 10>=2 THEN 1+2 ELSE 0 END";
|
|
const char* expr3 = "CASE WHEN 10>=2 THEN 10-2 ELSE SUM(10-2) END";
|
|
const char* expr4 = "1/2";
|
|
|
|
MAKE_RAW_EXPR_FROM_STR(expr1, expr);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
pos = 0;
|
|
ret = expr->get_name(buf, buf_len, pos);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
_OB_LOG(INFO, "%.*s", static_cast<int32_t>(pos), buf);
|
|
|
|
pos = 0;
|
|
MAKE_RAW_EXPR_FROM_STR(expr2, expr);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
ret = expr->get_name(buf, buf_len, pos);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
_OB_LOG(INFO, "%.*s", static_cast<int32_t>(pos), buf);
|
|
|
|
pos = 0;
|
|
MAKE_RAW_EXPR_FROM_STR(expr3, expr);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
ret = expr->get_name(buf, buf_len, pos);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
_OB_LOG(INFO, "%.*s", static_cast<int32_t>(pos), buf);
|
|
|
|
pos = 0;
|
|
MAKE_RAW_EXPR_FROM_STR(expr4, expr);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
ret = expr->get_name(buf, buf_len, pos);
|
|
EXPECT_TRUE(OB_SUCC(ret));
|
|
_OB_LOG(INFO, "%.*s", static_cast<int32_t>(pos), buf);
|
|
}
|
|
|
|
} // namespace test
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
oceanbase::common::ObLogger::get_logger().set_log_level("INFO");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|