ACE/tests/Intrusive_Auto_Ptr_Test.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    Intrusive_Auto_Ptr_Test.cpp
   5  *
   6  *  This test verifies the functionality of the <ACE_Intrusive_Auto_Ptr>
   7  *  implementation.
   8  *
   9  *  @author Iliyan Jeliazkov <iliyan@ociweb.com>
  10  */
  11 //=============================================================================
  12
  13
  14 #include "test_config.h"
  15 #include "ace/Intrusive_Auto_Ptr.h"
  16 #include "ace/Thread_Manager.h"
  17
  18 class One {
  19   static bool released;
  20
  21   int ref;
  22
  23 public:
  24   explicit One (int refcount) : ref(refcount)
  25   {
  26     released = false;
  27   }
  28
  29   ~One ()
  30   {
  31     released = true;
  32   }
  33
  34   bool has_refs (int howmany)
  35   {
  36     return this->ref == howmany;
  37   }
  38
  39   static bool was_released ()
  40   {
  41     return released;
  42   }
  43
  44   static void intrusive_add_ref (One *);
  45   static void intrusive_remove_ref (One *);
  46 };
  47
  48 bool One::released = true;
  49
  50 void
  51 One::intrusive_add_ref (One *one) {
  52   one->ref++;
  53 }
  54
  55 void
  56 One::intrusive_remove_ref (One *one) {
  57   one->ref--;
  58   if (one->ref == 0)
  59     delete one;
  60 }
  61
  62 int run_main (int, ACE_TCHAR *[])
  63 {
  64   ACE_START_TEST (ACE_TEXT ("Intrusive_Auto_Ptr_Test"));
  65
  66
  67   One *theone (new One(0));
  68
  69   {
  70     ACE_TEST_ASSERT (theone->has_refs (0));
  71     ACE_TEST_ASSERT (!One::was_released ());
  72
  73     ACE_Intrusive_Auto_Ptr<One> ip2(theone);
  74
  75     {
  76       ACE_TEST_ASSERT (theone->has_refs (1));
  77       ACE_TEST_ASSERT (!One::was_released ());
  78
  79       ACE_Intrusive_Auto_Ptr<One> ip2(theone);
  80       ACE_TEST_ASSERT (theone->has_refs (2));
  81       ACE_TEST_ASSERT (!One::was_released ());
  82     }
  83
  84     ACE_TEST_ASSERT (theone->has_refs (1));
  85     ACE_TEST_ASSERT (!One::was_released ());
  86   }
  87
  88   ACE_TEST_ASSERT (One::was_released());
  89
  90   // Test assigning zero.
  91   theone = new One(0);
  92   {
  93     ACE_TEST_ASSERT (theone->has_refs (0));
  94     ACE_TEST_ASSERT (!One::was_released ());
  95
  96     // Transfer object to instrusive auto ptr.
  97     // The reference is expected to increment to 1.
  98     ACE_Intrusive_Auto_Ptr<One> ip2(theone);
  99
 100     ACE_TEST_ASSERT (theone->has_refs (1));
 101     ACE_TEST_ASSERT (!One::was_released ());
 102
 103     // Assign a zero instrusive auto ptr.
 104     // The reference is expected to decrement to 0, so that the object is
 105     // expected to be released.
 106     ACE_Intrusive_Auto_Ptr<One> ip3;
 107     ip2 = ip3;
 108     ACE_TEST_ASSERT (One::was_released ());
 109   }
 110
 111   ACE_END_TEST;
 112   return 0;
 113 }