1 //===----------------------------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 // <memory>
11 
12 // shared_ptr
13 
14 // template<class Y, class D> void reset(Y* p, D d);
15 
16 #include <memory>
17 #include <cassert>
18 #include "../test_deleter.h"
19 
20 struct B
21 {
22     static int count;
23 
BB24     B() {++count;}
BB25     B(const B&) {++count;}
~BB26     virtual ~B() {--count;}
27 };
28 
29 int B::count = 0;
30 
31 struct A
32     : public B
33 {
34     static int count;
35 
AA36     A() {++count;}
AA37     A(const A&) {++count;}
~AA38     ~A() {--count;}
39 };
40 
41 int A::count = 0;
42 
main()43 int main()
44 {
45     {
46         std::shared_ptr<B> p(new B);
47         A* ptr = new A;
48         p.reset(ptr, test_deleter<A>(3));
49         assert(A::count == 1);
50         assert(B::count == 1);
51         assert(p.use_count() == 1);
52         assert(p.get() == ptr);
53         test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
54         assert(test_deleter<A>::count == 1);
55         assert(test_deleter<A>::dealloc_count == 0);
56         assert(d);
57         assert(d->state() == 3);
58     }
59     assert(A::count == 0);
60     assert(test_deleter<A>::count == 0);
61     assert(test_deleter<A>::dealloc_count == 1);
62     {
63         std::shared_ptr<B> p;
64         A* ptr = new A;
65         p.reset(ptr, test_deleter<A>(3));
66         assert(A::count == 1);
67         assert(B::count == 1);
68         assert(p.use_count() == 1);
69         assert(p.get() == ptr);
70         test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p);
71         assert(test_deleter<A>::count == 1);
72         assert(test_deleter<A>::dealloc_count == 1);
73         assert(d);
74         assert(d->state() == 3);
75     }
76     assert(A::count == 0);
77     assert(test_deleter<A>::count == 0);
78     assert(test_deleter<A>::dealloc_count == 2);
79 }
80