106 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			106 lines
		
	
	
		
			3.5 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);
 | |
| };
 | |
| #define T(expr1)                                  \
 | |
|   do {                                            \
 | |
|     MAKE_RAW_EXPR_FROM_STR(expr1, expr);          \
 | |
|     EXPECT_TRUE(OB_SUCC(ret));                    \
 | |
|     uint64_t hash = expr->hash(0);                \
 | |
|     _OB_LOG(INFO, "hash(%s) = %lu", expr1, hash); \
 | |
|   } while (0)
 | |
| TEST_F(TestRawExprToStr, basic)
 | |
| {
 | |
|   int ret = OB_SUCCESS;
 | |
|   // ObExprOperatorGFactory::get_instance()->init();
 | |
|   // 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";
 | |
| 
 | |
|   T("1+c1 > ? and SUM(1) OR 2 >= 1");
 | |
|   T("CASE WHEN 10>=2 THEN 1+2 ELSE 0 END");
 | |
|   T("1");
 | |
|   T("100");
 | |
|   T("sum(1)");
 | |
|   T("c1");
 | |
| }
 | |
| 
 | |
| }  // 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();
 | |
| }
 | 
