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

   1 // { dg-options "-std=gnu++0x" }
   2
   3 // Copyright (C) 2005-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
  25 struct A { };
  26 struct B : A { };
  27 struct D
  28 {
  29   void operator()(B* p) { delete p; ++delete_count; }
  30   static long delete_count;
  31 };
  32 long D::delete_count = 0;
  33
  34 // 20.6.6.2.4 shared_ptr modifiers [util.smartptr.shared.mod]
  35
  36 // reset
  37 int
  38 test01()
  39 {
  40   bool test __attribute__((unused)) = true;
  41
  42   A * const a = new A;
  43   std::shared_ptr<A> p1(a);
  44   std::shared_ptr<A> p2(p1);
  45   p1.reset();
  46   VERIFY( p1.get() == 0 );
  47   VERIFY( p2.get() == a );
  48
  49   return 0;
  50 }
  51
  52 int
  53 test02()
  54 {
  55   bool test __attribute__((unused)) = true;
  56
  57   A * const a = new A;
  58   B * const b = new B;
  59   std::shared_ptr<A> p1(a);
  60   std::shared_ptr<A> p2(p1);
  61   p1.reset(b);
  62   VERIFY( p1.get() == b );
  63   VERIFY( p2.get() == a );
  64
  65   return 0;
  66 }
  67
  68 int
  69 test03()
  70 {
  71   bool test __attribute__((unused)) = true;
  72
  73   {
  74     std::shared_ptr<A> p1;
  75     p1.reset(new B, D());
  76   }
  77   VERIFY( D::delete_count == 1 );
  78
  79   return 0;
  80 }
  81
  82 int
  83 main()
  84 {
  85   test01();
  86   test02();
  87   test03();
  88   return 0;
  89 }