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 protected:
  59   Updates* test_client_;
  60 };
  61
  62 /***************************************************************************/
  63
  64 class Updates : public Notify_Test_Client
  65 {
  66 public:
  67   // Initialization and termination code
  68   Updates ();
  69   virtual ~Updates ();
  70
  71   /// initialization.
  72   int init (int argc, ACE_TCHAR *argv []);
  73
  74   /// Run the test.
  75   void run_test ();
  76
  77   /// print_event_types
  78   void print_event_types (const CosNotification::EventTypeSeq &types);
  79
  80   /// Invoked by Supplier and Consumer when they receive updates
  81   void types_changed (const CosNotification::EventTypeSeq & added, const CosNotification::EventTypeSeq & removed);
  82
  83   /// Wait for expected count of updates.
  84   void wait_for_updates (int expected_added, int expected_removed);
  85
  86 protected:
  87   /// Create EC
  88   void create_EC ();
  89
  90   /// Helper to add types.
  91   void add_type (CosNotification::EventTypeSeq& type_seq, const char* type);
  92
  93   // test subscription_change
  94   void test_subscription_change ();
  95
  96   // test offer_change
  97   void test_offer_change ();
  98
  99   /// Reset the counts.
 100   void reset_counts ();
 101
 102   /// Lock to serialize internal state.
 103   TAO_SYNCH_MUTEX lock_;
 104
 105   /// Count of added updates received
 106   int added_count_;
 107
 108   /// Count of removed updates received
 109   /// Check if these are equal to <update_count_> at the end of the test.
 110   int removed_count_;
 111
 112   /// The one channel that we create using the factory.
 113   CosNotifyChannelAdmin::EventChannel_var ec_;
 114
 115   /// The consumer admin used by consumers.
 116   CosNotifyChannelAdmin::ConsumerAdmin_var consumer_admin_;
 117
 118   /// The supplier admin used by suppliers.
 119   CosNotifyChannelAdmin::SupplierAdmin_var supplier_admin_;
 120
 121   /// Consumer
 122   TAO_Notify_Tests_StructuredPushConsumer* consumer_;
 123
 124   /// Supplier
 125   TAO_Notify_Tests_StructuredPushSupplier* supplier_;
 126
 127 private:
 128   friend class Updates_StructuredPushSupplier;
 129   friend class Updates_StructuredPushConsumer;
 130 };
 131
 132 /***************************************************************************/
 133
 134 #if defined(_MSC_VER)
 135 #pragma warning(pop)
 136 #endif /* _MSC_VER */
 137
 138 #endif /* NOTIFY_TESTS_UPDATES_H */