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

   1 // Copyright (C) 2005, 2009 Free Software Foundation
   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   bool test __attribute__((unused)) = true;
  39
  40   A * const a = new A;
  41   std::tr1::shared_ptr<A> p1(a);
  42   std::tr1::shared_ptr<A> p2(p1);
  43   p1.reset();
  44   VERIFY( p1.get() == 0 );
  45   VERIFY( p2.get() == a );
  46
  47   return 0;
  48 }
  49
  50 int
  51 test02()
  52 {
  53   bool test __attribute__((unused)) = true;
  54
  55   A * const a = new A;
  56   B * const b = new B;
  57   std::tr1::shared_ptr<A> p1(a);
  58   std::tr1::shared_ptr<A> p2(p1);
  59   p1.reset(b);
  60   VERIFY( p1.get() == b );
  61   VERIFY( p2.get() == a );
  62
  63   return 0;
  64 }
  65
  66 int
  67 test03()
  68 {
  69   bool test __attribute__((unused)) = true;
  70
  71   {
  72     std::tr1::shared_ptr<A> p1;
  73     p1.reset(new B, D());
  74   }
  75   VERIFY( D::delete_count == 1 );
  76
  77   return 0;
  78 }
  79
  80 int
  81 main()
  82 {
  83   test01();
  84   test02();
  85   test03();
  86   return 0;
  87 }