@@ -1076,7 +1076,7 @@ TEST_P(MergeTreeWriterTest, TestCloseSkipsDeleteForUpgradedFilesInCompactAfter)
10761076 << " Intermediate file should be deleted because it's not in compact_before_" ;
10771077}
10781078
1079- TEST_P (MergeTreeWriterTest, TestSpillWithSameKeyDeduplicate) {
1079+ TEST_F (MergeTreeWriterTest, TestSpillWithSameKeyDeduplicate) {
10801080 ASSERT_OK_AND_ASSIGN (CoreOptions options,
10811081 CoreOptions::FromMap ({{Options::FILE_FORMAT , " orc" },
10821082 {Options::WRITE_BUFFER_SIZE , " 1" },
@@ -1111,9 +1111,8 @@ TEST_P(MergeTreeWriterTest, TestSpillWithSameKeyDeduplicate) {
11111111
11121112 WriteBatch (batch1, /* row_kinds=*/ {}, merge_writer.get ());
11131113 WriteBatch (batch2, /* row_kinds=*/ {}, merge_writer.get ());
1114- // Leveled compaction may merge 2 files into 1 when actual_max_fan_in_ is low.
1115- ASSERT_GE (2u , TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ));
1116- ASSERT_GE (TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ), 1u );
1114+ // actual_max_fan_in_=2 with 1-byte budget, so 2 spill files compact to 1.
1115+ ASSERT_EQ (1u , TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ));
11171116
11181117 std::shared_ptr<arrow::Array> batch3 =
11191118 arrow::ipc::internal::json::ArrayFromJSON (value_type_, R"( [
@@ -1142,7 +1141,7 @@ TEST_P(MergeTreeWriterTest, TestSpillWithSameKeyDeduplicate) {
11421141 CheckFileContent (expected_data_file_path, expected_array);
11431142}
11441143
1145- TEST_P (MergeTreeWriterTest, TestIntermediateMergeSpillFileBound) {
1144+ TEST_F (MergeTreeWriterTest, TestIntermediateMergeSpillFileBound) {
11461145 ASSERT_OK_AND_ASSIGN (CoreOptions options,
11471146 CoreOptions::FromMap ({{Options::FILE_FORMAT , " orc" },
11481147 {Options::WRITE_BUFFER_SIZE , " 1" },
@@ -1186,9 +1185,8 @@ TEST_P(MergeTreeWriterTest, TestIntermediateMergeSpillFileBound) {
11861185 ASSERT_EQ (1u , TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ));
11871186
11881187 WriteBatch (batch3, /* row_kinds=*/ {}, merge_writer.get ());
1189- // Leveled compaction keeps files across levels, so file count may be > 1.
1190- ASSERT_LE (TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ), 2u );
1191- ASSERT_GE (TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ), 1u );
1188+ // After 3rd spill: 1 file at level 0 (new) + 1 file at level 1 (from prior compact) = 2.
1189+ ASSERT_EQ (2u , TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ));
11921190
11931191 ASSERT_OK_AND_ASSIGN (CommitIncrement commit_increment,
11941192 merge_writer->PrepareCommit (/* wait_compaction=*/ false ));
@@ -1207,7 +1205,7 @@ TEST_P(MergeTreeWriterTest, TestIntermediateMergeSpillFileBound) {
12071205 CheckFileContent (expected_data_file_path, expected_array);
12081206}
12091207
1210- TEST_P (MergeTreeWriterTest, TestDiskQuotaExhaustedFallsBackToFlushWriteBuffer) {
1208+ TEST_F (MergeTreeWriterTest, TestDiskQuotaExhaustedFallsBackToFlushWriteBuffer) {
12111209 ASSERT_OK_AND_ASSIGN (CoreOptions options,
12121210 CoreOptions::FromMap ({{Options::FILE_FORMAT , " orc" },
12131211 {Options::WRITE_BUFFER_SIZE , " 1" },
@@ -1281,7 +1279,7 @@ TEST_P(MergeTreeWriterTest, TestDiskQuotaExhaustedFallsBackToFlushWriteBuffer) {
12811279 }
12821280}
12831281
1284- TEST_P (MergeTreeWriterTest, TestFlushMemoryQuotaExhaustedFallsBackToFlushWriteBuffer) {
1282+ TEST_F (MergeTreeWriterTest, TestFlushMemoryQuotaExhaustedFallsBackToFlushWriteBuffer) {
12851283 // WRITE_BUFFER_SIZE is large enough so WriteBatch does NOT auto-spill.
12861284 // SPILL_MAX_DISK_SIZE is tiny so the first FlushMemory() exhausts the quota,
12871285 // triggering the fallback path: FlushMemory() -> quota exhausted -> FlushWriteBuffer.
@@ -1328,7 +1326,7 @@ TEST_P(MergeTreeWriterTest, TestFlushMemoryQuotaExhaustedFallsBackToFlushWriteBu
13281326 ASSERT_OK (merge_writer->Close ());
13291327}
13301328
1331- TEST_P (MergeTreeWriterTest, TestCloseDeletesSpillTempFiles) {
1329+ TEST_F (MergeTreeWriterTest, TestCloseDeletesSpillTempFiles) {
13321330 ASSERT_OK_AND_ASSIGN (CoreOptions options,
13331331 CoreOptions::FromMap ({{Options::FILE_FORMAT , " orc" },
13341332 {Options::WRITE_BUFFER_SIZE , " 1" },
@@ -1360,7 +1358,7 @@ TEST_P(MergeTreeWriterTest, TestCloseDeletesSpillTempFiles) {
13601358 ASSERT_EQ (0u , TestHelper::CountChannelFiles (file_system_, dir->Str () + " /tmp" ));
13611359}
13621360
1363- TEST_P (MergeTreeWriterTest, TestMultiplePrepareCommitWithSpill) {
1361+ TEST_F (MergeTreeWriterTest, TestMultiplePrepareCommitWithSpill) {
13641362 ASSERT_OK_AND_ASSIGN (
13651363 CoreOptions options,
13661364 CoreOptions::FromMap ({{Options::FILE_FORMAT , " orc" }, {Options::WRITE_ONLY , " true" }}));
0 commit comments