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