1 files changed, 147 insertions, 0 deletions
diff --git a/ACE/TAO/tests/TransportCurrent/Framework/Current_Test_Impl.cpp b/ACE/TAO/tests/TransportCurrent/Framework/Current_Test_Impl.cpp
new file mode 100644
index 00000000000..0697c11ac4c
--- /dev/null
+++ b/ACE/TAO/tests/TransportCurrent/Framework/Current_Test_Impl.cpp
@@ -0,0 +1,147 @@
+// -*- C++ -*-
+
+#include "tao/TransportCurrent/TCC.h"
+
+#include "Current_Test_Impl.h"
+
+ACE_RCSID (Transport_Current,
+           Current_Test_Impl,
+           "$Id$")
+
+  Current_Test_Impl::Current_Test_Impl (CORBA::ORB_ptr orb,
+                                        PortableServer::POA_ptr poa,
+                                        int do_collocated_calls)
+    : orb_ (CORBA::ORB::_duplicate (orb))
+    , poa_ (PortableServer::POA::_duplicate (poa))
+    , do_collocated_calls_ (do_collocated_calls)
+{
+}
+
+Current_Test_Impl::~Current_Test_Impl (void)
+{
+}
+
+::CORBA::Long
+Current_Test_Impl::self_test (void)
+{
+  return 0;
+}
+
+void
+Current_Test_Impl::test_transport_current (void)
+{
+  if (TAO_debug_level > 2)
+    {
+      ACE_DEBUG ((LM_DEBUG,
+                  ACE_TEXT ("(%P|%t) server - Testing access to Transport Current ...\n")));
+    }
+
+  CORBA::Object_var tcobject =
+    orb_->resolve_initial_references ("TAO::Transport::Current");
+
+  TAO::Transport::Current_var tc =
+    TAO::Transport::Current::_narrow (tcobject.in ());
+
+  if (TAO_debug_level >= 1)
+    {
+      ::CORBA::Long id = tc->id ();
+
+      ::TAO::CounterT bs = tc->bytes_sent ();
+
+      ::TAO::CounterT br = tc->bytes_received ();
+
+      ::TAO::CounterT rs = tc->messages_sent ();
+
+      ::TAO::CounterT rr = tc->messages_received ();
+
+      ACE_DEBUG ((LM_DEBUG,
+                  ACE_TEXT ("Current_Test_Impl (%P|%t) Transport [%q] - Sent/Received [bytes=%q/%q, messages=%q/%q]\n"),
+                  (ACE_UINT64)id,
+                  (ACE_UINT64)bs,
+                  (ACE_UINT64)br,
+                  (ACE_UINT64)rs,
+                  (ACE_UINT64)rr));
+    }
+
+  if (TAO_debug_level > 2)
+    {
+      ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("(%P|%t) server - Successfully accessed Transport Current\n")));
+    }
+}
+
+void
+Current_Test_Impl::invoked_by_client (void)
+{
+  try
+    {
+      this->invoked_by_client_i ();
+    }
+  catch (const TAO::Transport::NoContext&)
+    {
+      ACE_ERROR ((LM_ERROR,
+                  ACE_TEXT ("(%P|%t) server - Caught a Transport::NoContext exception\n")));
+    }
+  catch (const CORBA::SystemException&)
+    {
+      ACE_ERROR ((LM_ERROR,
+                  ACE_TEXT ("(%P|%t) server - Caught a CORBA::SystemException exception\n")));
+    }
+}
+
+void
+Current_Test_Impl::invoked_by_client_i (void)
+{
+  if (TAO_debug_level > 2) {
+    ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("(%P|%t) server - Testing inside an upcall\n")));
+  }
+
+  this->test_transport_current ();
+
+  if (this->do_collocated_calls_)
+    {
+      CORBA::Object_var selfobject =
+        poa_->servant_to_reference (this);
+
+      Test::Transport::CurrentTest_var self =
+        Test::Transport::CurrentTest::_narrow (selfobject.in ());
+
+      if (TAO_debug_level >= 1)
+        {
+          ACE_DEBUG ((LM_DEBUG,
+                      ACE_TEXT ("(%P|%t) server - Testing a collocated call\n")));
+        }
+      self->invoked_during_upcall ();
+    }
+  else
+    {
+      if (TAO_debug_level >= 1)
+        {
+          ACE_DEBUG ((LM_DEBUG,
+                      ACE_TEXT ("(%P|%t) server - Skipping the collocated call test\n")));
+        }
+    }
+}
+
+void
+Current_Test_Impl::invoked_during_upcall (void)
+{
+  if (TAO_debug_level > 2) {
+    ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("(%P|%t) server - Test method invoked ")
+                ACE_TEXT ("by the server during upcall.\n")));
+  }
+
+}
+
+void
+Current_Test_Impl::shutdown (void)
+{
+  if (TAO_debug_level > 2) {
+    ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("(%P|%t) server - shutting down.\n")));
+  }
+
+  this->orb_->shutdown (0);
+}