Fix flaky allocationFromFreeReservedMemoryDuringGlobalArbitration (#1…

…1346) Summary: Pull Request resolved: #11346 Reviewed By: kagamiori Differential Revision: D64922211 fbshipit-source-id: fbcb6b8a3146ac4d03811385ac0f1dc184f7ae5c
facebookincubator · Oct 24, 2024 · cd58af0 · cd58af0
1 parent 3fa7acf
commit cd58af0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/velox/common/memory/tests/MockSharedArbitratorTest.cpp b/velox/common/memory/tests/MockSharedArbitratorTest.cpp
@@ -1543,15 +1543,15 @@ DEBUG_ONLY_TEST_F(
         runtimeStats[SharedArbitrator::kMemoryArbitrationWallNanos].count, 1);
     ASSERT_GT(
         runtimeStats[SharedArbitrator::kMemoryArbitrationWallNanos].sum, 0);
-    ASSERT_EQ(
-        runtimeStats[SharedArbitrator::kGlobalArbitrationWaitCount].count, 1);
-    ASSERT_EQ(
-        runtimeStats[SharedArbitrator::kGlobalArbitrationWaitCount].sum, 1);
+    ASSERT_GT(
+        runtimeStats[SharedArbitrator::kGlobalArbitrationWaitCount].count, 0);
+    ASSERT_GT(
+        runtimeStats[SharedArbitrator::kGlobalArbitrationWaitCount].sum, 0);
     ASSERT_EQ(runtimeStats[SharedArbitrator::kLocalArbitrationCount].count, 0);
     ASSERT_EQ(runtimeStats[SharedArbitrator::kLocalArbitrationCount].sum, 0);
-    ASSERT_EQ(
+    ASSERT_GT(
         runtimeStats[SharedArbitrator::kGlobalArbitrationWaitWallNanos].count,
-        1);
+        0);
     ASSERT_GT(
         runtimeStats[SharedArbitrator::kGlobalArbitrationWaitWallNanos].sum,
         1'000'000'000);