1 files changed, 145 insertions, 0 deletions
diff --git a/ACE/TAO/tests/POA/wait_for_completion/wait_for_completion.cpp b/ACE/TAO/tests/POA/wait_for_completion/wait_for_completion.cpp
new file mode 100644
index 00000000000..c91fbc51fc9
--- /dev/null
+++ b/ACE/TAO/tests/POA/wait_for_completion/wait_for_completion.cpp
@@ -0,0 +1,145 @@
+// $Id$
+
+//========================================================================
+//
+// = LIBRARY
+//     TAO/tests/POA/wait_for_completion
+//
+// = FILENAME
+//     wait_for_completion.cpp
+//
+// = DESCRIPTION
+//     This program tests the wait_for_completion feature of a POA.
+//
+// = AUTHOR
+//     Irfan Pyarali
+//
+//=========================================================================
+
+#include "testS.h"
+
+class test_i : public POA_test
+{
+public:
+  void destroy_poa (void);
+
+  void test_poa (PortableServer::POA_ptr poa);
+
+  PortableServer::POA_var poa_;
+};
+
+void
+test_i::test_poa (PortableServer::POA_ptr poa)
+{
+  this->poa_ = PortableServer::POA::_duplicate (poa);
+}
+
+void
+test_i::destroy_poa (void)
+{
+  CORBA::Boolean etherealize_objects = 1;
+  CORBA::Boolean wait_for_completion = 1;
+  this->poa_->destroy (etherealize_objects,
+                       wait_for_completion);
+}
+
+PortableServer::POA_ptr
+init_orb (int argc,
+          ACE_TCHAR **argv,
+          const char *orb_name,
+          CORBA::ORB_ptr &orb_ptr)
+{
+  // Initialize the ORB first.
+  CORBA::ORB_var orb = CORBA::ORB_init (argc,
+                                        argv,
+                                        orb_name);
+
+  // Obtain the RootPOA.
+  CORBA::Object_var obj =
+    orb->resolve_initial_references ("RootPOA");
+
+  // Get the POA_var object from Object_var.
+  PortableServer::POA_var root_poa =
+    PortableServer::POA::_narrow (obj.in ());
+
+  // Get the POAManager of the RootPOA.
+  PortableServer::POAManager_var poa_manager =
+    root_poa->the_POAManager ();
+
+  poa_manager->activate ();
+
+  orb_ptr = orb._retn ();
+  return root_poa._retn ();
+}
+
+int
+ACE_TMAIN (int argc, ACE_TCHAR *argv[])
+{
+  try
+    {
+      CORBA::ORB_var orb1;
+      PortableServer::POA_var first_poa =
+        init_orb (argc,
+                  argv,
+                  "first ORB",
+                  orb1.inout ());
+
+      CORBA::ORB_var orb2;
+      PortableServer::POA_var second_poa =
+        init_orb (argc,
+                  argv,
+                  "second ORB",
+                  orb2.inout ());
+
+      test_i servant;
+      PortableServer::ObjectId_var id =
+        first_poa->activate_object (&servant);
+
+      CORBA::Object_var object_act = first_poa->id_to_reference (id.in ());
+
+      test_var test_object = test::_narrow (object_act.in ());
+
+      int expected_exception_raised = 0;
+
+      try
+        {
+          servant.test_poa (first_poa.in ());
+
+          test_object->destroy_poa ();
+        }
+      catch (const CORBA::BAD_INV_ORDER& )
+        {
+          // This is the correct exception! Ignore
+          expected_exception_raised = 1;
+        }
+      catch (const CORBA::Exception& ex)
+        {
+          ex._tao_print_exception (
+            "Exception caught of incorrect type");
+          return -1;
+        }
+
+      // Make sure an exception was raised and it was of the correct
+      // type.
+      ACE_ASSERT (expected_exception_raised);
+
+      // In non-debug compiles, asserts will disappear.
+      ACE_UNUSED_ARG (expected_exception_raised);
+
+      servant.test_poa (second_poa.in ());
+
+      test_object->destroy_poa ();
+
+      first_poa->destroy (1, 1);
+
+      orb1->destroy ();
+      orb2->destroy ();
+    }
+  catch (const CORBA::Exception& ex)
+    {
+      ex._tao_print_exception ("Unexpected exception caught");
+      return -1;
+    }
+
+  return 0;
+}