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 // template<class Y, class D, class A> shared_ptr(Y* p, D d, A a); 13 14 #include <memory> 15 #include <cassert> 16 #include "deleter_types.h" 17 #include "test_allocator.h" 18 #include "min_allocator.h" 19 20 struct A 21 { 22 static int count; 23 24 A() {++count;} 25 A(const A&) {++count;} 26 ~A() {--count;} 27 }; 28 29 int A::count = 0; 30 31 32 int main() 33 { 34 { 35 A* ptr = new A; 36 std::shared_ptr<A> p(ptr, test_deleter<A>(3), test_allocator<A>(5)); 37 assert(A::count == 1); 38 assert(p.use_count() == 1); 39 assert(p.get() == ptr); 40 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p); 41 assert(test_deleter<A>::count == 1); 42 assert(test_deleter<A>::dealloc_count == 0); 43 assert(d); 44 assert(d->state() == 3); 45 assert(test_allocator<A>::count == 1); 46 assert(test_allocator<A>::alloc_count == 1); 47 } 48 assert(A::count == 0); 49 assert(test_deleter<A>::count == 0); 50 assert(test_deleter<A>::dealloc_count == 1); 51 assert(test_allocator<A>::count == 0); 52 assert(test_allocator<A>::alloc_count == 0); 53 test_deleter<A>::dealloc_count = 0; 54 // Test an allocator with a minimal interface 55 { 56 A* ptr = new A; 57 std::shared_ptr<A> p(ptr, test_deleter<A>(3), bare_allocator<void>()); 58 assert(A::count == 1); 59 assert(p.use_count() == 1); 60 assert(p.get() == ptr); 61 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p); 62 assert(test_deleter<A>::count == 1); 63 assert(test_deleter<A>::dealloc_count == 0); 64 assert(d); 65 assert(d->state() == 3); 66 } 67 assert(A::count == 0); 68 assert(test_deleter<A>::count == 0); 69 assert(test_deleter<A>::dealloc_count == 1); 70 test_deleter<A>::dealloc_count = 0; 71 #if TEST_STD_VER >= 11 72 // Test an allocator that returns class-type pointers 73 { 74 A* ptr = new A; 75 std::shared_ptr<A> p(ptr, test_deleter<A>(3), min_allocator<void>()); 76 assert(A::count == 1); 77 assert(p.use_count() == 1); 78 assert(p.get() == ptr); 79 test_deleter<A>* d = std::get_deleter<test_deleter<A> >(p); 80 assert(test_deleter<A>::count == 1); 81 assert(test_deleter<A>::dealloc_count == 0); 82 assert(d); 83 assert(d->state() == 3); 84 } 85 assert(A::count == 0); 86 assert(test_deleter<A>::count == 0); 87 assert(test_deleter<A>::dealloc_count == 1); 88 #endif 89 } 90
