unique_ptr.pass.cpp revision f5256e16dfc425c1d466f6308d4026d529ce9e0b 
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// 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        assert(A::count == 1);
81        assert(B::count == 1);
82        assert(ptr.get() == raw_ptr);
83    }
84    }
85    assert(A::count == 0);
86}
87