TAO/tests/Permanent_Forward/StubTest.cpp

   1 //========================================================================
   2 /**
   3  * @file  StubTest.cpp
   4  *
   5  * This program tests the basic functionality of the TAO_Stub in case
   6  * of permanent_forward
   7  *
   8  * @author Frank Rehberger
   9  */
  10 //=========================================================================
  11
  12
  13 #include "ace/Log_Msg.h"
  14 #include "ace/OS_NS_string.h"
  15 #include "tao/ORB.h"
  16 #include "tao/Environment.h"
  17 #include "tao/SystemException.h"
  18 #include "tao/Object.h"
  19 #include "tao/Stub.h"
  20 #include "tao/Profile.h"
  21 #include "tao/IIOP_Profile.h"
  22
  23 #ifndef FRANKS_ASSERT
  24 #define FRANKS_ASSERT(X) \
  25   ((X)                                                                  \
  26    ? static_cast<void>(0)                                               \
  27    : ACE_VERSIONED_NAMESPACE_NAME::__ace_assert(__FILE__, __LINE__, ACE_TEXT_CHAR_TO_TCHAR (#X)))
  28 #endif
  29
  30 static bool
  31 is_endpoint (TAO_Profile *profile, const char *host, unsigned short port)
  32   {
  33     TAO_IIOP_Profile *iiop = dynamic_cast<TAO_IIOP_Profile*> (profile);
  34     TAO_Endpoint *endpoint = iiop->endpoint ();
  35     TAO_IIOP_Endpoint *iiop_endpoint = dynamic_cast<TAO_IIOP_Endpoint*> (endpoint);
  36     const char     * endpoint_host = iiop_endpoint->host();
  37     unsigned short   endpoint_port = iiop_endpoint->port();
  38
  39     bool retval =
  40       ACE_OS::strcmp (endpoint_host, host)==0
  41       && endpoint_port == port;
  42
  43     return retval;
  44   }
  45
  46 static bool
  47 equal_endpoint (TAO_Profile *profile, TAO_Profile *other)
  48   {
  49     TAO_IIOP_Profile *iiop = dynamic_cast<TAO_IIOP_Profile*> (profile);
  50     TAO_Endpoint *endpoint = iiop->endpoint ();
  51     TAO_IIOP_Endpoint *iiop_endpoint = dynamic_cast<TAO_IIOP_Endpoint*> (endpoint);
  52     const char     * endpoint_host = iiop_endpoint->host();
  53     unsigned short   endpoint_port = iiop_endpoint->port();
  54
  55     TAO_IIOP_Profile *other_iiop = dynamic_cast<TAO_IIOP_Profile*> (other);
  56     TAO_Endpoint *other_endpoint = other_iiop->endpoint ();
  57     TAO_IIOP_Endpoint *other_iiop_endpoint = dynamic_cast<TAO_IIOP_Endpoint*> (other_endpoint);
  58     const char     * other_endpoint_host = other_iiop_endpoint->host();
  59     unsigned short   other_endpoint_port = other_iiop_endpoint->port();
  60
  61     bool retval =
  62       ACE_OS::strcmp (endpoint_host, other_endpoint_host)==0
  63       && endpoint_port == other_endpoint_port;
  64
  65     return retval;
  66   }
  67
  68 static bool
  69 marshaled_equal_to_other (CORBA::ORB_ptr orb, CORBA::Object_ptr obj, TAO_MProfile *other_mprofile)
  70 {
  71   CORBA::String_var str = orb->object_to_string (obj);
  72
  73   CORBA::Object_var obj_copy = orb->string_to_object (str.in ());
  74
  75   TAO_Stub *stub_copy = obj_copy->_stubobj ();
  76   TAO_MProfile *copy_mprofile  = &(stub_copy->base_profiles ());
  77
  78   if ( copy_mprofile->size() != other_mprofile->size())
  79     return false;
  80
  81   for (size_t i=0; i<copy_mprofile->size(); ++i)
  82     {
  83       TAO_Profile *copy_profile  = copy_mprofile->get_profile (i);
  84       TAO_Profile *other_profile = other_mprofile->get_profile (i);
  85
  86       if ( ! equal_endpoint (copy_profile, other_profile) )
  87         return false;
  88     }
  89
  90   return true;
  91 }
  92
  93 static void
  94 test_forward_permanent (CORBA::ORB_ptr orb)
  95 {
  96   CORBA::Object_var obj1 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:1111,iiop:192.168.1.3:1111/NameService");
  97
  98   // the permanent profile
  99   CORBA::Object_var obj4 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:4444,iiop:192.168.1.3:4444/NameService");
 100
 101   TAO_Stub *stub1 = obj1->_stubobj ();
 102   TAO_Stub *stub4 = obj4->_stubobj ();
 103
 104   TAO_Profile *profile = 0;
 105
 106   profile = stub1->profile_in_use ();
 107
 108   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 1111));
 109
 110   // ----- forward to obj2 permanently
 111   stub1->add_forward_profiles (stub4->base_profiles(), true /* permanent */ );
 112
 113   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 114
 115   profile = stub1->next_profile ();
 116
 117   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 4444));
 118
 119   // ----- stringified object reference must be equal to obj4->base_prpfiles().
 120   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub4->base_profiles())) );
 121
 122   // ----- consume second profile of obj2
 123
 124   profile = stub1->next_profile ();
 125
 126   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 4444));
 127
 128
 129   // ----- reached end, next_profile() must yield NULL
 130
 131   profile = stub1->next_profile ();
 132
 133   FRANKS_ASSERT (profile == 0);
 134
 135   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 136 }
 137
 138 static void
 139 test_forward_permanent_mix (CORBA::ORB_ptr orb)
 140 {
 141   CORBA::Object_var obj1 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:1111,iiop:192.168.1.3:1111/NameService");
 142
 143   CORBA::Object_var obj2 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:2222,iiop:192.168.1.3:2222/NameService");
 144
 145   CORBA::Object_var obj3 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:3333,iiop:192.168.1.3:3333/NameService");
 146
 147   CORBA::Object_var obj4 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:4444,iiop:192.168.1.3:4444/NameService");
 148
 149   CORBA::Object_var obj5 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:5555,iiop:192.168.1.3:5555/NameService");
 150
 151   TAO_Stub *stub1 = obj1->_stubobj ();
 152   TAO_Stub *stub2 = obj2->_stubobj ();
 153   TAO_Stub *stub3 = obj3->_stubobj ();
 154   TAO_Stub *stub4 = obj4->_stubobj ();
 155   TAO_Stub *stub5 = obj5->_stubobj ();
 156
 157   TAO_Profile *profile = 0;
 158
 159   profile = stub1->profile_in_use ();
 160
 161   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 1111));
 162
 163   // ----- forward to obj2
 164   stub1->add_forward_profiles (stub2->base_profiles(), false /* permanent */ );
 165
 166   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 167
 168   profile = stub1->next_profile ();
 169
 170   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 2222));
 171
 172   // ----- forward to obj3
 173   stub1->add_forward_profiles (stub3->base_profiles(), false /* permanent */ );
 174
 175   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 176
 177   profile = stub1->next_profile ();
 178
 179   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 3333));
 180
 181   // ----- stringified object reference must be equal to obj1->base_prpfiles().
 182   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub1->base_profiles())) );
 183
 184   // ----- forward to obj4 permanently
 185   stub1->add_forward_profiles (stub4->base_profiles(), true /* permanent */ );
 186
 187   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 188
 189   profile = stub1->next_profile ();
 190
 191   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 4444));
 192
 193   // ----- stringified object reference must be equal to obj4->base_prpfiles().
 194   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub4->base_profiles())) );
 195
 196   // ----- consume second profile from obj4
 197   profile = stub1->next_profile ();
 198
 199   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 4444));
 200
 201   // ----- forward to obj2
 202   stub1->add_forward_profiles (stub2->base_profiles(), false /* permanent */ );
 203
 204   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 205
 206   profile = stub1->next_profile ();
 207
 208   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 2222));
 209
 210   // ----- forward to obj3
 211   stub1->add_forward_profiles (stub3->base_profiles(), false /* permanent */ );
 212
 213   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 214
 215   profile = stub1->next_profile ();
 216
 217   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 3333));
 218
 219   // ----- stringified object reference must be equal to obj4->base_prpfiles().
 220   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub4->base_profiles())) );
 221
 222   // ----- forward to obj5 permanently
 223   stub1->add_forward_profiles (stub5->base_profiles(), true /* permanent */ );
 224
 225   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 226
 227   profile = stub1->next_profile ();
 228
 229   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 5555));
 230
 231   // ----- stringified object reference must be equal to obj5->base_prpfiles().
 232   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub5->base_profiles())) );
 233
 234   // ----- consume second profile from obj5
 235   profile = stub1->next_profile ();
 236
 237   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 5555));
 238
 239   // ----- stringified object reference must be equal to obj1->base_prpfiles().
 240   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub5->base_profiles())) );
 241
 242   // reached end of profiles, next_profile must return NULL
 243
 244   profile = stub1->next_profile ();
 245
 246   FRANKS_ASSERT (profile == 0);
 247 }
 248
 249 static void
 250 test_forward (CORBA::ORB_ptr orb)
 251 {
 252   CORBA::Object_var obj1 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:1111,iiop:192.168.1.3:1111/NameService");
 253
 254   CORBA::Object_var obj2 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:2222,iiop:192.168.1.3:2222/NameService");
 255
 256   CORBA::Object_var obj3 = orb->string_to_object ("corbaloc:iiop:192.168.1.2:3333,iiop:192.168.1.3:3333/NameService");
 257
 258   TAO_Stub *stub1 = obj1->_stubobj ();
 259   TAO_Stub *stub2 = obj2->_stubobj ();
 260   TAO_Stub *stub3 = obj3->_stubobj ();
 261
 262   TAO_Profile *profile = 0;
 263
 264   profile = stub1->profile_in_use ();
 265
 266   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 1111));
 267
 268   // ----- forward to obj2
 269   stub1->add_forward_profiles (stub2->base_profiles(), false /* permanent */ );
 270
 271   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 272
 273   profile = stub1->next_profile ();
 274
 275   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 2222));
 276
 277   // ----- forward to obj3
 278
 279   stub1->add_forward_profiles (stub3->base_profiles(), false /* permanent */ );
 280
 281   FRANKS_ASSERT (stub1->forward_profiles () != 0);
 282
 283   profile = stub1->next_profile ();
 284
 285   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.2", 3333));
 286
 287   // ----- stringified object reference must be equal to obj1->base_prpfiles().
 288   FRANKS_ASSERT (marshaled_equal_to_other (orb, obj1.in (), &(stub1->base_profiles())) );
 289
 290   // ----- consume second profile of obj3
 291
 292   profile = stub1->next_profile ();
 293
 294   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 3333));
 295
 296   // ----- consume second profile of obj2
 297
 298   profile = stub1->next_profile ();
 299
 300   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 2222));
 301
 302   // ----- consume second profile of obj1
 303
 304   profile = stub1->next_profile ();
 305
 306   FRANKS_ASSERT (is_endpoint (profile, "192.168.1.3", 1111));
 307
 308   FRANKS_ASSERT (stub1->forward_profiles () == 0);
 309
 310   // ----- reached end, next_profile() must yield NULL
 311
 312   profile = stub1->next_profile ();
 313 }
 314
 315 int
 316 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 317 {
 318   ACE_DEBUG ((LM_DEBUG, "---------------------------------------------\n"));
 319   ACE_DEBUG ((LM_DEBUG, "Running the Stub Tests.\n"));
 320
 321   try
 322     {
 323       // Retrieve the ORB.
 324       CORBA::ORB_var orb = CORBA::ORB_init (argc,
 325                                             argv);
 326
 327       test_forward (orb.in());
 328       test_forward_permanent (orb.in());
 329       test_forward_permanent_mix (orb.in());
 330     }
 331   catch (const CORBA::SystemException& sysex)
 332     {
 333       sysex._tao_print_exception ("Unexpected system Exception!!\n");
 334       return -1;
 335     }
 336   catch (const CORBA::Exception& ex)
 337     {
 338       ex._tao_print_exception ("Unexpected CORBA Exception!\n");
 339       return -1;
 340     }
 341
 342   ACE_DEBUG ((LM_DEBUG, "Stub Tests Successfully Completed!\n"));
 343   ACE_DEBUG ((LM_DEBUG, "---------------------------------------------\n"));
 344
 345   return 0;
 346 }