init push

unittest/sql/resolver/expr/test_raw_expr_to_str.cpp

@@ -0,0 +1,120 @@
+/**
+ * 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();
+}