Method TestBinary

be/src/exprs/iceberg-functions-test.cc:227–260

}

void IcebergTruncatePartitionTransformTests::TestBinary() {
  MemTracker m;
  MemPool pool(&m);

  // UTF-8 compatible bytes: "Hello " + UTF-8 encoded "世界" (world in Chinese) + "!"
  uint8_t binary_data[] = {0x48, 0x65, 0x6C, 0x6C, 0x6F, 0x20, 0xE4, 0xB8,
                           0x96, 0xE7, 0x95, 0x8C, 0x21};
  StringVal input(binary_data, sizeof(binary_data));
  FunctionContext* ctx = CreateFunctionContext(&pool);

  for (size_t width = 1; width <= sizeof(binary_data); ++width) {
    StringVal ret_val = IcebergFunctions::TruncatePartitionTransform<true>(
        ctx, input, IntVal(width));
    EXPECT_EQ(ret_val.len, width);
    EXPECT_EQ(memcmp((char*)input.ptr, (char*)ret_val.ptr, width), 0);
  }

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

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

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

  pool.FreeAll();
}

void IcebergTruncatePartitionTransformTests::TestDecimal() {
  // Testing decimal in unit tests by invoking TruncatePartitionTransform seems