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 // unique_ptr 13 14 // Test unique_ptr(pointer, deleter) ctor 15 16 #include <memory> 17 #include <cassert> 18 19 // unique_ptr(pointer, deleter) should work with derived pointers 20 21 struct A 22 { 23 static int count; 24 A() {++count;} 25 A(const A&) {++count;} 26 virtual ~A() {--count;} 27 }; 28 29 int A::count = 0; 30 31 struct B 32 : public A 33 { 34 static int count; 35 B() {++count;} 36 B(const B&) {++count;} 37 virtual ~B() {--count;} 38 }; 39 40 int B::count = 0; 41 42 class Deleter 43 { 44 int state_; 45 46 public: 47 Deleter() : state_(5) {} 48 49 int state() const {return state_;} 50 51 void operator()(A* p) {delete p;} 52 }; 53 54 int main() 55 { 56 { 57 B* p = new B; 58 assert(A::count == 1); 59 assert(B::count == 1); 60 std::unique_ptr<A, Deleter> s(p, Deleter()); 61 assert(s.get() == p); 62 assert(s.get_deleter().state() == 5); 63 } 64 assert(A::count == 0); 65 assert(B::count == 0); 66 } 67
