TAO/orbsvcs/tests/Notify/Basic/Events.h

   1 /* -*- C++ -*- */
   2 //=============================================================================
   3 /**
   4  *  @file   Events.h
   5  *
   6  * Test to check if events are received by all 3 types of consumers.
   7  * This is intended to be a simple test without any filters with default subscription (all events).
   8  *
   9  *  @author Pradeep Gore <pradeep@cs.wustl.edu>
  10  */
  11 //=============================================================================
  12
  13
  14 #ifndef NOTIFY_TESTS_EventS_H
  15 #define NOTIFY_TESTS_EventS_H
  16
  17 #include "Notify_Test_Client.h"
  18 #include "Notify_StructuredPushConsumer.h"
  19 #include "Notify_StructuredPushSupplier.h"
  20
  21 #if defined(_MSC_VER)
  22 #pragma warning(push)
  23 #pragma warning(disable:4250)
  24 #endif /* _MSC_VER */
  25
  26 class Events;
  27
  28 class Event_StructuredPushConsumer : public TAO_Notify_Tests_StructuredPushConsumer
  29 {
  30 public:
  31   /// Constructor.
  32   Event_StructuredPushConsumer (Events *test_client);
  33
  34   // = StructuredPushSupplier methods.
  35   virtual void push_structured_event (
  36       const CosNotification::StructuredEvent & notification);
  37
  38 protected:
  39   Events * test_client_;
  40 };
  41
  42 /***************************************************************************/
  43
  44 class Event_StructuredPushSupplier : public TAO_Notify_Tests_StructuredPushSupplier
  45 {
  46 public:
  47   /// Constructor.
  48   Event_StructuredPushSupplier (Events * test_client);
  49
  50   /// Destructor.
  51   virtual ~Event_StructuredPushSupplier ();
  52
  53 protected:
  54   Events* test_client_;
  55 };
  56
  57 /***************************************************************************/
  58
  59 class Events : public Notify_Test_Client
  60 {
  61 public:
  62   // Initialization and termination code.
  63   Events ();
  64   virtual ~Events ();
  65
  66   int parse_args (int argc,
  67                   ACE_TCHAR *argv[]) ;
  68
  69   /// Initialization.
  70   int init (int argc,
  71             ACE_TCHAR *argv []);
  72
  73   /// Called when an event is received.
  74   void on_event_received ();
  75
  76   /// Run the test.
  77   void run_test ();
  78
  79   /// End the test.
  80   void end_test ();
  81
  82   /// check if we got the expected results.
  83   int check_results ();
  84
  85 protected:
  86   /// Create EC
  87   void create_EC ();
  88
  89   /// Number of events received so far.
  90   ACE_Atomic_Op <TAO_SYNCH_MUTEX, int> result_count_;
  91
  92   /// Use the default admins.
  93   int use_default_admin_;
  94
  95   /// Number of events to send
  96   int event_count_;
  97
  98   /// The one channel that we create using the factory.
  99   CosNotifyChannelAdmin::EventChannel_var ec_;
 100
 101   /// The consumer admin used by consumers.
 102   CosNotifyChannelAdmin::ConsumerAdmin_var consumer_admin_;
 103
 104   /// The supplier admin used by suppliers.
 105   CosNotifyChannelAdmin::SupplierAdmin_var supplier_admin_;
 106
 107   /// Consumer
 108   TAO_Notify_Tests_StructuredPushConsumer* consumer_;
 109
 110   /// Supplier
 111   TAO_Notify_Tests_StructuredPushSupplier* supplier_;
 112
 113 private:
 114   friend class Event_StructuredPushSupplier;
 115   friend class Event_StructuredPushConsumer;
 116 };
 117
 118 /***************************************************************************/
 119
 120 #if defined(_MSC_VER)
 121 #pragma warning(pop)
 122 #endif /* _MSC_VER */
 123
 124 #endif /* NOTIFY_TESTS_EventS_H */