libstdc++-v3/testsuite/tr1/2_general_utilities/shared_ptr/modifiers/reset.cc

   1 // Copyright (C) 2005-2019 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17
  18 // TR1 2.2.2 Template class shared_ptr [tr.util.smartptr.shared]
  19
  20 #include <tr1/memory>
  21 #include <testsuite_hooks.h>
  22
  23 struct A { };
  24 struct B : A { };
  25 struct D
  26 {
  27   void operator()(B* p) { delete p; ++delete_count; }
  28   static long delete_count;
  29 };
  30 long D::delete_count = 0;
  31
  32 // 2.2.3.4 shared_ptr modifiers [tr.util.smartptr.shared.mod]
  33
  34 // reset
  35 int
  36 test01()
  37 {
  38   A * const a = new A;
  39   std::tr1::shared_ptr<A> p1(a);
  40   std::tr1::shared_ptr<A> p2(p1);
  41   p1.reset();
  42   VERIFY( p1.get() == 0 );
  43   VERIFY( p2.get() == a );
  44
  45   return 0;
  46 }
  47
  48 int
  49 test02()
  50 {
  51   A * const a = new A;
  52   B * const b = new B;
  53   std::tr1::shared_ptr<A> p1(a);
  54   std::tr1::shared_ptr<A> p2(p1);
  55   p1.reset(b);
  56   VERIFY( p1.get() == b );
  57   VERIFY( p2.get() == a );
  58
  59   return 0;
  60 }
  61
  62 int
  63 test03()
  64 {
  65   {
  66     std::tr1::shared_ptr<A> p1;
  67     p1.reset(new B, D());
  68   }
  69   VERIFY( D::delete_count == 1 );
  70
  71   return 0;
  72 }
  73
  74 int
  75 main()
  76 {
  77   test01();
  78   test02();
  79   test03();
  80   return 0;
  81 }