fix ut regression

pingcap · May 24, 2022 · 6bc27c0 · 6bc27c0
1 parent 158403c
commit 6bc27c0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/dbms/src/Storages/Page/V3/tests/entries_helper.h b/dbms/src/Storages/Page/V3/tests/entries_helper.h
@@ -234,6 +234,14 @@ inline ::testing::AssertionResult getEntryNotExist(
             id_entry.first.low,
             toDebugString(id_entry.second));
     }
+    catch (DB::Exception & ex)
+    {
+        if (ex.code() == ErrorCodes::PS_ENTRY_NOT_EXISTS || ex.code() == ErrorCodes::PS_ENTRY_NO_VALID_VERSION)
+            return ::testing::AssertionSuccess();
+        else
+            error = ex.displayText();
+        return ::testing::AssertionFailure(::testing::Message(error.c_str()));
+    }
     catch (...)
     {
         error = getCurrentExceptionMessage(true);