TAO/tests/Portable_Interceptors/PICurrent/client.cpp

   1 // -*- C++ -*-
   2 #include "ace/Get_Opt.h"
   3 #include "ace/OS_NS_unistd.h"
   4
   5 #include "testC.h"
   6 #include "ClientORBInitializer.h"
   7
   8 #include "tao/ORBInitializer_Registry.h"
   9 #include "tao/PI/PI.h"
  10
  11 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
  12
  13 int
  14 parse_args (int argc, ACE_TCHAR *argv[])
  15 {
  16   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:"));
  17   int c;
  18
  19   while ((c = get_opts ()) != -1)
  20     switch (c)
  21       {
  22       case 'k':
  23         ior = get_opts.opt_arg ();
  24         break;
  25       default:
  26         ACE_ERROR_RETURN ((LM_ERROR,
  27                            "Usage: %s "
  28                            "-k IOR "
  29                            "\n",
  30                            argv[0]),
  31                           -1);
  32       }
  33   return 0;
  34 }
  35
  36 int
  37 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  38 {
  39   try
  40     {
  41       PortableInterceptor::ORBInitializer_ptr temp_initializer =
  42         PortableInterceptor::ORBInitializer::_nil ();
  43
  44       ACE_NEW_RETURN (temp_initializer,
  45                       ClientORBInitializer,
  46                       -1);  // No exceptions yet!
  47       PortableInterceptor::ORBInitializer_var orb_initializer =
  48         temp_initializer;
  49
  50       PortableInterceptor::register_orb_initializer (orb_initializer.in ());
  51
  52       CORBA::ORB_var orb =
  53         CORBA::ORB_init (argc,
  54                          argv,
  55                          "client_orb");
  56
  57       if (parse_args (argc, argv) != 0)
  58         return 1;
  59
  60       // Get the PICurrent object.
  61       CORBA::Object_var obj =
  62         orb->resolve_initial_references ("PICurrent");
  63
  64       PortableInterceptor::Current_var pi_current =
  65         PortableInterceptor::Current::_narrow (obj.in ());
  66
  67       if (CORBA::is_nil (pi_current.in ()))
  68         {
  69           ACE_ERROR_RETURN ((LM_ERROR,
  70                              "(%P|%t) ERROR: Could not resolve "
  71                              "PICurrent object.\n"),
  72                             -1);
  73         }
  74
  75       // Insert some data into the allocated PICurrent slot.
  76       CORBA::Any data;
  77       CORBA::Long number = 46;
  78
  79       data <<= number;
  80
  81       // Now reset the contents of our slot in the thread-scope
  82       // current (TSC).
  83       pi_current->set_slot (::slot_id,
  84                             data);
  85
  86       // Resolve the target object, and perform the invocation.
  87       obj =
  88         orb->string_to_object (ior);
  89
  90       PICurrentTest::test_var server =
  91         PICurrentTest::test::_narrow (obj.in ());
  92
  93       if (CORBA::is_nil (server.in ()))
  94         {
  95           ACE_ERROR_RETURN ((LM_ERROR,
  96                              "Object reference <%s> is nil.\n",
  97                              ior),
  98                             1);
  99         }
 100
 101       // BUG 2656 testing - _get_policy_overrides() should return an empty
 102       // sequence rather than nill.
 103       CORBA::PolicyTypeSeq types;
 104       CORBA::PolicyList_var policies = server->_get_policy_overrides(types);
 105
 106       if (policies.ptr () == 0)
 107         {
 108             ACE_ERROR ((LM_ERROR,
 109                         "(%P|%t) _get_policy_overrides returned nill pointer\n"));
 110             throw CORBA::INTERNAL ();
 111         }
 112       else
 113         {
 114           CORBA::ULong const list_size = policies->length();
 115           if (list_size != 0)
 116             {
 117               ACE_ERROR ((LM_ERROR,
 118                           "(%P|%t) _get_policy_overrides returned list with size not equal 0\n"));
 119               throw CORBA::INTERNAL ();
 120             }
 121         }
 122
 123       server->invoke_me ();
 124
 125       CORBA::Any_var new_data =
 126         pi_current->get_slot (::slot_id);
 127
 128       // The original data in the TSC was of type CORBA::Long.  If the
 129       // following extraction from the CORBA::Any fails, then the
 130       // original data in the TSC was not replaced within the client
 131       // request interceptor, as this test should do.
 132       const char *str = 0;
 133       if (new_data.in () >>= str)
 134         {
 135           ACE_DEBUG ((LM_DEBUG,
 136                       "(%P|%t) Retrieved \"%C\" from the TSC.\n",
 137                       str));
 138         }
 139       else
 140         {
 141           ACE_ERROR ((LM_ERROR,
 142                       "(%P|%t) Unable to extract data (a string) "
 143                       "from the TSC.\n"));
 144
 145           throw CORBA::INTERNAL ();
 146         }
 147
 148       server->invoke_we ();
 149
 150       server->shutdown ();
 151
 152       ACE_OS::sleep(1);
 153       orb->destroy ();
 154     }
 155   catch (const CORBA::Exception& ex)
 156     {
 157       ex._tao_print_exception ("PICurrent test (client-side):");
 158       return -1;
 159     }
 160
 161   return 0;
 162 }