testsuite/supernova/ticket_scheduler_test.cpp

   1 #include <boost/test/unit_test.hpp>
   2 #include <boost/thread.hpp>
   3
   4 #define HAVE_SEMAPHORE_H
   5 #include "utilities/ticket_scheduler.hpp"
   6
   7
   8 namespace
   9 {
  10
  11 int thread_count = 5;
  12
  13 nova::ticket_scheduler sched;
  14
  15 void test_fn(std::size_t check)
  16 {
  17     static std::size_t count = 0;
  18
  19     BOOST_REQUIRE_EQUAL(count, check);
  20     ++count;
  21 }
  22
  23 const int iterations = 150;
  24
  25
  26 void runme_nonblocking(void)
  27 {
  28     for (int i = 0; i != iterations; ++i)
  29     {
  30         std::size_t ticket = sched.get_ticket();
  31
  32         boost::thread::yield();
  33
  34         for(;;)
  35         {
  36             if (sched.run_nonblocking(ticket, boost::bind(test_fn, ticket)))
  37                 break;
  38             else
  39                 boost::thread::yield();
  40         }
  41     }
  42 }
  43
  44
  45 void runme(void)
  46 {
  47     for (int i = 0; i != iterations; ++i)
  48     {
  49         std::size_t ticket = sched.get_ticket();
  50
  51         boost::thread::yield();
  52
  53         sched.run(ticket, boost::bind(test_fn, ticket));
  54     }
  55 }
  56
  57 }
  58
  59
  60 BOOST_AUTO_TEST_CASE( ticket_scheduler_test )
  61 {
  62     {
  63         boost::thread_group threads;
  64
  65         for (int i = 0; i != thread_count; ++i)
  66         {
  67             threads.create_thread(runme_nonblocking);
  68         }
  69
  70         threads.join_all();
  71     }
  72
  73     {
  74         boost::thread_group threads;
  75
  76         for (int i = 0; i != thread_count; ++i)
  77         {
  78             threads.create_thread(runme);
  79         }
  80
  81         threads.join_all();
  82     }
  83 }