unique_ptr_Y.pass.cpp revision b64f8b07c104c6cc986570ac8ee0ed16a9f23976 
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// shared_ptr
13
14// template <class Y, class D> shared_ptr& operator=(unique_ptr<Y, D>&& r);
15
16#include <memory>
17#include <type_traits>
18#include <cassert>
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::unique_ptr<A> pA(new A);
47        A* ptrA = pA.get();
48        {
49            std::shared_ptr<B> pB(new B);
50            pB = std::move(pA);
51            assert(B::count == 1);
52            assert(A::count == 1);
53            assert(pB.use_count() == 1);
54            assert(pA.get() == 0);
55            assert(pB.get() == ptrA);
56        }
57        assert(B::count == 0);
58        assert(A::count == 0);
59    }
60    assert(B::count == 0);
61    assert(A::count == 0);
62    {
63        std::unique_ptr<A> pA;
64        A* ptrA = pA.get();
65        {
66            std::shared_ptr<B> pB(new B);
67            pB = std::move(pA);
68            assert(B::count == 0);
69            assert(A::count == 0);
70            assert(pB.use_count() == 1);
71            assert(pA.get() == 0);
72            assert(pB.get() == ptrA);
73        }
74        assert(B::count == 0);
75        assert(A::count == 0);
76    }
77    assert(B::count == 0);
78    assert(A::count == 0);
79    {
80        std::unique_ptr<A> pA(new A);
81        A* ptrA = pA.get();
82        {
83            std::shared_ptr<B> pB;
84            pB = std::move(pA);
85            assert(B::count == 1);
86            assert(A::count == 1);
87            assert(pB.use_count() == 1);
88            assert(pA.get() == 0);
89            assert(pB.get() == ptrA);
90        }
91        assert(B::count == 0);
92        assert(A::count == 0);
93    }
94    assert(B::count == 0);
95    assert(A::count == 0);
96    {
97        std::unique_ptr<A> pA;
98        A* ptrA = pA.get();
99        {
100            std::shared_ptr<B> pB;
101            pB = std::move(pA);
102            assert(B::count == 0);
103            assert(A::count == 0);
104            assert(pB.use_count() == 1);
105            assert(pA.get() == 0);
106            assert(pB.get() == ptrA);
107        }
108        assert(B::count == 0);
109        assert(A::count == 0);
110    }
111    assert(B::count == 0);
112    assert(A::count == 0);
113}
114