TAO/orbsvcs/tests/unit/Notify/MC/Control/Control.cpp

   1 #include "ace/Log_Msg.h"
   2
   3 #include "orbsvcs/Notify/MonitorControl/Control.h"
   4
   5 #if defined (TAO_HAS_MONITOR_FRAMEWORK) && (TAO_HAS_MONITOR_FRAMEWORK == 1)
   6
   7 static ACE_CString command;
   8
   9 class ControlTest : public TAO_NS_Control
  10 {
  11 public:
  12   ControlTest (const ACE_CString& name)
  13     : TAO_NS_Control (name.c_str ())
  14   {
  15   }
  16
  17   virtual bool execute (const char* cmd)
  18   {
  19     command = cmd;
  20     return (ACE_OS::strcmp (cmd, TAO_NS_CONTROL_SHUTDOWN) == 0);
  21   }
  22 };
  23
  24 void
  25 error (const char* msg)
  26 {
  27   ACE_ERROR ((LM_ERROR, "%s\n", msg));
  28   ACE_OS::exit (1);
  29 }
  30
  31 #endif /* TAO_HAS_MONITOR_FRAMEWORK==1 */
  32
  33 int
  34 ACE_TMAIN (int, ACE_TCHAR*[])
  35 {
  36 #if defined (TAO_HAS_MONITOR_FRAMEWORK) && (TAO_HAS_MONITOR_FRAMEWORK == 1)
  37
  38   try
  39     {
  40       ACE_CString name ("Control Test");
  41       ControlTest ct (name);
  42
  43       if (name != ct.name ())
  44         {
  45           error ("The TAO_NS_Control name does not work");
  46         }
  47
  48       if (ct.execute (TAO_NS_CONTROL_REMOVE_CONSUMER))
  49         {
  50           error ("The control object should have returned false");
  51         }
  52
  53       if (!ct.execute (TAO_NS_CONTROL_SHUTDOWN))
  54         {
  55           error ("The control object shouldn't have returned false");
  56         }
  57
  58       if (command != TAO_NS_CONTROL_SHUTDOWN)
  59         {
  60           error ("The TAO_NS_Control callback does not work");
  61         }
  62     }
  63   catch(...)
  64     {
  65       error ("Caught an unexpected exception type");
  66     }
  67
  68 #endif /* TAO_HAS_MONITOR_FRAMEWORK==1 */
  69
  70   return 0;
  71 }
  72