Function TEST_P

contrib/pax_storage/src/cpp/storage/orc/orc_test.cc:755–818

class OrcCompressTest : public OrcEncodingTest {};

TEST_P(OrcEncodingTest, ReadTupleWithEncoding) {
  TupleTableSlot *tuple_slot = nullptr;
  auto encoding_kind = GetParam();

  auto tuple_desc = reinterpret_cast<TupleDescData *>(
      cbdb::Palloc0(sizeof(TupleDescData) + sizeof(FormData_pg_attribute) * 2));

  tuple_desc->natts = 2;
  InitAttribute_int8(&tuple_desc->attrs[0]);
  InitAttribute_int8(&tuple_desc->attrs[1]);

  tuple_slot = MakeTupleTableSlot(tuple_desc, &TTSOpsVirtual);
  bool *fake_is_null =
      reinterpret_cast<bool *>(cbdb::Palloc0(sizeof(bool) * COLUMN_NUMS));
  fake_is_null[0] = false;
  fake_is_null[1] = false;

  tuple_slot->tts_values[0] = Int64GetDatum(0);
  tuple_slot->tts_values[1] = Int64GetDatum(1);
  tuple_slot->tts_isnull = fake_is_null;

  auto local_fs = Singleton<LocalFileSystem>::GetInstance();
  ASSERT_NE(nullptr, local_fs);

  std::unique_ptr<File> file_ptr = local_fs->Open(file_name_, fs::kWriteMode);
  EXPECT_NE(nullptr, file_ptr.get());

  std::vector<pax::porc::proto::Type_Kind> types;
  types.emplace_back(pax::porc::proto::Type_Kind::Type_Kind_LONG);
  types.emplace_back(pax::porc::proto::Type_Kind::Type_Kind_LONG);
  std::vector<std::tuple<ColumnEncoding_Kind, int>> types_encoding;
  types_encoding.emplace_back(std::make_tuple(encoding_kind, 0));
  types_encoding.emplace_back(std::make_tuple(encoding_kind, 0));
  MicroPartitionWriter::WriterOptions writer_options;
  writer_options.encoding_opts = types_encoding;
  writer_options.rel_tuple_desc = tuple_desc;

  auto writer = new OrcWriter(writer_options, types, std::move(file_ptr));

  for (size_t i = 0; i < 10000; i++) {
    tuple_slot->tts_values[0] = Int64GetDatum(i);
    tuple_slot->tts_values[1] = Int64GetDatum(i + 1);
    writer->WriteTuple(tuple_slot);
  }

  writer->Close();

  file_ptr = local_fs->Open(file_name_, fs::kReadMode);

  MicroPartitionReader::ReaderOptions reader_options;
  auto reader = new OrcReader(std::move(file_ptr));
  reader->Open(reader_options);
  EXPECT_EQ(1UL, reader->GetGroupNums());
  for (size_t i = 0; i < 10000; i++) {
    ASSERT_TRUE(reader->ReadTuple(tuple_slot));
    ASSERT_EQ(tuple_slot->tts_values[0], i);
    ASSERT_EQ(tuple_slot->tts_values[1], i + 1);
  }