TAO/tests/POA/Default_Servant/Default_Servant.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     Default_Servant.cpp
   5  *
   6  *   This program tests the behavior of POA::id_to_servant() and
   7  *   POA::reference_to_servant() with the use of default servants.
   8  *
   9  *  @author  Irfan Pyarali
  10  */
  11 //=============================================================================
  12
  13
  14 #include "testS.h"
  15 #include "ace/SString.h"
  16 #include "tao/PortableServer/ServantManagerC.h"
  17
  18 class test_i : public POA_test
  19 {
  20 };
  21
  22 void
  23 test_get_servant_manager (PortableServer::POA_ptr poa)
  24 {
  25   bool succeed = false;
  26   try
  27     {
  28       // Getting the servant manager should give a wrong policy exception
  29       // exception
  30       PortableServer::ServantManager_ptr servant_manager =
  31         poa->get_servant_manager ();
  32
  33       ACE_UNUSED_ARG (servant_manager);
  34     }
  35   catch (const PortableServer::POA::WrongPolicy& )
  36     {
  37       succeed = true;
  38     }
  39   catch (const CORBA::Exception&)
  40     {
  41     }
  42
  43   if (!succeed)
  44   {
  45     ACE_ERROR ((LM_ERROR,
  46                 "(%t) ERROR, get servant manager failed, should give an exception\n"));
  47   }
  48 }
  49
  50 void
  51 test_set_servant_manager (PortableServer::POA_ptr poa)
  52 {
  53   bool succeed = false;
  54   try
  55     {
  56       // Setting the servant manager should give a wrong policy exception
  57       // exception
  58       poa->set_servant_manager (PortableServer::ServantManager::_nil());
  59     }
  60   catch (const PortableServer::POA::WrongPolicy& )
  61     {
  62       succeed = true;
  63     }
  64   catch (const CORBA::Exception&)
  65     {
  66     }
  67
  68   if (!succeed)
  69   {
  70     ACE_ERROR ((LM_ERROR,
  71                 "(%t) ERROR, set servant manager failed, should give an exception\n"));
  72   }
  73 }
  74
  75 void
  76 test_get_servant_with_no_set (PortableServer::POA_ptr poa)
  77 {
  78   bool succeed = false;
  79   try
  80     {
  81       // Getting the default servant without one set whould give a NoServant
  82       // exception
  83       PortableServer::Servant servant =
  84         poa->get_servant ();
  85
  86       ACE_UNUSED_ARG (servant);
  87     }
  88   catch (const PortableServer::POA::NoServant& )
  89     {
  90       succeed = true;
  91     }
  92   catch (const CORBA::Exception&)
  93     {
  94     }
  95
  96   if (!succeed)
  97   {
  98     ACE_ERROR ((LM_ERROR,
  99                 "(%t) ERROR, get servant without one set failed\n"));
 100   }
 101 }
 102
 103 void
 104 test_reference_to_servant_active_object(PortableServer::POA_ptr root_poa)
 105 {
 106   test_i test;
 107   CORBA::ULong expected_refcount = 1;
 108
 109   PortableServer::ObjectId_var id =
 110     root_poa->activate_object (&test);
 111   expected_refcount++;
 112
 113   CORBA::Object_var object =
 114     root_poa->id_to_reference (id.in ());
 115
 116   PortableServer::ServantBase_var servant =
 117     root_poa->reference_to_servant (object.in ());
 118   ++expected_refcount;
 119
 120   root_poa->deactivate_object (id.in ());
 121   --expected_refcount;
 122
 123   CORBA::ULong refcount =
 124     test._refcount_value ();
 125
 126   ACE_UNUSED_ARG (refcount);
 127   ACE_UNUSED_ARG (expected_refcount);
 128   ACE_ASSERT (expected_refcount == refcount);
 129 }
 130
 131
 132 int
 133 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 134 {
 135   try
 136     {
 137       // Initialize the ORB.
 138       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 139
 140       // Obtain the RootPOA.
 141       CORBA::Object_var object =
 142         orb->resolve_initial_references ("RootPOA");
 143
 144       // Narrow to POA.
 145       PortableServer::POA_var root_poa =
 146         PortableServer::POA::_narrow (object.in ());
 147
 148       // Get the POAManager of the RootPOA.
 149       PortableServer::POAManager_var poa_manager =
 150         root_poa->the_POAManager ();
 151
 152       // Policies for the new POA.
 153       CORBA::PolicyList policies (3);
 154       policies.length (3);
 155
 156       // Request Processing Policy.
 157       policies[0] =
 158         root_poa->create_request_processing_policy (PortableServer::USE_DEFAULT_SERVANT);
 159
 160       // Id Uniqueness Policy.
 161       policies[1] =
 162         root_poa->create_id_uniqueness_policy (PortableServer::MULTIPLE_ID);
 163
 164       // Servant Retention Policy.
 165       policies[2] =
 166         root_poa->create_servant_retention_policy (PortableServer::NON_RETAIN);
 167
 168       // Create POA to host default servant.
 169       ACE_CString name = "Default Servant";
 170       PortableServer::POA_var default_servant_poa =
 171         root_poa->create_POA (name.c_str (),
 172                               poa_manager.in (),
 173                               policies);
 174
 175       // Destroy policies.
 176       for (CORBA::ULong i = 0;
 177            i < policies.length ();
 178            ++i)
 179         {
 180           CORBA::Policy_ptr policy = policies[i];
 181           policy->destroy ();
 182         }
 183
 184       // Activate POA manager.
 185       poa_manager->activate ();
 186
 187       test_reference_to_servant_active_object(root_poa.in ());
 188
 189       // Test servant.
 190       test_i test;
 191       CORBA::ULong expected_refcount = 1;
 192
 193       (void) test_get_servant_with_no_set (default_servant_poa.in());
 194
 195       (void) test_get_servant_manager (default_servant_poa.in());
 196
 197       (void) test_set_servant_manager (default_servant_poa.in());
 198
 199       // Register default servant.
 200       default_servant_poa->set_servant (&test);
 201       expected_refcount++;
 202
 203       // Create dummy id.
 204       PortableServer::ObjectId_var id =
 205         PortableServer::string_to_ObjectId ("id");
 206
 207       // Create dummy object.
 208       object =
 209         default_servant_poa->create_reference ("IDL:test:1.0");
 210
 211       // Invoke id_to_servant(). Should retrieve default servant.
 212       PortableServer::ServantBase_var servant =
 213         default_servant_poa->id_to_servant (id.in ());
 214       expected_refcount++;
 215
 216       // Assert correctness.
 217       ACE_ASSERT (&test == servant.in());
 218
 219       // Invoke reference_to_servant(). Should retrieve default servant.
 220       servant =
 221         default_servant_poa->reference_to_servant (object.in ());
 222       expected_refcount++;
 223
 224       // Assert correctness.
 225       ACE_ASSERT (&test == servant.in());
 226
 227       // Report success.
 228       ACE_DEBUG ((LM_DEBUG,
 229                   "Default_Servant test successful\n"));
 230
 231       CORBA::ULong refcount =
 232         test._refcount_value ();
 233
 234       ACE_UNUSED_ARG (refcount);
 235       ACE_UNUSED_ARG (expected_refcount);
 236       ACE_ASSERT (expected_refcount == refcount);
 237
 238       // Destroy the ORB.
 239       orb->destroy ();
 240     }
 241   catch (const CORBA::Exception& ex)
 242     {
 243       ex._tao_print_exception ("Exception caught");
 244       return -1;
 245     }
 246
 247   return 0;
 248 }