From f93eae6534623e1dd5842cb246d2bad72820a69f Mon Sep 17 00:00:00 2001 From: Xiaoxuan Meng Date: Mon, 21 Oct 2024 23:17:14 -0700 Subject: [PATCH] Fix the flaky arbitrationAbort test (#11316) Summary: Pull Request resolved: https://github.com/facebookincubator/velox/pull/11316 The doesn't require timeout setting and also only work for debug build Reviewed By: amitkdutta, tanjialiang Differential Revision: D64736824 fbshipit-source-id: 9b9d31fe51810f8ef84e6325a7d0d364b4677676 --- velox/common/memory/tests/MockSharedArbitratorTest.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/velox/common/memory/tests/MockSharedArbitratorTest.cpp b/velox/common/memory/tests/MockSharedArbitratorTest.cpp index 0f5d45fc58ff..9d819482df26 100644 --- a/velox/common/memory/tests/MockSharedArbitratorTest.cpp +++ b/velox/common/memory/tests/MockSharedArbitratorTest.cpp @@ -2219,10 +2219,9 @@ TEST_F(MockSharedArbitrationTest, ensureNodeMaxCapacity) { } } -TEST_F(MockSharedArbitrationTest, arbitrationAbort) { +DEBUG_ONLY_TEST_F(MockSharedArbitrationTest, arbitrationAbort) { uint64_t memoryCapacity = 256 * MB; - setupMemory( - memoryCapacity, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1.0, nullptr, true, 1'000); + setupMemory(memoryCapacity); std::shared_ptr task1 = addTask(memoryCapacity); auto* op1 = task1->addMemoryOp(true, [&](MemoryPool* /*unsed*/, uint64_t /*unsed*/) {