1 // Copyright (c) 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 "base/memory/ref_counted.h"
6 #include "testing/gtest/include/gtest/gtest.h"
7
8 namespace {
9
10 class SelfAssign : public base::RefCounted<SelfAssign> {
11 friend class base::RefCounted<SelfAssign>;
12
~SelfAssign()13 ~SelfAssign() {}
14 };
15
16 class CheckDerivedMemberAccess : public scoped_refptr<SelfAssign> {
17 public:
CheckDerivedMemberAccess()18 CheckDerivedMemberAccess() {
19 // This shouldn't compile if we don't have access to the member variable.
20 SelfAssign** pptr = &ptr_;
21 EXPECT_EQ(*pptr, ptr_);
22 }
23 };
24
25 class ScopedRefPtrToSelf : public base::RefCounted<ScopedRefPtrToSelf> {
26 public:
ScopedRefPtrToSelf()27 ScopedRefPtrToSelf() : self_ptr_(this) {}
28
was_destroyed()29 static bool was_destroyed() { return was_destroyed_; }
30
SelfDestruct()31 void SelfDestruct() { self_ptr_ = NULL; }
32
33 private:
34 friend class base::RefCounted<ScopedRefPtrToSelf>;
~ScopedRefPtrToSelf()35 ~ScopedRefPtrToSelf() { was_destroyed_ = true; }
36
37 static bool was_destroyed_;
38
39 scoped_refptr<ScopedRefPtrToSelf> self_ptr_;
40 };
41
42 bool ScopedRefPtrToSelf::was_destroyed_ = false;
43
44 } // end namespace
45
TEST(RefCountedUnitTest,TestSelfAssignment)46 TEST(RefCountedUnitTest, TestSelfAssignment) {
47 SelfAssign* p = new SelfAssign;
48 scoped_refptr<SelfAssign> var(p);
49 var = var;
50 EXPECT_EQ(var.get(), p);
51 }
52
TEST(RefCountedUnitTest,ScopedRefPtrMemberAccess)53 TEST(RefCountedUnitTest, ScopedRefPtrMemberAccess) {
54 CheckDerivedMemberAccess check;
55 }
56
TEST(RefCountedUnitTest,ScopedRefPtrToSelf)57 TEST(RefCountedUnitTest, ScopedRefPtrToSelf) {
58 ScopedRefPtrToSelf* check = new ScopedRefPtrToSelf();
59 EXPECT_FALSE(ScopedRefPtrToSelf::was_destroyed());
60 check->SelfDestruct();
61 EXPECT_TRUE(ScopedRefPtrToSelf::was_destroyed());
62 }
63
TEST(RefCountedUnitTest,ScopedRefPtrBooleanOperations)64 TEST(RefCountedUnitTest, ScopedRefPtrBooleanOperations) {
65 scoped_refptr<SelfAssign> p1 = new SelfAssign;
66 scoped_refptr<SelfAssign> p2;
67
68 EXPECT_TRUE(p1);
69 EXPECT_FALSE(!p1);
70
71 EXPECT_TRUE(!p2);
72 EXPECT_FALSE(p2);
73
74 EXPECT_NE(p1, p2);
75
76 SelfAssign* raw_p = new SelfAssign;
77 p2 = raw_p;
78 EXPECT_NE(p1, p2);
79 EXPECT_EQ(raw_p, p2);
80
81 p2 = p1;
82 EXPECT_NE(raw_p, p2);
83 EXPECT_EQ(p1, p2);
84 }
85