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 move assignment
15 
16 // test move assignment.  Should only require a MoveConstructible deleter, or if
17 //    deleter is a reference, not even that.
18 
19 #include <memory>
20 #include <cassert>
21 
22 #include "../../deleter.h"
23 
24 struct A
25 {
26     static int count;
27     A() {++count;}
28     A(const A&) {++count;}
29     ~A() {--count;}
30 };
31 
32 int A::count = 0;
33 
34 int main()
35 {
36     {
37     std::unique_ptr<A> s1(new A);
38     A* p = s1.get();
39     std::unique_ptr<A> s2(new A);
40     assert(A::count == 2);
41     s2 = std::move(s1);
42     assert(A::count == 1);
43     assert(s2.get() == p);
44     assert(s1.get() == 0);
45     }
46     assert(A::count == 0);
47     {
48     std::unique_ptr<A, Deleter<A> > s1(new A, Deleter<A>(5));
49     A* p = s1.get();
50     std::unique_ptr<A, Deleter<A> > s2(new A);
51     assert(A::count == 2);
52     s2 = std::move(s1);
53     assert(s2.get() == p);
54     assert(s1.get() == 0);
55     assert(A::count == 1);
56     assert(s2.get_deleter().state() == 5);
57     assert(s1.get_deleter().state() == 0);
58     }
59     assert(A::count == 0);
60     {
61     CDeleter<A> d1(5);
62     std::unique_ptr<A, CDeleter<A>&> s1(new A, d1);
63     A* p = s1.get();
64     CDeleter<A> d2(6);
65     std::unique_ptr<A, CDeleter<A>&> s2(new A, d2);
66     s2 = std::move(s1);
67     assert(s2.get() == p);
68     assert(s1.get() == 0);
69     assert(A::count == 1);
70     assert(d1.state() == 5);
71     assert(d2.state() == 5);
72     }
73     assert(A::count == 0);
74 }
75