Add PassRef and createRefCounted so we can improve creation of RefCounted objects
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WTF / Ref.cpp
1 /*
2  * Copyright (C) 2013 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27
28 #include "RefLogger.h"
29 #include <wtf/PassRef.h>
30 #include <wtf/Ref.h>
31 #include <wtf/RefPtr.h>
32
33 namespace TestWebKitAPI {
34
35 TEST(WTF_Ref, Basic)
36 {
37     DerivedRefLogger a("a");
38
39     {
40         Ref<RefLogger> ptr(a);
41         ASSERT_EQ(&a, &ptr.get());
42         ASSERT_EQ(&a.name, &ptr->name);
43     }
44     ASSERT_STREQ("ref(a) deref(a) ", takeLogStr().c_str());
45
46     {
47         Ref<RefLogger> ptr(adoptRef(a));
48         ASSERT_EQ(&a, &ptr.get());
49         ASSERT_EQ(&a.name, &ptr->name);
50     }
51     ASSERT_STREQ("deref(a) ", takeLogStr().c_str());
52 }
53
54 TEST(WTF_Ref, Assignment)
55 {
56     DerivedRefLogger a("a");
57     RefLogger b("b");
58     DerivedRefLogger c("c");
59
60     {
61         Ref<RefLogger> ptr(a);
62         ASSERT_EQ(&a, &ptr.get());
63         log() << "| ";
64         ptr = b;
65         ASSERT_EQ(&b, &ptr.get());
66         log() << "| ";
67     }
68     ASSERT_STREQ("ref(a) | ref(b) deref(a) | deref(b) ", takeLogStr().c_str());
69
70     {
71         Ref<RefLogger> ptr(a);
72         ASSERT_EQ(&a, &ptr.get());
73         log() << "| ";
74         ptr = c;
75         ASSERT_EQ(&c, &ptr.get());
76         log() << "| ";
77     }
78     ASSERT_STREQ("ref(a) | ref(c) deref(a) | deref(c) ", takeLogStr().c_str());
79
80     {
81         Ref<RefLogger> ptr(a);
82         ASSERT_EQ(&a, &ptr.get());
83         log() << "| ";
84         ptr = adoptRef(b);
85         ASSERT_EQ(&b, &ptr.get());
86         log() << "| ";
87     }
88     ASSERT_STREQ("ref(a) | deref(a) | deref(b) ", takeLogStr().c_str());
89
90     {
91         Ref<RefLogger> ptr(a);
92         ASSERT_EQ(&a, &ptr.get());
93         log() << "| ";
94         ptr = adoptRef(c);
95         ASSERT_EQ(&c, &ptr.get());
96         log() << "| ";
97     }
98     ASSERT_STREQ("ref(a) | deref(a) | deref(c) ", takeLogStr().c_str());
99 }
100
101 PassRef<RefLogger> passWithPassRef(PassRef<RefLogger> reference)
102 {
103     return reference;
104 }
105
106 RefPtr<RefLogger> passWithPassRefPtr(PassRefPtr<RefLogger> reference)
107 {
108     return reference;
109 }
110
111 TEST(WTF_Ref, ReturnValue)
112 {
113     DerivedRefLogger a("a");
114     RefLogger b("b");
115     DerivedRefLogger c("c");
116
117     {
118         Ref<RefLogger> ptr(passWithPassRef(a));
119         ASSERT_EQ(&a, &ptr.get());
120     }
121     ASSERT_STREQ("ref(a) deref(a) ", takeLogStr().c_str());
122
123     {
124         Ref<RefLogger> ptr(a);
125         ASSERT_EQ(&a, &ptr.get());
126         log() << "| ";
127         ptr = passWithPassRef(b);
128         ASSERT_EQ(&b, &ptr.get());
129         log() << "| ";
130     }
131     ASSERT_STREQ("ref(a) | ref(b) deref(a) | deref(b) ", takeLogStr().c_str());
132
133     {
134         RefPtr<RefLogger> ptr(passWithPassRef(a));
135         ASSERT_EQ(&a, ptr.get());
136     }
137     ASSERT_STREQ("ref(a) deref(a) ", takeLogStr().c_str());
138
139     {
140         RefPtr<RefLogger> ptr(passWithPassRefPtr(passWithPassRef(a)));
141         ASSERT_EQ(&a, ptr.get());
142     }
143     ASSERT_STREQ("ref(a) deref(a) ", takeLogStr().c_str());
144 }
145
146 } // namespace TestWebKitAPI