TAO/orbsvcs/tests/Notify/Bug_2415_Regression/Consumer.cpp

   1 // ******************************************************************
   2 // Include Section
   3 // ******************************************************************
   4
   5 #include "ace/Get_Opt.h"
   6 #include "ace/OS_NS_unistd.h"
   7
   8 #include "orbsvcs/CosNotifyCommC.h"
   9 #include "orbsvcs/CosNamingC.h"
  10 #include "Notify_Push_Consumer.h"
  11 #include "goC.h"
  12
  13 #include "Notify_Test_Client.h"
  14
  15 #define TEST_OP 1
  16
  17 // ******************************************************************
  18 // Data Section
  19 // ******************************************************************
  20
  21 static const ACE_TCHAR *ior = ACE_TEXT ("file://supplier.ior");
  22 static unsigned int expect_fail = 0;
  23 static Notify_Push_Consumer* consumer_1 = 0;
  24
  25 // ******************************************************************
  26 // Subroutine Section
  27 // ******************************************************************
  28
  29 class Consumer_Client : public Notify_Test_Client
  30 {
  31 public:
  32   virtual int parse_args (int argc, ACE_TCHAR *argv[]);
  33 };
  34
  35
  36 int
  37 Consumer_Client::parse_args (int argc, ACE_TCHAR *argv[])
  38 {
  39   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:l:h:d:f:"));
  40   int c;
  41
  42   while ((c = get_opts ()) != -1)
  43     switch (c)
  44       {
  45       case 'k':
  46         ior = get_opts.optarg;
  47         break;
  48
  49       case 'f':
  50         expect_fail = ACE_OS::atoi (get_opts.optarg);
  51         break;
  52
  53       default:
  54         ACE_ERROR_RETURN ((LM_ERROR,
  55                            "usage:  %s "
  56                            "-k <ior> "
  57                            "-f <expect failure> "
  58                            "\n",
  59                            argv [0]),
  60                           -1);
  61       }
  62   // Indicates successful parsing of the command line
  63   return 0;
  64 }
  65
  66 static CosNotifyFilter::Filter_ptr
  67 create_proxyFilter (CosNotifyChannelAdmin::EventChannel_ptr ec)
  68 {
  69   CosNotifyFilter::FilterFactory_var ffact =
  70     ec->default_filter_factory ();
  71
  72   CosNotifyFilter::Filter_var filter =
  73     ffact->create_filter ("EXTENDED_TCL");
  74
  75   if(!CORBA::is_nil(filter.in()))
  76     {
  77     CosNotifyFilter::ConstraintExpSeq constraint_list (1);
  78     constraint_list.length(1);
  79
  80     constraint_list[0].event_types.length (0);
  81
  82     constraint_list[0].constraint_expr =
  83       CORBA::string_dup ("$.domain_name == 'domain1'");
  84
  85     CosNotifyFilter::ConstraintInfoSeq_var cons_info = filter->add_constraints (constraint_list);
  86     }
  87   return filter._retn();
  88 }
  89
  90 static void
  91 create_consumers (CosNotifyChannelAdmin::ConsumerAdmin_ptr admin,
  92                   Notify_Test_Client* client,
  93                   CosNotifyFilter::Filter_ptr filter)
  94 {
  95   // startup the consumer
  96   ACE_NEW_THROW_EX (consumer_1,
  97                     Notify_Push_Consumer (*client),
  98                     CORBA::NO_MEMORY ());
  99
 100   consumer_1->init (client->root_poa ());
 101
 102   consumer_1->_connect (admin, filter);
 103 }
 104
 105 // ******************************************************************
 106 // Main Section
 107 // ******************************************************************
 108
 109 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 110 {
 111   int status = 0;
 112   try
 113     {
 114       Consumer_Client client;
 115
 116       status = client.init (argc, argv);
 117
 118       if (status == 0)
 119         {
 120           CosNotifyChannelAdmin::EventChannel_var ec =
 121             client.create_event_channel ("MyEventChannel", 1);
 122
 123           CORBA::ORB_ptr orb = client.orb ();
 124           CORBA::Object_var object =
 125                               orb->string_to_object (ior);
 126
 127           sig_var sig = sig::_narrow (object.in ());
 128
 129           if (CORBA::is_nil (sig.in ()))
 130             {
 131               ACE_ERROR_RETURN ((LM_ERROR,
 132                                  "Object reference <%s> is nil\n",
 133                                  ior),
 134                                 1);
 135             }
 136
 137           CosNotifyChannelAdmin::ConsumerAdmin_var admin =
 138             ec->default_consumer_admin();
 139
 140           CosNotifyFilter::Filter_var filter =
 141             create_proxyFilter (ec.in());
 142
 143           if (!CORBA::is_nil (admin.in ()))
 144             {
 145               create_consumers (admin.in (), &client, filter.in());
 146
 147               // Tell the supplier to go
 148               sig->go ();
 149
 150               ACE_Time_Value tv(10, 0);
 151               client.ORB_run(tv);
 152               ACE_DEBUG((LM_DEBUG, "Consumer done.\n"));
 153
 154               sig->done ();
 155
 156               ACE_DEBUG((LM_DEBUG, "Expected %d message(s) and received %d message(s).\n",
 157                 expect_fail == 1 ? 0 : 1, consumer_1->messages_received()));
 158
 159               if (1 == expect_fail)
 160                 status = consumer_1->messages_received() == 0 ? 0 : 1;
 161               else
 162                 status = consumer_1->messages_received() == 1 ? 0 : 1;
 163
 164               ACE_DEBUG((LM_DEBUG, "Test status is %s.\n", status == 1 ? "fail" : "pass"));
 165             }
 166         }
 167     }
 168   catch (const CORBA::Exception& e)
 169     {
 170       e._tao_print_exception ("Error: Consumer exception: ");
 171       status = 1;
 172     }
 173
 174   return status;
 175 }