libstdc++-v3/testsuite/20_util/shared_ptr/cons/alloc.cc

   1 // { dg-options "-std=gnu++0x" }
   2
   3 // Copyright (C) 2007 Free Software Foundation
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 2, or (at your option)
   9 // any later version.
  10
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License along
  17 // with this library; see the file COPYING.  If not, write to the Free
  18 // Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  19 // USA.
  20
  21 // 20.6.6.2 Template class shared_ptr [util.smartptr.shared]
  22
  23 #include <memory>
  24 #include <testsuite_hooks.h>
  25 #include <testsuite_allocator.h>
  26
  27 using __gnu_test::tracker_allocator_counter;
  28 using __gnu_test::tracker_allocator;
  29
  30 struct A { };
  31 void deletefunc(A* p) { delete p; }
  32 struct D
  33 {
  34   void operator()(A* p) { delete p; ++delete_count; }
  35   static long delete_count;
  36 };
  37 long D::delete_count = 0;
  38
  39 // 20.6.6.2.1 shared_ptr constructors [util.smartptr.shared.const]
  40
  41 // Construction with allocator
  42 int
  43 test01()
  44 {
  45   bool test __attribute__((unused)) = true;
  46   tracker_allocator_counter::reset();
  47
  48   std::shared_ptr<A> p1(new A, deletefunc, tracker_allocator<A>());
  49   std::size_t const sz = tracker_allocator_counter::get_allocation_count();
  50   VERIFY( sz > 0 );
  51   {
  52     std::shared_ptr<A> p2(p1);
  53     VERIFY( p2.use_count() == 2 );
  54     VERIFY( tracker_allocator_counter::get_allocation_count() == sz );
  55     VERIFY( tracker_allocator_counter::get_deallocation_count() == 0 );
  56   }
  57   VERIFY( p1.use_count() == 1 );
  58   VERIFY( tracker_allocator_counter::get_allocation_count() == sz );
  59   VERIFY( tracker_allocator_counter::get_deallocation_count() == 0 );
  60   p1.reset();
  61   VERIFY( p1.use_count() == 0 );
  62   VERIFY( tracker_allocator_counter::get_allocation_count() == sz );
  63   VERIFY( tracker_allocator_counter::get_deallocation_count() == sz );
  64
  65   return 0;
  66 }
  67
  68 // Construction with allocator
  69 int
  70 test02()
  71 {
  72   bool test __attribute__((unused)) = true;
  73   tracker_allocator_counter::reset();
  74
  75   std::shared_ptr<A> p1(new A, deletefunc, tracker_allocator<A>());
  76   std::size_t const sz1 = tracker_allocator_counter::get_allocation_count();
  77   VERIFY( sz1 > 0 );
  78   std::shared_ptr<A> p2(new A, D(), tracker_allocator<A>());
  79   std::size_t const sz2 = tracker_allocator_counter::get_allocation_count();
  80   VERIFY( sz2 > sz1 );
  81   VERIFY( tracker_allocator_counter::get_deallocation_count() == 0 );
  82   p1 = p2;
  83   VERIFY( p2.use_count() == 2 );
  84   VERIFY( tracker_allocator_counter::get_allocation_count() == sz2 );
  85   VERIFY( tracker_allocator_counter::get_deallocation_count() == sz1 );
  86   p1.reset();
  87   VERIFY( p2.use_count() == 1 );
  88   VERIFY( tracker_allocator_counter::get_allocation_count() == sz2 );
  89   VERIFY( tracker_allocator_counter::get_deallocation_count() == sz1 );
  90   p2.reset();
  91   VERIFY( tracker_allocator_counter::get_allocation_count() == sz2 );
  92   VERIFY( tracker_allocator_counter::get_deallocation_count() == sz2 );
  93   VERIFY( D::delete_count == 1 );
  94
  95   return 0;
  96 }
  97
  98 int
  99 main()
 100 {
 101   test01();
 102   test02();
 103   return 0;
 104 }