commit f59e58f2024cfe12178d75784da23a7ddab1f089
author Teng Liang <philoliang2011@gmail.com> Fri Sep 07 16:41:54 2018 -0700
committer Teng Liang <philoliang2011@gmail.com> Mon Sep 10 12:42:01 2018 -0700
tree 38a96d7820948855339d7bd2e36a006d026da0ce
parent 52009043030df599db1bb6c9ee5ddde30c7f7aca

core: add a facility to execute functions on the main io_service

Change-Id: I38e5f4ad5ed6798a14e0075fb7a14b792f8b2413
refs: #4683

tests/core/global-io.t.cpp

diff --git a/tests/core/global-io.t.cpp b/tests/core/global-io.t.cpp
index 8ae9126..5aa3bc6 100644
--- a/tests/core/global-io.t.cpp
+++ b/tests/core/global-io.t.cpp
@@ -51,20 +51,31 @@
   BOOST_CHECK(s1 != s2);
 }
 
-BOOST_FIXTURE_TEST_CASE(RibIoService, RibIoFixture)
+BOOST_FIXTURE_TEST_CASE(MainRibIoService, RibIoFixture)
 {
   boost::asio::io_service* mainIo = &g_io;
   boost::asio::io_service* ribIo = g_ribIo;
 
   BOOST_CHECK(mainIo != ribIo);
   BOOST_CHECK(&getGlobalIoService() == mainIo);
+  BOOST_CHECK(&getMainIoService() == mainIo);
   BOOST_CHECK(&getRibIoService() == ribIo);
   auto mainThreadId = boost::this_thread::get_id();
 
   runOnRibIoService([&] {
     BOOST_CHECK(mainThreadId != boost::this_thread::get_id());
-    BOOST_CHECK(&getRibIoService() == ribIo);
     BOOST_CHECK(&getGlobalIoService() == ribIo);
+    BOOST_CHECK(&getMainIoService() == mainIo);
+    BOOST_CHECK(&getRibIoService() == ribIo);
+  });
+
+  runOnRibIoService([&] {
+    runOnMainIoService([&] {
+      BOOST_CHECK(mainThreadId == boost::this_thread::get_id());
+      BOOST_CHECK(&getGlobalIoService() == mainIo);
+      BOOST_CHECK(&getMainIoService() == mainIo);
+      BOOST_CHECK(&getRibIoService() == ribIo);
+    });
   });
 }
 
@@ -80,10 +91,10 @@
 
   hasRibRun = false;
   bool hasMainRun = false;
-  g_io.post([&] {
-      hasMainRun = true;
-      runOnRibIoService([&] { hasRibRun = true; });
-    });
+  runOnMainIoService([&] {
+    hasMainRun = true;
+    runOnRibIoService([&] { hasRibRun = true; });
+  });
   BOOST_CHECK_EQUAL(hasMainRun, false);
   BOOST_CHECK_EQUAL(hasRibRun, false);
 
@@ -105,9 +116,9 @@
   hasRibRun = false;
   bool hasMainRun = false;
   scheduler::schedule(250_ms, [&] {
-      hasMainRun = true;
-      runOnRibIoService([&] { hasRibRun = true; });
-    });
+    hasMainRun = true;
+    runOnRibIoService([&] { hasRibRun = true; });
+  });
   BOOST_CHECK_EQUAL(hasMainRun, false);
   BOOST_CHECK_EQUAL(hasRibRun, false);