reset_pointer_deleter.pass.cpp revision bc8d3f97eb5c958007f2713238472e0c1c8fe02c
1//===----------------------------------------------------------------------===// 2// 3// ��������������������The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. 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 20struct B 21{ 22 static int count; 23 24 B() {++count;} 25 B(const B&) {++count;} 26 virtual ~B() {--count;} 27}; 28 29int B::count = 0; 30 31struct A 32 : public B 33{ 34 static int count; 35 36 A() {++count;} 37 A(const A&) {++count;} 38 ~A() {--count;} 39}; 40 41int A::count = 0; 42 43int 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