Method TestString

be/src/exprs/iceberg-functions-test.cc:190–225

}

void IcebergTruncatePartitionTransformTests::TestString() {
  MemTracker m;
  MemPool pool(&m);
  StringVal input("Some test input");
  FunctionContext* ctx = CreateFunctionContext(&pool);

  int width = 10;
  StringVal ret_val = IcebergFunctions::TruncatePartitionTransform<false>(
      ctx, input, IntVal(width));
  EXPECT_EQ(ret_val.len, width);
  EXPECT_EQ(strncmp((char*)input.ptr, (char*)ret_val.ptr, width), 0);

  // Truncate width is longer than the input string.
  ret_val = IcebergFunctions::TruncatePartitionTransform<false>(ctx, input, 100);
  EXPECT_EQ(ret_val.len, input.len);
  EXPECT_EQ(strncmp((char*)input.ptr, (char*)ret_val.ptr, input.len), 0);

  // Truncate width is the same as the the input string length.
  ret_val = IcebergFunctions::TruncatePartitionTransform<false>(ctx, input, input.len);
  EXPECT_EQ(ret_val.len, input.len);
  EXPECT_EQ(strncmp((char*)input.ptr, (char*)ret_val.ptr, input.len), 0);

  // Test NULL input.
  ret_val = IcebergFunctions::TruncatePartitionTransform<false>(
      nullptr, StringVal::null(), IntVal(10));
  EXPECT_TRUE(ret_val.is_null);

  // Test empty string input.
  ctx = CreateFunctionContext(&pool);
  ret_val = IcebergFunctions::TruncatePartitionTransform<false>(ctx, "", 10);
  EXPECT_EQ(ret_val.len, 0);

  TestIncorrectWidthParameter<StringVal, IntVal>(StringVal("input"));

  pool.FreeAll();
}

void IcebergTruncatePartitionTransformTests::TestBinary() {
  MemTracker m;