[Bug] fix mem_tracker use-after-free & add UT for it (#3899)

This commit is contained in:
HuangWei
2020-06-20 19:08:53 +08:00
committed by GitHub
parent 8e895958d6
commit fdd65c50c4
7 changed files with 345 additions and 282 deletions

View File

@ -16,11 +16,11 @@
// under the License.
#include "exprs/string_functions.h"
#include "util/logging.h"
#include "testutil/function_utils.h"
#include "exprs/anyval_util.h"
#include <iostream>
#include <string>
#include "exprs/anyval_util.h"
#include "testutil/function_utils.h"
#include "util/logging.h"
#include <gtest/gtest.h>
@ -34,9 +34,7 @@ public:
utils = new FunctionUtils();
ctx = utils->get_fn_ctx();
}
void TearDown() {
delete utils;
}
void TearDown() { delete utils; }
private:
FunctionUtils* utils;
@ -69,7 +67,8 @@ TEST_F(StringFunctionsTest, money_format_large_int) {
__int128 value;
ss >> value;
StringVal result = StringFunctions::money_format(context, doris_udf::LargeIntVal(value));
StringVal expected = AnyValUtil::from_string_temp(context, std::string("170,141,183,460,469,231,731,687,303,715,884,105,727.00"));
StringVal expected = AnyValUtil::from_string_temp(
context, std::string("170,141,183,460,469,231,731,687,303,715,884,105,727.00"));
ASSERT_EQ(expected, result);
delete context;
}
@ -140,35 +139,35 @@ TEST_F(StringFunctionsTest, money_format_decimal_v2) {
TEST_F(StringFunctionsTest, split_part) {
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("hello")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("hello")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("word")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 2));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("word")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 2));
ASSERT_EQ(StringVal::null(),
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 3));
StringFunctions::split_part(context, StringVal("hello word"), StringVal(" "), 3));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal("hello"), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal("hello"), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string(" word")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal("hello"), 2));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string(" word")),
StringFunctions::split_part(context, StringVal("hello word"), StringVal("hello"), 2));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("2019年9")),
StringFunctions::split_part(context, StringVal("2019年9月8日"), StringVal(""), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("2019年9")),
StringFunctions::split_part(context, StringVal("2019年9月8日"), StringVal(""), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("bcd")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 2));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("bcd")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 2));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("bd")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 3));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("bd")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 3));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 4));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::split_part(context, StringVal("abcdabda"), StringVal("a"), 4));
delete context;
}
@ -192,11 +191,14 @@ TEST_F(StringFunctionsTest, ends_with) {
ASSERT_EQ(falseRet, StringFunctions::ends_with(context, StringVal(" "), StringVal("hello")));
ASSERT_EQ(falseRet, StringFunctions::ends_with(context, StringVal("hello doris"), StringVal("hello")));
ASSERT_EQ(falseRet,
StringFunctions::ends_with(context, StringVal("hello doris"), StringVal("hello")));
ASSERT_EQ(trueRet, StringFunctions::ends_with(context, StringVal("hello doris"), StringVal("doris")));
ASSERT_EQ(trueRet,
StringFunctions::ends_with(context, StringVal("hello doris"), StringVal("doris")));
ASSERT_EQ(trueRet, StringFunctions::ends_with(context, StringVal("hello doris"), StringVal("hello doris")));
ASSERT_EQ(trueRet, StringFunctions::ends_with(context, StringVal("hello doris"),
StringVal("hello doris")));
ASSERT_EQ(nullRet, StringFunctions::ends_with(context, StringVal("hello"), StringVal::null()));
@ -220,21 +222,27 @@ TEST_F(StringFunctionsTest, starts_with) {
ASSERT_EQ(falseRet, StringFunctions::starts_with(context, StringVal(""), StringVal("hello")));
ASSERT_EQ(trueRet, StringFunctions::starts_with(context, StringVal("hello"), StringVal("hello")));
ASSERT_EQ(trueRet,
StringFunctions::starts_with(context, StringVal("hello"), StringVal("hello")));
ASSERT_EQ(falseRet, StringFunctions::starts_with(context, StringVal("hello"), StringVal(" ")));
ASSERT_EQ(falseRet, StringFunctions::starts_with(context, StringVal(" "), StringVal("world")));
ASSERT_EQ(trueRet, StringFunctions::starts_with(context, StringVal("hello world"), StringVal("hello")));
ASSERT_EQ(trueRet,
StringFunctions::starts_with(context, StringVal("hello world"), StringVal("hello")));
ASSERT_EQ(falseRet, StringFunctions::starts_with(context, StringVal("hello world"), StringVal("world")));
ASSERT_EQ(falseRet,
StringFunctions::starts_with(context, StringVal("hello world"), StringVal("world")));
ASSERT_EQ(trueRet, StringFunctions::starts_with(context, StringVal("hello world"), StringVal("hello world")));
ASSERT_EQ(trueRet, StringFunctions::starts_with(context, StringVal("hello world"),
StringVal("hello world")));
ASSERT_EQ(nullRet, StringFunctions::starts_with(context, StringVal("hello world"), StringVal::null()));
ASSERT_EQ(nullRet,
StringFunctions::starts_with(context, StringVal("hello world"), StringVal::null()));
ASSERT_EQ(nullRet, StringFunctions::starts_with(context, StringVal::null(), StringVal("hello world")));
ASSERT_EQ(nullRet,
StringFunctions::starts_with(context, StringVal::null(), StringVal("hello world")));
ASSERT_EQ(nullRet, StringFunctions::starts_with(context, StringVal::null(), StringVal::null()));
delete context;
@ -264,98 +272,90 @@ TEST_F(StringFunctionsTest, null_or_empty) {
TEST_F(StringFunctionsTest, substring) {
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("hello")),
StringFunctions::substring(context, StringVal("hello word"), 1, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("hello")),
StringFunctions::substring(context, StringVal("hello word"), 1, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("word")),
StringFunctions::substring(context, StringVal("hello word"), 7, 4));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("word")),
StringFunctions::substring(context, StringVal("hello word"), 7, 4));
ASSERT_EQ(StringVal::null(),
StringFunctions::substring(context, StringVal::null(), 1, 0));
ASSERT_EQ(StringVal::null(), StringFunctions::substring(context, StringVal::null(), 1, 0));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::substring(context, StringVal("hello word"), 1, 0));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::substring(context, StringVal("hello word"), 1, 0));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string(" word")),
StringFunctions::substring(context, StringVal("hello word"), -5, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string(" word")),
StringFunctions::substring(context, StringVal("hello word"), -5, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("hello word 你")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 12));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("hello word 你")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 12));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::substring(context, StringVal("hello word 你好"), 13, 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::substring(context, StringVal("hello word 你好"), 13, 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 0));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 0));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("rd 你好")),
StringFunctions::substring(context, StringVal("hello word 你好"), -5, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("rd 你好")),
StringFunctions::substring(context, StringVal("hello word 你好"), -5, 5));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("h")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 1));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("h")),
StringFunctions::substring(context, StringVal("hello word 你好"), 1, 1));
delete context;
}
TEST_F(StringFunctionsTest, reverse) {
FunctionUtils fu;
doris_udf::FunctionContext* context = fu.get_fn_ctx();
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("olleh")),
StringFunctions::reverse(context, StringVal("hello")));
ASSERT_EQ(StringVal::null(),
StringFunctions::reverse(context, StringVal::null()));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("olleh")),
StringFunctions::reverse(context, StringVal("hello")));
ASSERT_EQ(StringVal::null(), StringFunctions::reverse(context, StringVal::null()));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("")),
StringFunctions::reverse(context, StringVal("")));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("")),
StringFunctions::reverse(context, StringVal("")));
ASSERT_EQ(AnyValUtil::from_string_temp(context,std::string("好你olleh")),
StringFunctions::reverse(context, StringVal("hello你好")));
ASSERT_EQ(AnyValUtil::from_string_temp(context, std::string("好你olleh")),
StringFunctions::reverse(context, StringVal("hello你好")));
}
TEST_F(StringFunctionsTest, length) {
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
ASSERT_EQ(IntVal(5),
StringFunctions::length(context, StringVal("hello")));
ASSERT_EQ(IntVal(5),
StringFunctions::char_utf8_length(context, StringVal("hello")));
ASSERT_EQ(IntVal::null(),
StringFunctions::length(context, StringVal::null()));
ASSERT_EQ(IntVal::null(),
StringFunctions::char_utf8_length(context, StringVal::null()));
ASSERT_EQ(IntVal(5), StringFunctions::length(context, StringVal("hello")));
ASSERT_EQ(IntVal(5), StringFunctions::char_utf8_length(context, StringVal("hello")));
ASSERT_EQ(IntVal::null(), StringFunctions::length(context, StringVal::null()));
ASSERT_EQ(IntVal::null(), StringFunctions::char_utf8_length(context, StringVal::null()));
ASSERT_EQ(IntVal(0),
StringFunctions::length(context, StringVal("")));
ASSERT_EQ(IntVal(0),
StringFunctions::char_utf8_length(context, StringVal("")));
ASSERT_EQ(IntVal(0), StringFunctions::length(context, StringVal("")));
ASSERT_EQ(IntVal(0), StringFunctions::char_utf8_length(context, StringVal("")));
ASSERT_EQ(IntVal(11),
StringFunctions::length(context, StringVal("hello你好")));
ASSERT_EQ(IntVal(11), StringFunctions::length(context, StringVal("hello你好")));
ASSERT_EQ(IntVal(7),
StringFunctions::char_utf8_length(context, StringVal("hello你好")));
ASSERT_EQ(IntVal(7), StringFunctions::char_utf8_length(context, StringVal("hello你好")));
delete context;
}
TEST_F(StringFunctionsTest, append_trailing_char_if_absent) {
ASSERT_EQ(StringVal("ac"), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal("a"), StringVal("c")));
ASSERT_EQ(StringVal("ac"),
StringFunctions::append_trailing_char_if_absent(ctx, StringVal("a"), StringVal("c")));
ASSERT_EQ(StringVal("c"), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal("c"), StringVal("c")));
ASSERT_EQ(StringVal("c"),
StringFunctions::append_trailing_char_if_absent(ctx, StringVal("c"), StringVal("c")));
ASSERT_EQ(StringVal("123c"), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal("123c"), StringVal("c")));
ASSERT_EQ(StringVal("123c"), StringFunctions::append_trailing_char_if_absent(
ctx, StringVal("123c"), StringVal("c")));
ASSERT_EQ(StringVal("c"), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal(""), StringVal("c")));
ASSERT_EQ(StringVal("c"),
StringFunctions::append_trailing_char_if_absent(ctx, StringVal(""), StringVal("c")));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal::null(), StringVal("c")));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(
ctx, StringVal::null(), StringVal("c")));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal("a"), StringVal::null()));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(
ctx, StringVal("a"), StringVal::null()));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(ctx,
StringVal("a"), StringVal("abc")));
ASSERT_EQ(StringVal::null(), StringFunctions::append_trailing_char_if_absent(
ctx, StringVal("a"), StringVal("abc")));
}
TEST_F(StringFunctionsTest, instr) {
@ -374,77 +374,127 @@ TEST_F(StringFunctionsTest, instr) {
ASSERT_EQ(IntVal(3), StringFunctions::instr(context, StringVal("你好abc"), StringVal("abc")));
ASSERT_EQ(IntVal::null(), StringFunctions::instr(context, StringVal::null(), StringVal("2")));
ASSERT_EQ(IntVal::null(), StringFunctions::instr(context, StringVal(""), StringVal::null()));
ASSERT_EQ(IntVal::null(), StringFunctions::instr(context, StringVal::null(), StringVal::null()));
ASSERT_EQ(IntVal::null(),
StringFunctions::instr(context, StringVal::null(), StringVal::null()));
delete context;
}
TEST_F(StringFunctionsTest, locate) {
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
ASSERT_EQ(IntVal(4), StringFunctions::locate(context, StringVal("bar"), StringVal("foobarbar")));
ASSERT_EQ(IntVal(4),
StringFunctions::locate(context, StringVal("bar"), StringVal("foobarbar")));
ASSERT_EQ(IntVal(0), StringFunctions::locate(context, StringVal("xbar"), StringVal("foobar")));
ASSERT_EQ(IntVal(2), StringFunctions::locate(context, StringVal("234"), StringVal("123456234")));
ASSERT_EQ(IntVal(2),
StringFunctions::locate(context, StringVal("234"), StringVal("123456234")));
ASSERT_EQ(IntVal(0), StringFunctions::locate(context, StringVal("567"), StringVal("123456")));
ASSERT_EQ(IntVal(2), StringFunctions::locate(context, StringVal(".234"), StringVal("1.234")));
ASSERT_EQ(IntVal(1), StringFunctions::locate(context, StringVal(""), StringVal("1.234")));
ASSERT_EQ(IntVal(0), StringFunctions::locate(context, StringVal("123"), StringVal("")));
ASSERT_EQ(IntVal(1), StringFunctions::locate(context, StringVal(""), StringVal("")));
ASSERT_EQ(IntVal(3), StringFunctions::locate(context, StringVal("世界"), StringVal("你好世界")));
ASSERT_EQ(IntVal(0), StringFunctions::locate(context, StringVal("您好"), StringVal("你好世界")));
ASSERT_EQ(IntVal(3),
StringFunctions::locate(context, StringVal("世界"), StringVal("你好世界")));
ASSERT_EQ(IntVal(0),
StringFunctions::locate(context, StringVal("您好"), StringVal("你好世界")));
ASSERT_EQ(IntVal(3), StringFunctions::locate(context, StringVal("a"), StringVal("你好abc")));
ASSERT_EQ(IntVal(3), StringFunctions::locate(context, StringVal("abc"), StringVal("你好abc")));
ASSERT_EQ(IntVal::null(), StringFunctions::locate(context, StringVal::null(), StringVal("2")));
ASSERT_EQ(IntVal::null(), StringFunctions::locate(context, StringVal(""), StringVal::null()));
ASSERT_EQ(IntVal::null(), StringFunctions::locate(context, StringVal::null(), StringVal::null()));
ASSERT_EQ(IntVal::null(),
StringFunctions::locate(context, StringVal::null(), StringVal::null()));
delete context;
}
TEST_F(StringFunctionsTest, locate_pos) {
doris_udf::FunctionContext* context = new doris_udf::FunctionContext();
ASSERT_EQ(IntVal(7), StringFunctions::locate_pos(context, StringVal("bar"), StringVal("foobarbar"), IntVal(5)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("xbar"), StringVal("foobar"), IntVal(1)));
ASSERT_EQ(IntVal(2), StringFunctions::locate_pos(context, StringVal(""), StringVal("foobar"), IntVal(2)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("foobar"), StringVal(""), IntVal(1)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal(""), StringVal(""), IntVal(2)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("A"), StringVal("AAAAAA"), IntVal(0)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"), IntVal(0)));
ASSERT_EQ(IntVal(2), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"), IntVal(1)));
ASSERT_EQ(IntVal(2), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"), IntVal(2)));
ASSERT_EQ(IntVal(5), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"), IntVal(3)));
ASSERT_EQ(IntVal(7), StringFunctions::locate_pos(context, StringVal("BaR"), StringVal("foobarBaR"), IntVal(5)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal::null(), StringVal("2"), IntVal(1)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal(""), StringVal::null(), IntVal(4)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal::null(), StringVal::null(), IntVal(4)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal::null(), StringVal::null(), IntVal(-1)));
ASSERT_EQ(IntVal(7), StringFunctions::locate_pos(context, StringVal("bar"),
StringVal("foobarbar"), IntVal(5)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("xbar"),
StringVal("foobar"), IntVal(1)));
ASSERT_EQ(IntVal(2),
StringFunctions::locate_pos(context, StringVal(""), StringVal("foobar"), IntVal(2)));
ASSERT_EQ(IntVal(0),
StringFunctions::locate_pos(context, StringVal("foobar"), StringVal(""), IntVal(1)));
ASSERT_EQ(IntVal(0),
StringFunctions::locate_pos(context, StringVal(""), StringVal(""), IntVal(2)));
ASSERT_EQ(IntVal(0),
StringFunctions::locate_pos(context, StringVal("A"), StringVal("AAAAAA"), IntVal(0)));
ASSERT_EQ(IntVal(0), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"),
IntVal(0)));
ASSERT_EQ(IntVal(2), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"),
IntVal(1)));
ASSERT_EQ(IntVal(2), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"),
IntVal(2)));
ASSERT_EQ(IntVal(5), StringFunctions::locate_pos(context, StringVal("A"), StringVal("大A写的A"),
IntVal(3)));
ASSERT_EQ(IntVal(7), StringFunctions::locate_pos(context, StringVal("BaR"),
StringVal("foobarBaR"), IntVal(5)));
ASSERT_EQ(IntVal::null(),
StringFunctions::locate_pos(context, StringVal::null(), StringVal("2"), IntVal(1)));
ASSERT_EQ(IntVal::null(),
StringFunctions::locate_pos(context, StringVal(""), StringVal::null(), IntVal(4)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal::null(),
StringVal::null(), IntVal(4)));
ASSERT_EQ(IntVal::null(), StringFunctions::locate_pos(context, StringVal::null(),
StringVal::null(), IntVal(-1)));
delete context;
}
TEST_F(StringFunctionsTest, lpad) {
ASSERT_EQ(StringVal("???hi"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("?")));
ASSERT_EQ(StringVal("g8%7IgY%AHx7luNtf8Kh"), StringFunctions::lpad(ctx, StringVal("g8%7IgY%AHx7luNtf8Kh"), IntVal(20), StringVal("")));
ASSERT_EQ(StringVal("h"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""), StringFunctions::lpad(ctx, StringVal("你好"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(0), StringVal("?")));
ASSERT_EQ(StringVal::null(), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(-1), StringVal("?")));
ASSERT_EQ(StringVal("h"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(1), StringVal("")));
ASSERT_EQ(StringVal::null(), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("")));
ASSERT_EQ(StringVal("abahi"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("ab")));
ASSERT_EQ(StringVal("ababhi"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(6), StringVal("ab")));
ASSERT_EQ(StringVal("呵呵呵hi"), StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("呵呵")));
ASSERT_EQ(StringVal("hih呵呵"), StringFunctions::lpad(ctx, StringVal("呵呵"), IntVal(5), StringVal("hi")));
ASSERT_EQ(StringVal("???hi"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("?")));
ASSERT_EQ(StringVal("g8%7IgY%AHx7luNtf8Kh"),
StringFunctions::lpad(ctx, StringVal("g8%7IgY%AHx7luNtf8Kh"), IntVal(20),
StringVal("")));
ASSERT_EQ(StringVal("h"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""),
StringFunctions::lpad(ctx, StringVal("你好"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(0), StringVal("?")));
ASSERT_EQ(StringVal::null(),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(-1), StringVal("?")));
ASSERT_EQ(StringVal("h"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(1), StringVal("")));
ASSERT_EQ(StringVal::null(),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("")));
ASSERT_EQ(StringVal("abahi"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("ab")));
ASSERT_EQ(StringVal("ababhi"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(6), StringVal("ab")));
ASSERT_EQ(StringVal("呵呵呵hi"),
StringFunctions::lpad(ctx, StringVal("hi"), IntVal(5), StringVal("呵呵")));
ASSERT_EQ(StringVal("hih呵呵"),
StringFunctions::lpad(ctx, StringVal("呵呵"), IntVal(5), StringVal("hi")));
}
TEST_F(StringFunctionsTest, rpad) {
ASSERT_EQ(StringVal("hi???"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("?")));
ASSERT_EQ(StringVal("g8%7IgY%AHx7luNtf8Kh"), StringFunctions::rpad(ctx, StringVal("g8%7IgY%AHx7luNtf8Kh"), IntVal(20), StringVal("")));
ASSERT_EQ(StringVal("h"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""), StringFunctions::rpad(ctx, StringVal("你好"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(0), StringVal("?")));
ASSERT_EQ(StringVal::null(), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(-1), StringVal("?")));
ASSERT_EQ(StringVal("h"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(1), StringVal("")));
ASSERT_EQ(StringVal::null(), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("")));
ASSERT_EQ(StringVal("hiaba"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("ab")));
ASSERT_EQ(StringVal("hiabab"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(6), StringVal("ab")));
ASSERT_EQ(StringVal("hi呵呵呵"), StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("呵呵")));
ASSERT_EQ(StringVal("呵呵hih"), StringFunctions::rpad(ctx, StringVal("呵呵"), IntVal(5), StringVal("hi")));
}
ASSERT_EQ(StringVal("hi???"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("?")));
ASSERT_EQ(StringVal("g8%7IgY%AHx7luNtf8Kh"),
StringFunctions::rpad(ctx, StringVal("g8%7IgY%AHx7luNtf8Kh"), IntVal(20),
StringVal("")));
ASSERT_EQ(StringVal("h"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""),
StringFunctions::rpad(ctx, StringVal("你好"), IntVal(1), StringVal("?")));
ASSERT_EQ(StringVal(""),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(0), StringVal("?")));
ASSERT_EQ(StringVal::null(),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(-1), StringVal("?")));
ASSERT_EQ(StringVal("h"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(1), StringVal("")));
ASSERT_EQ(StringVal::null(),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("")));
ASSERT_EQ(StringVal("hiaba"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("ab")));
ASSERT_EQ(StringVal("hiabab"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(6), StringVal("ab")));
ASSERT_EQ(StringVal("hi呵呵呵"),
StringFunctions::rpad(ctx, StringVal("hi"), IntVal(5), StringVal("呵呵")));
ASSERT_EQ(StringVal("呵呵hih"),
StringFunctions::rpad(ctx, StringVal("呵呵"), IntVal(5), StringVal("hi")));
}
} // namespace doris
int main(int argc, char** argv) {
std::string conffile = std::string(getenv("DORIS_HOME")) + "/conf/be.conf";