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

   1 /* -*- C++ -*- */
   2 //=============================================================================
   3 /**
   4  *  @file   Simple.h
   5  *
   6  * Simple test any supplier to any consumer.
   7  *
   8  *  @author Pradeep Gore <pradeep@cs.wustl.edu>
   9  */
  10 //=============================================================================
  11
  12
  13 #ifndef NOTIFY_TESTS_SIMPLE_H
  14 #define NOTIFY_TESTS_SIMPLE_H
  15
  16 #include "Notify_Test_Client.h"
  17 #include "Notify_PushConsumer.h"
  18 #include "Notify_PushSupplier.h"
  19
  20 #if defined(_MSC_VER)
  21 #pragma warning(push)
  22 #pragma warning(disable:4250)
  23 #endif /* _MSC_VER */
  24
  25 class Simple_Test;
  26
  27 /***************************************************************************/
  28
  29 class Event_AnyPushConsumer : public TAO_Notify_Tests_PushConsumer
  30 {
  31 public:
  32   /// Constructor.
  33   Event_AnyPushConsumer (Simple_Test *test_client);
  34
  35   // = PushSupplier methods
  36   virtual void push (const CORBA::Any & data);
  37
  38 protected:
  39   Simple_Test * test_client_;
  40 };
  41
  42 /***************************************************************************/
  43
  44 class Event_AnyPushSupplier : public TAO_Notify_Tests_PushSupplier
  45 {
  46 public:
  47   /// Constructor.
  48   Event_AnyPushSupplier (Simple_Test * test_client);
  49
  50   /// Destructor.
  51   virtual ~Event_AnyPushSupplier ();
  52
  53 protected:
  54   Simple_Test* test_client_;
  55 };
  56
  57 /***************************************************************************/
  58
  59 class Simple_Test : public Notify_Test_Client
  60 {
  61 public:
  62   // Initialization and termination code
  63   Simple_Test ();
  64   virtual ~Simple_Test ();
  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   /// Number of events to send
  93   int event_count_;
  94
  95   /// The one channel that we create using the factory.
  96   CosNotifyChannelAdmin::EventChannel_var ec_;
  97
  98   /// The consumer admin used by consumers.
  99   CosNotifyChannelAdmin::ConsumerAdmin_var consumer_admin_;
 100
 101   /// The supplier admin used by suppliers.
 102   CosNotifyChannelAdmin::SupplierAdmin_var supplier_admin_;
 103
 104   /// Consumer.
 105   TAO_Notify_Tests_PushConsumer* consumer_;
 106
 107   /// Supplier.
 108   TAO_Notify_Tests_PushSupplier* supplier_;
 109
 110 private:
 111   friend class Event_AnyPushSupplier;
 112   friend class Event_AnyPushConsumer;
 113 };
 114
 115 /***************************************************************************/
 116
 117 #if defined(_MSC_VER)
 118 #pragma warning(pop)
 119 #endif /* _MSC_VER */
 120
 121 #endif /* NOTIFY_TESTS_SIMPLE_H */