Function ASSERT_OK_AND_ASSIGN

cpp/src/parquet/column_writer_test.cc:1976–2382

  ASSERT_THAT(key_value_metadata, NotNull());
  ASSERT_EQ(2, key_value_metadata->size());
  ASSERT_OK_AND_ASSIGN(auto value, key_value_metadata->Get("hello"));
  ASSERT_EQ("world", value);
  ASSERT_OK_AND_ASSIGN(value, key_value_metadata->Get("bye"));
  ASSERT_EQ("moon", value);
}

TEST_F(TestInt32Writer, ResetKeyValueMetadata) {
  auto writer = this->BuildWriter();
  writer->AddKeyValueMetadata(KeyValueMetadata::Make({"hello"}, {"world"}));
  writer->ResetKeyValueMetadata();
  writer->Close();
  auto key_value_metadata = metadata_key_value_metadata();
  ASSERT_THAT(key_value_metadata, IsNull());
}

TEST_F(TestInt32Writer, WriteKeyValueMetadataEndToEnd) {
  auto sink = CreateOutputStream();
  {
    auto file_writer = ParquetFileWriter::Open(
        sink, std::dynamic_pointer_cast<schema::GroupNode>(schema_.schema_root()));
    auto rg_writer = file_writer->AppendRowGroup();
    auto col_writer = rg_writer->NextColumn();
    col_writer->AddKeyValueMetadata(KeyValueMetadata::Make({"foo"}, {"bar"}));
    file_writer->Close();
  }
  ASSERT_OK_AND_ASSIGN(auto buffer, sink->Finish());
  auto file_reader =
      ParquetFileReader::Open(std::make_shared<::arrow::io::BufferReader>(buffer));
  auto key_value_metadata =
      file_reader->metadata()->RowGroup(0)->ColumnChunk(0)->key_value_metadata();
  ASSERT_THAT(key_value_metadata, NotNull());
  ASSERT_EQ(1U, key_value_metadata->size());
  ASSERT_OK_AND_ASSIGN(auto value, key_value_metadata->Get("foo"));
  ASSERT_EQ("bar", value);
}

TEST_F(TestValuesWriterInt32Type, AllNullsCompressionInPageV2) {
  // GH-31992: In DataPageV2, the levels and data will not be compressed together,
  // so, when all values are null, the compressed values should be empty. And
  // we should handle this case correctly.
  std::vector<Compression::type> compressions = {Compression::SNAPPY, Compression::GZIP,
                                                 Compression::ZSTD, Compression::BROTLI,
                                                 Compression::LZ4};
  for (auto compression : compressions) {
    if (!Codec::IsAvailable(compression)) {
      continue;
    }
    ARROW_SCOPED_TRACE("compression = ", Codec::GetCodecAsString(compression));
    // Optional and non-repeated, with definition levels
    // but no repetition levels
    this->SetUpSchema(Repetition::OPTIONAL);
    this->GenerateData(SMALL_SIZE);
    std::fill(this->def_levels_.begin(), this->def_levels_.end(), 0);
    ColumnProperties column_properties;
    column_properties.set_compression(compression);

    auto writer =
        this->BuildWriter(SMALL_SIZE, column_properties, ParquetVersion::PARQUET_2_LATEST,