TAO/tests/Mixed_Sync_ASync_Events/test_i.h

   1
   2 //=============================================================================
   3 /**
   4  *  @file   test_i.h
   5  */
   6 //=============================================================================
   7
   8 #ifndef TAO_MSAE_TEST_I_H
   9 #define TAO_MSAE_TEST_I_H
  10
  11 #include "testS.h"
  12
  13 /**
  14  * @class Test_i
  15  *
  16  * @brief Test implementation
  17  *
  18  * Implements the Test interface in test.idl
  19  */
  20 class Test_i : public POA_A::Test
  21 {
  22 public:
  23   /// ctor
  24   Test_i (CORBA::ORB_ptr orb,
  25           A::AMI_TestHandler_ptr rh,
  26           CORBA::ULong max_count = 10,
  27           A::RunMode mode = A::RM_SLAVE);
  28
  29   void set_opponent (A::Test_ptr opp);
  30
  31   virtual void request (
  32     ::A::RequestMode mode,
  33     ::CORBA::ULong & counter,
  34     ::A::FollowUp_out follow_up);
  35
  36   virtual void report (
  37     const char * msg);
  38
  39   void shutdown (void);
  40
  41 private:
  42   CORBA::ORB_var orb_;
  43
  44   A::AMI_TestHandler_var rh_;
  45
  46   A::Test_var opponent_;
  47
  48   CORBA::ULong max_count_;
  49
  50   A::RunMode mode_;
  51
  52   unsigned int seed_;
  53 };
  54
  55 class TestHandler
  56   : public ACE_Event_Handler
  57 {
  58 public:
  59   TestHandler (CORBA::ORB_ptr orb,
  60                CORBA::ULong max_count,
  61                A::RunMode mode);
  62
  63   void set_counter (CORBA::ULong counter);
  64
  65   void set_opponent (A::Test_ptr opp);
  66
  67   void set_reply_handler (A::AMI_TestHandler_ptr rh);
  68
  69   virtual int handle_timeout (const ACE_Time_Value &tv,
  70                               const void *arg);
  71
  72   virtual int handle_exception (ACE_HANDLE fc = ACE_INVALID_HANDLE);
  73
  74   void handle_followup (A::FollowUp fup, CORBA::ULong counter);
  75
  76   void start ();
  77
  78   CORBA::ORB_ptr orb () { return this->orb_.in (); }
  79   A::Test_ptr opponent () { return this->opponent_.in (); }
  80   CORBA::ULong max_count () { return this->max_count_; }
  81   A::RunMode mode () { return this->mode_; }
  82
  83 private:
  84   CORBA::ORB_var orb_;
  85   A::Test_var opponent_;
  86   A::AMI_TestHandler_var rh_;
  87   CORBA::ULong max_count_;
  88   A::RunMode mode_;
  89   CORBA::ULong counter_;
  90   unsigned int seed_;
  91 };
  92
  93 class Test_Reply_i : public POA_A::AMI_TestHandler
  94 {
  95 public:
  96   Test_Reply_i (CORBA::ORB_ptr orb,
  97       CORBA::ULong max_count,
  98       A::RunMode mode);
  99
 100   virtual void request (
 101     ::CORBA::ULong counter,
 102     ::A::FollowUp follow_up);
 103
 104   virtual void request_excep (
 105     ::Messaging::ExceptionHolder * excep_holder);
 106
 107   virtual void report (
 108     void);
 109
 110   virtual void report_excep (
 111     ::Messaging::ExceptionHolder * excep_holder);
 112
 113   TestHandler& test_handler () { return this->evh_; }
 114
 115 private:
 116   TestHandler evh_;
 117 };
 118
 119
 120
 121 #endif /* TAO_MSAE_TEST_I_H */
 122