ACE/tests/WFMO_Reactor_Test.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    WFMO_Reactor_Test.cpp
   5  *
   6  *  This is a simple test of the WFMO_Reactor.  It makes sure that
   7  *  removals and suspensions work correctly.
   8  *
   9  *  @author Irfan Pyarali <irfan@oomworks.com>
  10  */
  11 //=============================================================================
  12
  13 #include "test_config.h"
  14 #include "ace/Reactor.h"
  15 #include "ace/WFMO_Reactor.h"
  16 #include "ace/Pipe.h"
  17
  18 #if defined (ACE_WIN32)
  19
  20 static int number_of_handlers = 6;
  21 static int number_of_closes = 0;
  22
  23 class Event_Handler : public ACE_Event_Handler
  24 {
  25 public:
  26   Event_Handler (ACE_Reactor &reactor);
  27
  28   ~Event_Handler ();
  29
  30   ACE_Pipe pipe_;
  31 };
  32
  33 Event_Handler::Event_Handler (ACE_Reactor &reactor)
  34 {
  35   this->reference_counting_policy ().value
  36     (ACE_Event_Handler::Reference_Counting_Policy::ENABLED);
  37
  38   ACE_DEBUG ((LM_DEBUG,
  39               "Reference count in Event_Handler() is %d\n",
  40               this->reference_count_.load ()));
  41
  42   this->reactor (&reactor);
  43
  44   int result =
  45     this->pipe_.open ();
  46
  47   ACE_TEST_ASSERT (result == 0);
  48   ACE_UNUSED_ARG (result);
  49
  50   this->reactor ()->register_handler (this->pipe_.read_handle (),
  51                                       this,
  52                                       ACE_Event_Handler::READ_MASK);
  53   ACE_TEST_ASSERT (result == 0);
  54
  55   this->reactor ()->register_handler (this->pipe_.write_handle (),
  56                                       this,
  57                                       ACE_Event_Handler::READ_MASK);
  58   ACE_TEST_ASSERT (result == 0);
  59 }
  60
  61 Event_Handler::~Event_Handler ()
  62 {
  63   ACE_DEBUG ((LM_DEBUG,
  64               "Reference count in ~Event_Handler() is %d\n",
  65               this->reference_count_.load ()));
  66
  67   ++number_of_closes;
  68 }
  69
  70 void
  71 test ()
  72 {
  73   int result = 0;
  74   int i = 0;
  75
  76   ACE_Reactor reactor (new ACE_WFMO_Reactor, true);
  77
  78   ACE_Event_Handler_var *safe_event_handlers =
  79     new ACE_Event_Handler_var[number_of_handlers];
  80
  81   Event_Handler **event_handlers =
  82     new Event_Handler*[number_of_handlers];
  83
  84   for (i = 0; i < number_of_handlers; ++i)
  85     {
  86       event_handlers[i] =
  87         new Event_Handler (reactor);
  88
  89       safe_event_handlers[i] =
  90         event_handlers[i];
  91     }
  92
  93   ACE_Time_Value timeout (0, 500 * 1000);
  94
  95   result = reactor.run_reactor_event_loop (timeout);
  96   ACE_TEST_ASSERT (result != -1);
  97
  98   for (i = 0; i < number_of_handlers; ++i)
  99     {
 100       if (i % 2 == 0)
 101         continue;
 102
 103       result = reactor.suspend_handler (event_handlers[i]->pipe_.read_handle ());
 104       ACE_TEST_ASSERT (result == 0);
 105
 106       result = reactor.suspend_handler (event_handlers[i]->pipe_.write_handle ());
 107       ACE_TEST_ASSERT (result == 0);
 108     }
 109
 110   result = reactor.run_reactor_event_loop (timeout);
 111   ACE_TEST_ASSERT (result != -1);
 112
 113   delete[] safe_event_handlers;
 114   delete[] event_handlers;
 115 }
 116
 117 int
 118 run_main (int, ACE_TCHAR *[])
 119 {
 120   ACE_START_TEST (ACE_TEXT ("WFMO_Reactor_Test"));
 121
 122   test ();
 123
 124   ACE_TEST_ASSERT (number_of_closes == number_of_handlers);
 125
 126   ACE_END_TEST;
 127
 128   return 0;
 129 }
 130
 131 #else /* ACE_WIN32 */
 132
 133 int
 134 run_main (int, ACE_TCHAR *[])
 135 {
 136   ACE_START_TEST (ACE_TEXT ("WFMO_Reactor_Test"));
 137
 138   ACE_ERROR ((LM_INFO,
 139               ACE_TEXT ("WFMO_Reactor not supported on this platform\n")));
 140
 141   ACE_END_TEST;
 142
 143   return 0;
 144 }
 145
 146 #endif /* ACE_WIN32 */