tsan test fixes.

boostorg · Oct 19, 2023 · 210dce3 · 210dce3
1 parent 19f8487
commit 210dce3
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 25 deletions.
diff --git a/test/main.cpp b/test/main.cpp
@@ -1,17 +1,22 @@
-// Copyright (c) 2022 Klemens D. Morgenstern
-//
-// Distributed under the Boost Software License, Version 1.0. (See accompanying
-// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
-
 #include <boost/cobalt.hpp>
 #include <boost/cobalt/main.hpp>
-#include <boost/asio/steady_timer.hpp>
 
+namespace cobalt = boost::cobalt;
 
-boost::cobalt::main co_main(int argc, char *argv[])
+auto cobalt_task() -> cobalt::task<int>
 {
-  boost::asio::steady_timer tim{co_await boost::asio::this_coro::executor, std::chrono::milliseconds(50)};
-  co_await tim.async_wait(boost::cobalt::use_op);
-
   co_return 0;
 }
+
+
+cobalt::main co_main(int argc, char * argv[])
+{
+  constexpr std::size_t nb = 10000000;
+
+  for (std::size_t i = 0;i <nb; ++i)
+  {
+
+    co_await cobalt_task();
+  }
+  co_return 0;  
+}
diff --git a/test/strand.cpp b/test/strand.cpp
@@ -26,8 +26,6 @@ cobalt::promise<void> do_the_thing()
 
   asio::steady_timer tim{co_await cobalt::this_coro::executor, std::chrono::milliseconds(50)};
   co_await tim.async_wait(cobalt::use_op);
-
-
   unique++;
   BOOST_CHECK(unique == 1);
   unique--;
@@ -61,8 +59,6 @@ BOOST_AUTO_TEST_CASE(strand)
 
   for (auto & th : ths)
     th.join();
-
-
 }
 
 BOOST_AUTO_TEST_SUITE_END();

diff --git a/test/task.cpp b/test/task.cpp
@@ -264,7 +264,7 @@ CO_TEST_CASE(reawait)
 
 cobalt::task<int> test_strand1(asio::any_io_executor exec)
 {
-  BOOST_REQUIRE(exec == co_await cobalt::this_coro::executor);
+  BOOST_ASSERT(exec == co_await cobalt::this_coro::executor);
   co_await asio::post(co_await cobalt::this_coro::executor, cobalt::use_task);
   co_return 31;
 }
@@ -282,15 +282,15 @@ BOOST_AUTO_TEST_CASE(stranded)
 
   asio::thread_pool ctx;
   boost::cobalt::this_thread::set_executor(ctx.get_executor());
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { BOOST_CHECK(!ep); });
-  ctx.join();
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  cobalt::spawn(asio::make_strand(ctx.get_executor()), test_strand(),[](std::exception_ptr ep) { if (ep) std::rethrow_exception(ep); });
+  BOOST_CHECK_NO_THROW(ctx.join());
 }
 
 #endif

diff --git a/test/thread.cpp b/test/thread.cpp
@@ -38,7 +38,7 @@ boost::cobalt::thread thr_stop()
 #if !defined(BOOST_COBALT_USE_IO_CONTEXT)
   auto exec = co_await boost::asio::this_coro::executor;
   auto &exc = *exec.target<boost::asio::io_context::executor_type>();
-  BOOST_REQUIRE(&exc != nullptr);
+  BOOST_ASSERT(&exc != nullptr);
 #else
   auto exc = co_await boost::asio::this_coro::executor;
 #endif