unique_ptr.pass.cpp revision 541cb301a123086ac7889215aa032f7f81ea96b9
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> explicit shared_ptr(unique_ptr<Y, D>&&r); 13 14#include <memory> 15#include <new> 16#include <cstdlib> 17#include <cassert> 18 19bool throw_next = false; 20 21void* operator new(std::size_t s) throw(std::bad_alloc) 22{ 23 if (throw_next) 24 throw std::bad_alloc(); 25 return std::malloc(s); 26} 27 28void operator delete(void* p) throw() 29{ 30 std::free(p); 31} 32 33struct B 34{ 35 static int count; 36 37 B() {++count;} 38 B(const B&) {++count;} 39 virtual ~B() {--count;} 40}; 41 42int B::count = 0; 43 44struct A 45 : public B 46{ 47 static int count; 48 49 A() {++count;} 50 A(const A&) {++count;} 51 ~A() {--count;} 52}; 53 54int A::count = 0; 55 56int main() 57{ 58 { 59 std::unique_ptr<A> ptr(new A); 60 A* raw_ptr = ptr.get(); 61 std::shared_ptr<B> p(std::move(ptr)); 62 assert(A::count == 1); 63 assert(B::count == 1); 64 assert(p.use_count() == 1); 65 assert(p.get() == raw_ptr); 66 assert(ptr.get() == 0); 67 } 68 assert(A::count == 0); 69 { 70 std::unique_ptr<A> ptr(new A); 71 A* raw_ptr = ptr.get(); 72 throw_next = true; 73 try 74 { 75 std::shared_ptr<B> p(std::move(ptr)); 76 assert(false); 77 } 78 catch (...) 79 { 80#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES 81 assert(A::count == 1); 82 assert(B::count == 1); 83 assert(ptr.get() == raw_ptr); 84#else 85 assert(A::count == 0); 86 assert(B::count == 0); 87 assert(ptr.get() == 0); 88#endif 89 } 90 } 91 assert(A::count == 0); 92} 93