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 21struct A 22{ 23 static int count; 24 A() {++count;} 25 A(const A&) {++count;} 26 virtual ~A() {--count;} 27}; 28 29int A::count = 0; 30 31struct B 32 : public A 33{ 34 static int count; 35 B() {++count;} 36 B(const B&) {++count;} 37 virtual ~B() {--count;} 38}; 39 40int B::count = 0; 41 42class Deleter 43{ 44 int state_; 45 46public: 47 Deleter() : state_(5) {} 48 49 int state() const {return state_;} 50 51 void operator()(A* p) {delete p;} 52}; 53 54int 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