1 // Copyright 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "skia/ext/refptr.h"
6
7 #include "testing/gtest/include/gtest/gtest.h"
8
9 namespace skia {
10 namespace {
11
TEST(RefPtrTest,ReferenceCounting)12 TEST(RefPtrTest, ReferenceCounting) {
13 SkRefCnt* ref = new SkRefCnt();
14 EXPECT_EQ(1, ref->getRefCnt());
15
16 {
17 // Adopt the reference from the caller on creation.
18 RefPtr<SkRefCnt> refptr1 = AdoptRef(ref);
19 EXPECT_EQ(1, ref->getRefCnt());
20 EXPECT_EQ(1, refptr1->getRefCnt());
21
22 EXPECT_EQ(ref, &*refptr1);
23 EXPECT_EQ(ref, refptr1.get());
24
25 {
26 // Take a second reference for the second instance.
27 RefPtr<SkRefCnt> refptr2(refptr1);
28 EXPECT_EQ(2, ref->getRefCnt());
29
30 RefPtr<SkRefCnt> refptr3;
31 EXPECT_FALSE(refptr3);
32
33 // Take a third reference for the third instance.
34 refptr3 = refptr1;
35 EXPECT_EQ(3, ref->getRefCnt());
36
37 // Same object, so should have the same refcount.
38 refptr2 = refptr3;
39 EXPECT_EQ(3, ref->getRefCnt());
40
41 // Drop the object from refptr2, so it should lose its reference.
42 EXPECT_TRUE(refptr2);
43 refptr2.clear();
44 EXPECT_EQ(2, ref->getRefCnt());
45
46 EXPECT_FALSE(refptr2);
47 EXPECT_EQ(NULL, refptr2.get());
48
49 EXPECT_TRUE(refptr3);
50 EXPECT_EQ(2, refptr3->getRefCnt());
51 EXPECT_EQ(ref, &*refptr3);
52 EXPECT_EQ(ref, refptr3.get());
53 }
54
55 // Drop a reference when the third object is destroyed.
56 EXPECT_EQ(1, ref->getRefCnt());
57 }
58 }
59
TEST(RefPtrTest,Construct)60 TEST(RefPtrTest, Construct) {
61 SkRefCnt* ref = new SkRefCnt();
62 EXPECT_EQ(1, ref->getRefCnt());
63
64 // Adopt the reference from the caller on creation.
65 RefPtr<SkRefCnt> refptr1(AdoptRef(ref));
66 EXPECT_EQ(1, ref->getRefCnt());
67 EXPECT_EQ(1, refptr1->getRefCnt());
68
69 EXPECT_EQ(ref, &*refptr1);
70 EXPECT_EQ(ref, refptr1.get());
71
72 RefPtr<SkRefCnt> refptr2(refptr1);
73 EXPECT_EQ(2, ref->getRefCnt());
74 }
75
TEST(RefPtrTest,DeclareAndAssign)76 TEST(RefPtrTest, DeclareAndAssign) {
77 SkRefCnt* ref = new SkRefCnt();
78 EXPECT_EQ(1, ref->getRefCnt());
79
80 // Adopt the reference from the caller on creation.
81 RefPtr<SkRefCnt> refptr1 = AdoptRef(ref);
82 EXPECT_EQ(1, ref->getRefCnt());
83 EXPECT_EQ(1, refptr1->getRefCnt());
84
85 EXPECT_EQ(ref, &*refptr1);
86 EXPECT_EQ(ref, refptr1.get());
87
88 RefPtr<SkRefCnt> refptr2 = refptr1;
89 EXPECT_EQ(2, ref->getRefCnt());
90 }
91
TEST(RefPtrTest,Assign)92 TEST(RefPtrTest, Assign) {
93 SkRefCnt* ref = new SkRefCnt();
94 EXPECT_EQ(1, ref->getRefCnt());
95
96 // Adopt the reference from the caller on creation.
97 RefPtr<SkRefCnt> refptr1;
98 refptr1 = AdoptRef(ref);
99 EXPECT_EQ(1, ref->getRefCnt());
100 EXPECT_EQ(1, refptr1->getRefCnt());
101
102 EXPECT_EQ(ref, &*refptr1);
103 EXPECT_EQ(ref, refptr1.get());
104
105 RefPtr<SkRefCnt> refptr2;
106 refptr2 = refptr1;
107 EXPECT_EQ(2, ref->getRefCnt());
108 }
109
110 class Subclass : public SkRefCnt {};
111
TEST(RefPtrTest,Upcast)112 TEST(RefPtrTest, Upcast) {
113 RefPtr<Subclass> child = AdoptRef(new Subclass());
114 EXPECT_EQ(1, child->getRefCnt());
115
116 RefPtr<SkRefCnt> parent = child;
117 EXPECT_TRUE(child);
118 EXPECT_TRUE(parent);
119
120 EXPECT_EQ(2, child->getRefCnt());
121 EXPECT_EQ(2, parent->getRefCnt());
122 }
123
124 } // namespace
125 } // namespace skia
126