Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Merge branch 'main' into avro_writer
  • Loading branch information
shangxinli authored Dec 29, 2025
commit 323768f5544870afde93ef8a5bd6483c72145274
23 changes: 23 additions & 0 deletions src/iceberg/test/avro_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -945,4 +945,27 @@ INSTANTIATE_TEST_SUITE_P(DirectEncoderModes, AvroWriterParameterizedTest,
return info.param ? "DirectEncoder" : "GenericDatum";
});

TEST_F(AvroReaderTest, BufferSizeConfiguration) {
// Test default buffer size
auto properties1 = ReaderProperties::default_properties();
ASSERT_EQ(properties1->Get(ReaderProperties::kAvroBufferSize), 1024 * 1024);

// Test setting custom buffer size
auto properties2 = ReaderProperties::default_properties();
constexpr int64_t kCustomBufferSize = 2 * 1024 * 1024; // 2MB
properties2->Set(ReaderProperties::kAvroBufferSize, kCustomBufferSize);
ASSERT_EQ(properties2->Get(ReaderProperties::kAvroBufferSize), kCustomBufferSize);

// Test setting via FromMap
std::unordered_map<std::string, std::string> config_map = {
{"read.avro.buffer-size", "4194304"} // 4MB
};
auto properties3 = ReaderProperties::FromMap(config_map);
ASSERT_EQ(properties3->Get(ReaderProperties::kAvroBufferSize), 4194304);

// Test that unset returns to default
properties2->Unset(ReaderProperties::kAvroBufferSize);
ASSERT_EQ(properties2->Get(ReaderProperties::kAvroBufferSize), 1024 * 1024);
}

} // namespace iceberg::avro
Loading
You are viewing a condensed version of this merge commit. You can view the full changes here.