Add unit test for thread_pool.

asio/src/Makefile.msc

@@ -200,6 +200,7 @@ UNIT_TEST_EXES = \
 	tests\unit\system_timer.exe \
 	tests\unit\this_coro.exe \
 	tests\unit\thread.exe \
+	tests\unit\thread_pool.exe \
 	tests\unit\time_traits.exe \
 	tests\unit\ts\buffer.exe \
 	tests\unit\ts\executor.exe \

asio/src/tests/Makefile.am

@@ -104,6 +104,7 @@ check_PROGRAMS = \
 	unit/system_timer \
 	unit/this_coro \
 	unit/thread \
+	unit/thread_pool \
 	unit/time_traits \
 	unit/ts/buffer \
 	unit/ts/executor \
@@ -239,6 +240,7 @@ TESTS = \
 	unit/system_timer \
 	unit/this_coro \
 	unit/thread \
+	unit/thread_pool \
 	unit/time_traits \
 	unit/ts/buffer \
 	unit/ts/executor \
@@ -384,6 +386,7 @@ unit_system_executor_SOURCES = unit/system_executor.cpp
 unit_system_timer_SOURCES = unit/system_timer.cpp
 unit_this_coro_SOURCES = unit/this_coro.cpp
 unit_thread_SOURCES = unit/thread.cpp
+unit_thread_pool_SOURCES = unit/thread_pool.cpp
 unit_time_traits_SOURCES = unit/time_traits.cpp
 unit_ts_buffer_SOURCES = unit/ts/buffer.cpp
 unit_ts_executor_SOURCES = unit/ts/executor.cpp

asio/src/tests/unit/.gitignore

@@ -67,6 +67,7 @@ system_executor
 system_timer
 this_coro
 thread
+thread_pool
 time_traits
 use_awaitable
 use_future

asio/src/tests/unit/thread_pool.cpp

@@ -0,0 +1,166 @@
+//
+// thread_pool.cpp
+// ~~~~~~~~~~~~~~~
+//
+// Copyright (c) 2003-2020 Christopher M. Kohlhoff (chris at kohlhoff dot com)
+//
+// 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)
+//
+
+// Disable autolinking for unit tests.
+#if !defined(BOOST_ALL_NO_LIB)
+#define BOOST_ALL_NO_LIB 1
+#endif // !defined(BOOST_ALL_NO_LIB)
+
+// Test that header file is self-contained.
+#include "asio/thread_pool.hpp"
+
+#include "asio/dispatch.hpp"
+#include "asio/post.hpp"
+#include "unit_test.hpp"
+
+#if defined(ASIO_HAS_BOOST_BIND)
+# include <boost/bind/bind.hpp>
+#else // defined(ASIO_HAS_BOOST_BIND)
+# include <functional>
+#endif // defined(ASIO_HAS_BOOST_BIND)
+
+using namespace asio;
+
+#if defined(ASIO_HAS_BOOST_BIND)
+namespace bindns = boost;
+#else // defined(ASIO_HAS_BOOST_BIND)
+namespace bindns = std;
+#endif
+
+void increment(int* count)
+{
+  ++(*count);
+}
+
+void decrement_to_zero(thread_pool* pool, int* count)
+{
+  if (*count > 0)
+  {
+    --(*count);
+
+    int before_value = *count;
+    asio::post(*pool, bindns::bind(decrement_to_zero, pool, count));
+
+    // Handler execution cannot nest, so count value should remain unchanged.
+    ASIO_CHECK(*count == before_value);
+  }
+}
+
+void nested_decrement_to_zero(thread_pool* pool, int* count)
+{
+  if (*count > 0)
+  {
+    --(*count);
+
+    asio::dispatch(*pool,
+        bindns::bind(nested_decrement_to_zero, pool, count));
+
+    // Handler execution is nested, so count value should now be zero.
+    ASIO_CHECK(*count == 0);
+  }
+}
+
+void thread_pool_test()
+{
+  thread_pool pool(1);
+
+  int count1 = 0;
+  asio::post(pool, bindns::bind(increment, &count1));
+
+  int count2 = 10;
+  asio::post(pool, bindns::bind(decrement_to_zero, &pool, &count2));
+
+  int count3 = 10;
+  asio::post(pool, bindns::bind(nested_decrement_to_zero, &pool, &count3));
+
+  pool.join();
+
+  ASIO_CHECK(count1 == 1);
+  ASIO_CHECK(count2 == 0);
+  ASIO_CHECK(count3 == 0);
+}
+
+class test_service : public asio::execution_context::service
+{
+public:
+  typedef test_service key_type;
+
+  test_service(asio::execution_context& ctx)
+    : asio::execution_context::service(ctx)
+  {
+  }
+
+private:
+  virtual void shutdown() {}
+};
+
+void thread_pool_service_test()
+{
+  asio::thread_pool pool1(1);
+  asio::thread_pool pool2(1);
+  asio::thread_pool pool3(1);
+
+  // Implicit service registration.
+
+  asio::use_service<test_service>(pool1);
+
+  ASIO_CHECK(asio::has_service<test_service>(pool1));
+
+  test_service* svc1 = new test_service(pool1);
+  try
+  {
+    asio::add_service(pool1, svc1);
+    ASIO_ERROR("add_service did not throw");
+  }
+  catch (asio::service_already_exists&)
+  {
+  }
+  delete svc1;
+
+  // Explicit service registration.
+
+  test_service& svc2 = asio::make_service<test_service>(pool2);
+
+  ASIO_CHECK(asio::has_service<test_service>(pool2));
+  ASIO_CHECK(&asio::use_service<test_service>(pool2) == &svc2);
+
+  test_service* svc3 = new test_service(pool2);
+  try
+  {
+    asio::add_service(pool2, svc3);
+    ASIO_ERROR("add_service did not throw");
+  }
+  catch (asio::service_already_exists&)
+  {
+  }
+  delete svc3;
+
+  // Explicit registration with invalid owner.
+
+  test_service* svc4 = new test_service(pool2);
+  try
+  {
+    asio::add_service(pool3, svc4);
+    ASIO_ERROR("add_service did not throw");
+  }
+  catch (asio::invalid_service_owner&)
+  {
+  }
+  delete svc4;
+
+  ASIO_CHECK(!asio::has_service<test_service>(pool3));
+}
+
+ASIO_TEST_SUITE
+(
+  "thread_pool",
+  ASIO_TEST_CASE(thread_pool_test)
+  ASIO_TEST_CASE(thread_pool_service_test)
+)