fix tests

Author: ChaomingZhangCN

src/paimon/common/sst/sst_file_io_test.cpp

@@ -52,7 +52,7 @@ class SstFileIOTest : public ::testing::Test {
     }
 
     void TearDown() override {
-        fs_->Delete(dir_->Str());
+        ASSERT_OK(fs_->Delete(dir_->Str()));
     }
 
  protected:
@@ -71,7 +71,7 @@ TEST_F(SstFileIOTest, TestSimple) {
     auto bf = BloomFilter::Create(30, 0.01);
     auto seg_for_bf = MemorySegment::AllocateHeapMemory(bf->ByteLength(), pool_.get());
     auto seg_ptr = std::make_shared<MemorySegment>(seg_for_bf);
-    bf->SetMemorySegment(seg_ptr);
+    ASSERT_OK(bf->SetMemorySegment(seg_ptr));
     auto writer = std::make_shared<SstFileWriter>(out, pool_, bf, 50);
     std::set<int32_t> value_hash;
     for (size_t i = 1; i <= 5; i++) {
@@ -120,13 +120,13 @@ TEST_F(SstFileIOTest, TestSimple) {
     auto entries = bloom_filter_handle->ExpectedEntries();
     auto offset = bloom_filter_handle->Offset();
     auto size = bloom_filter_handle->Size();
-    in->Seek(offset, SeekOrigin::FS_SEEK_SET);
+    ASSERT_OK(in->Seek(offset, SeekOrigin::FS_SEEK_SET));
     auto bloom_filer_bytes = Bytes::AllocateBytes(size, pool_.get());
-    in->Read(bloom_filer_bytes->data(), bloom_filer_bytes->size());
+    ASSERT_OK(in->Read(bloom_filer_bytes->data(), bloom_filer_bytes->size()));
     auto seg = MemorySegment::Wrap(std::move(bloom_filer_bytes));
     auto ptr = std::make_shared<MemorySegment>(seg);
     auto bloom_filter = std::make_shared<BloomFilter>(entries, size);
-    bloom_filter->SetMemorySegment(ptr);
+    ASSERT_OK(bloom_filter->SetMemorySegment(ptr));
     for (const auto& value : value_hash) {
         ASSERT_TRUE(bloom_filter->TestHash(value));
     }

src/paimon/common/utils/bit_set_test.cpp

@@ -34,9 +34,9 @@ TEST(BitSetTest, TestBitSet) {
     auto pool = GetDefaultPool();
     auto seg = MemorySegment::AllocateHeapMemory(1024, pool.get());
     auto ptr = std::make_shared<MemorySegment>(seg);
-    bit_set->SetMemorySegment(ptr);
+    ASSERT_OK(bit_set->SetMemorySegment(ptr));
     for (int i = 0; i < 100; i++) {
-        bit_set->Set(i * 2 + 1);
+        ASSERT_OK(bit_set->Set(i * 2 + 1));
     }
     for (int i = 0; i < 100; i++) {
         ASSERT_TRUE(bit_set->Get(i * 2 + 1));

src/paimon/common/utils/bloom_filter_test.cpp

@@ -35,15 +35,15 @@ TEST(BloomFilterTest, TestOneSegmentBuilder) {
     auto bloom_filter = BloomFilter::Create(items, 0.01);
     auto seg = MemorySegment::AllocateHeapMemory(1024, pool.get());
     auto ptr = std::make_shared<MemorySegment>(seg);
-    bloom_filter->SetMemorySegment(ptr);
+    ASSERT_OK(bloom_filter->SetMemorySegment(ptr));
 
     std::mt19937_64 engine(std::random_device{}());  // NOLINT(whitespace/braces)
     std::uniform_int_distribution<int32_t> distribution(0, items);
     std::set<int32_t> test_data;
     for (int32_t i = 0; i < items; i++) {
         int32_t random = distribution(engine);
         test_data.insert(random);
-        bloom_filter->AddHash(random);
+        ASSERT_OK(bloom_filter->AddHash(random));
     }
 
     for (const auto& value : test_data) {
@@ -109,13 +109,13 @@ TEST(BloomFilterTest, TestBloomFilter) {
     // segments 1
     auto seg1 = MemorySegment::AllocateHeapMemory(1024, pool.get());
     auto ptr1 = std::make_shared<MemorySegment>(seg1);
-    bloom_filter->SetMemorySegment(ptr1);
+    ASSERT_OK(bloom_filter->SetMemorySegment(ptr1));
 
     std::set<int32_t> test_data1;
     for (int32_t i = 0; i < items; i++) {
         int32_t random = distribution(engine);
         test_data1.insert(random);
-        bloom_filter->AddHash(random);
+        ASSERT_OK(bloom_filter->AddHash(random));
     }
     for (const auto& value : test_data1) {
         ASSERT_TRUE(bloom_filter->TestHash(value));
@@ -125,11 +125,11 @@ TEST(BloomFilterTest, TestBloomFilter) {
     std::set<int32_t> test_data2;
     auto seg2 = MemorySegment::AllocateHeapMemory(1024, pool.get());
     auto ptr2 = std::make_shared<MemorySegment>(seg2);
-    bloom_filter->SetMemorySegment(ptr2);
+    ASSERT_OK(bloom_filter->SetMemorySegment(ptr2));
     for (int32_t i = 0; i < items; i++) {
         int32_t random = distribution(engine);
         test_data2.insert(random);
-        bloom_filter->AddHash(random);
+        ASSERT_OK(bloom_filter->AddHash(random));
     }
     for (const auto& value : test_data2) {
         ASSERT_TRUE(bloom_filter->TestHash(value));
@@ -147,7 +147,7 @@ TEST(BloomFilterTest, TestBloomFilter) {
     }
 
     // switch to segment2 and clear
-    bloom_filter->SetMemorySegment(ptr2);
+    ASSERT_OK(bloom_filter->SetMemorySegment(ptr2));
     bloom_filter->Reset();
     for (const auto& value : test_data2) {
         ASSERT_FALSE(bloom_filter->TestHash(value));