libstdc++-v3/testsuite/20_util/shared_ptr/modifiers/reset_alloc.cc

   1 // { dg-options "-std=gnu++0x" }
   2
   3 // Copyright (C) 2007-2014 Free Software Foundation, Inc.
   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 3, 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 COPYING3.  If not see
  18 // <http://www.gnu.org/licenses/>.
  19
  20 // 20.6.6.2 Template class shared_ptr [util.smartptr.shared]
  21
  22 #include <memory>
  23 #include <testsuite_hooks.h>
  24 #include <testsuite_allocator.h>
  25
  26 using __gnu_test::tracker_allocator_counter;
  27 using __gnu_test::tracker_allocator;
  28
  29 struct A { };
  30 struct B : A { };
  31 struct D
  32 {
  33   void operator()(B* p) { delete p; ++delete_count; }
  34   static long delete_count;
  35 };
  36 long D::delete_count = 0;
  37
  38 // 20.6.6.2.4 shared_ptr modifiers [util.smartptr.shared.mod]
  39
  40 // Reset with allocator
  41 int
  42 test01()
  43 {
  44   bool test __attribute__((unused)) = true;
  45   tracker_allocator_counter::reset();
  46
  47   {
  48     std::shared_ptr<A> p1;
  49     p1.reset(new B, D(), tracker_allocator<B>());
  50     VERIFY( tracker_allocator_counter::get_allocation_count() > 0 );
  51   }
  52   VERIFY( D::delete_count == 1 );
  53   VERIFY( tracker_allocator_counter::get_allocation_count() == tracker_allocator_counter::get_deallocation_count() );
  54
  55   return 0;
  56 }
  57
  58 int
  59 main()
  60 {
  61   test01();
  62   return 0;
  63 }