porting Clickhouse #49755, Fix unit test ExceptionFromWait (#787)

timeplus-io · Jul 26, 2024 · 53ac6ae · 53ac6ae
1 parent ccd37d6
commit 53ac6ae
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/Common/tests/gtest_thread_pool_schedule_exception.cpp b/src/Common/tests/gtest_thread_pool_schedule_exception.cpp
@@ -49,8 +49,17 @@ static bool check2()
 {
     ThreadPool pool(2);
 
-    pool.scheduleOrThrowOnError([&]{ throw std::runtime_error("Hello, world!"); });
-    pool.scheduleOrThrowOnError([]{});
+    try
+    {
+        pool.scheduleOrThrowOnError([&]{ throw std::runtime_error("Hello, world!"); });
+        pool.scheduleOrThrowOnError([]{});
+    }
+    catch (const std::runtime_error &)
+    {
+        /// Sometimes exception may be thrown from schedule.
+        /// Just retry test in that case.
+        return true;
+    }
 
     try
     {