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) ctor
15
16#include <memory>
17#include <cassert>
18
19// unique_ptr(pointer) ctor 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
46    Deleter(Deleter&);
47    Deleter& operator=(Deleter&);
48
49public:
50    Deleter() : state_(5) {}
51
52    int state() const {return state_;}
53
54    void operator()(A* p) {delete p;}
55};
56
57int main()
58{
59    {
60    B* p = new B;
61    assert(A::count == 1);
62    assert(B::count == 1);
63    std::unique_ptr<A> s(p);
64    assert(s.get() == p);
65    }
66    assert(A::count == 0);
67    assert(B::count == 0);
68    {
69    B* p = new B;
70    assert(A::count == 1);
71    assert(B::count == 1);
72    std::unique_ptr<A, Deleter> s(p);
73    assert(s.get() == p);
74    assert(s.get_deleter().state() == 5);
75    }
76    assert(A::count == 0);
77    assert(B::count == 0);
78}
79