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

   1 /* -*- C++ -*- */
   2 //=============================================================================
   3 /**
   4  *  @file   Updates.h
   5  *
   6  * Test to check if <offer_change> and <subscription_change> messages are
   7  * received correctly.
   8  *
   9  *  @author Pradeep Gore <pradeep@cs.wustl.edu>
  10  */
  11 //=============================================================================
  12
  13
  14 #ifndef NOTIFY_TESTS_UPDATES_H
  15 #define NOTIFY_TESTS_UPDATES_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 Updates;
  27
  28 class Updates_StructuredPushConsumer : public TAO_Notify_Tests_StructuredPushConsumer
  29 {
  30 public:
  31   /// Constructor.
  32   Updates_StructuredPushConsumer (Updates *test_client);
  33
  34   /// Offer change is conveyed here.
  35   virtual void offer_change (const CosNotification::EventTypeSeq & added,
  36                              const CosNotification::EventTypeSeq & removed);
  37
  38 protected:
  39   Updates * test_client_;
  40 };
  41
  42 /***************************************************************************/
  43
  44 class Updates_StructuredPushSupplier : public TAO_Notify_Tests_StructuredPushSupplier
  45 {
  46 public:
  47   /// Constructor.
  48   Updates_StructuredPushSupplier (Updates * test_client);
  49
  50   /// Destructor.
  51   virtual ~Updates_StructuredPushSupplier ();
  52
  53   /// Subscription change is conveyed here.
  54   virtual void subscription_change (
  55       const CosNotification::EventTypeSeq & added,
  56       const CosNotification::EventTypeSeq & removed
  57     );
  58
  59 protected:
  60   Updates* test_client_;
  61 };
  62
  63 /***************************************************************************/
  64
  65 class Updates : public Notify_Test_Client
  66 {
  67 public:
  68   // Initialization and termination code
  69   Updates (void);
  70   virtual ~Updates ();
  71
  72   /// initialization.
  73   int init (int argc, ACE_TCHAR *argv []);
  74
  75   /// Run the test.
  76   void run_test (void);
  77
  78   /// print_event_types
  79   void print_event_types (const CosNotification::EventTypeSeq &types);
  80
  81   /// Invoked by Supplier and Consumer when they receive updates
  82   void types_changed (const CosNotification::EventTypeSeq & added, const CosNotification::EventTypeSeq & removed);
  83
  84   /// Wait for expected count of updates.
  85   void wait_for_updates (int expected_added, int expected_removed);
  86
  87 protected:
  88   /// Create EC
  89   void create_EC (void);
  90
  91   /// Helper to add types.
  92   void add_type (CosNotification::EventTypeSeq& type_seq, const char* type);
  93
  94   // test subscription_change
  95   void test_subscription_change (void);
  96
  97   // test offer_change
  98   void test_offer_change (void);
  99
 100   /// Reset the counts.
 101   void reset_counts (void);
 102
 103   /// Lock to serialize internal state.
 104   TAO_SYNCH_MUTEX lock_;
 105
 106   /// Count of added updates received
 107   int added_count_;
 108
 109   /// Count of removed updates received
 110   /// Check if these are equal to <update_count_> at the end of the test.
 111   int removed_count_;
 112
 113   /// The one channel that we create using the factory.
 114   CosNotifyChannelAdmin::EventChannel_var ec_;
 115
 116   /// The consumer admin used by consumers.
 117   CosNotifyChannelAdmin::ConsumerAdmin_var consumer_admin_;
 118
 119   /// The supplier admin used by suppliers.
 120   CosNotifyChannelAdmin::SupplierAdmin_var supplier_admin_;
 121
 122   /// Consumer
 123   TAO_Notify_Tests_StructuredPushConsumer* consumer_;
 124
 125   /// Supplier
 126   TAO_Notify_Tests_StructuredPushSupplier* supplier_;
 127
 128 private:
 129   friend class Updates_StructuredPushSupplier;
 130   friend class Updates_StructuredPushConsumer;
 131 };
 132
 133 /***************************************************************************/
 134
 135 #if defined(_MSC_VER)
 136 #pragma warning(pop)
 137 #endif /* _MSC_VER */
 138
 139 #endif /* NOTIFY_TESTS_UPDATES_H */