add ut

Signed-off-by: luohaha <[email protected]>
StarRocks · Feb 11, 2025 · a0c4903 · a0c4903
1 parent 90fb1d7
commit a0c4903
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/be/src/storage/lake/delta_writer.cpp b/be/src/storage/lake/delta_writer.cpp
@@ -334,7 +334,7 @@ inline Status DeltaWriterImpl::flush_async() {
                         }
                         VLOG(2) << "flush memtable, tablet=" << _tablet_id << ", txn=" << _txn_id
                                 << " _immutable_tablet_size=" << _immutable_tablet_size
-                                << ", segment_size=" << (seg ? seg->data_size() : 0)
+                                << ", flush data size=" << (seg ? seg->data_size() : flush_data_size)
                                 << ", in_writing_data_size=" << _tablet_manager->in_writing_data_size(_tablet_id)
                                 << ", is_immutable=" << _is_immutable.load(std::memory_order_relaxed);
                     }

diff --git a/be/test/storage/lake/async_delta_writer_test.cpp b/be/test/storage/lake/async_delta_writer_test.cpp
@@ -594,6 +594,7 @@ TEST_F(LakeAsyncDeltaWriterTest, test_block_merger) {
             latch.count_down();
         });
     }
+    ASSERT_TRUE(_tablet_mgr->in_writing_data_size(tablet_id) > 0);
     latch.wait();
     config::write_buffer_size = old_val;
     config::enable_load_spill = old_val2;

diff --git a/be/test/storage/memtable_flush_executor_test.cpp b/be/test/storage/memtable_flush_executor_test.cpp
@@ -394,7 +394,7 @@ TEST_F(MemTableFlushExecutorTest, testMemtableFlushStatusNotOk) {
     flush_token->set_status(Status::NotSupported("Not Suppoted"));
     ASSERT_FALSE(flush_token->status().ok());
 
-    flush_token->_flush_memtable(nullptr, nullptr, false);
+    flush_token->_flush_memtable(nullptr, nullptr, false, nullptr);
 
     ASSERT_TRUE(MemTableFlushExecutor::calc_max_threads_for_lake_table(data_dirs) > 0);
 }