1 files changed, 163 insertions, 0 deletions
diff --git a/examples/Reactor/WFMO_Reactor/test_suspended_removals.cpp b/examples/Reactor/WFMO_Reactor/test_suspended_removals.cpp
new file mode 100644
index 00000000000..4dc7cc33bb7
--- /dev/null
+++ b/examples/Reactor/WFMO_Reactor/test_suspended_removals.cpp
@@ -0,0 +1,163 @@
+// $Id$
+//
+// ============================================================================
+//
+// = LIBRARY
+//    examples
+//
+// = FILENAME
+//    test_suspended_removals.cpp
+//
+// = DESCRIPTION
+//
+//    Tests the Reactor's ability to handle removal of suspended
+//    handles.
+//
+// = AUTHOR
+//    Irfan Pyarali
+//
+// ============================================================================
+
+#include "ace/Reactor.h"
+#include "ace/WFMO_Reactor.h"
+#include "ace/Synch.h"
+
+ACE_RCSID(WFMO_Reactor, test_suspended_removals, "$Id$")
+
+class Event_Handler : public ACE_Event_Handler
+{
+public:
+
+  ACE_HANDLE get_handle (void) const
+  {
+    return this->event_.handle ();
+  }
+
+  ACE_Event event_;
+};
+
+class ACE_WFMO_Reactor_Test
+{
+public:
+  static void check_for_valid_state (ACE_WFMO_Reactor &wfmo_reactor,
+                                     size_t handles_to_be_added,
+                                     size_t handles_to_be_suspended,
+                                     size_t handles_to_be_resumed,
+                                     size_t handles_to_be_deleted)
+  {
+    ACE_ASSERT (wfmo_reactor.handler_rep_.handles_to_be_added_ == handles_to_be_added);
+    ACE_ASSERT (wfmo_reactor.handler_rep_.handles_to_be_suspended_ == handles_to_be_suspended);
+    ACE_ASSERT (wfmo_reactor.handler_rep_.handles_to_be_resumed_ == handles_to_be_resumed);
+    ACE_ASSERT (wfmo_reactor.handler_rep_.handles_to_be_deleted_ == handles_to_be_deleted);
+  }
+};
+
+int
+main (int argc, char *argv[])
+{
+  Event_Handler handler;
+  ACE_WFMO_Reactor reactor;
+  ACE_Reactor base_reactor (&reactor);
+  ACE_Time_Value time (1);
+
+  int result =
+    reactor.register_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                1, 0, 0, 0);
+
+  result =
+    reactor.remove_handler (&handler,
+                            ACE_Event_Handler::DONT_CALL |
+                            ACE_Event_Handler::ALL_EVENTS_MASK);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                1, 0, 0, 1);
+
+  result = base_reactor.run_reactor_event_loop (time);
+  ACE_ASSERT (result != -1);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 0);
+
+  result =
+    reactor.register_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                1, 0, 0, 0);
+
+  result = base_reactor.run_reactor_event_loop (time);
+  ACE_ASSERT (result != -1);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 0);
+
+  result =
+    reactor.suspend_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 1, 0, 0);
+
+  result =
+    reactor.remove_handler (&handler,
+                            ACE_Event_Handler::DONT_CALL |
+                            ACE_Event_Handler::ALL_EVENTS_MASK);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 1);
+
+  result = base_reactor.run_reactor_event_loop (time);
+  ACE_ASSERT (result != -1);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 0);
+
+  result =
+    reactor.register_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                1, 0, 0, 0);
+
+  result =
+    reactor.suspend_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                1, 1, 0, 0);
+
+  result = base_reactor.run_reactor_event_loop (time);
+  ACE_ASSERT (result != -1);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 0);
+
+  result =
+    reactor.resume_handler (&handler);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 1, 0);
+
+  result =
+    reactor.remove_handler (&handler,
+                            ACE_Event_Handler::DONT_CALL |
+                            ACE_Event_Handler::ALL_EVENTS_MASK);
+  ACE_ASSERT (result == 0);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 1);
+
+  result = base_reactor.run_reactor_event_loop (time);
+  ACE_ASSERT (result != -1);
+
+  ACE_WFMO_Reactor_Test::check_for_valid_state (reactor,
+                                                0, 0, 0, 0);
+
+  return 0;
+}