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