base/memory/ref_counted_unittest.cc

   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
  13   ~SelfAssign() {}
  14 };
  15
  16 class CheckDerivedMemberAccess : public scoped_refptr<SelfAssign> {
  17  public:
  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:
  27   ScopedRefPtrToSelf() : self_ptr_(this) {}
  28
  29   static bool was_destroyed() { return was_destroyed_; }
  30
  31   void SelfDestruct() { self_ptr_ = NULL; }
  32
  33  private:
  34   friend class base::RefCounted<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
  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
  53 TEST(RefCountedUnitTest, ScopedRefPtrMemberAccess) {
  54   CheckDerivedMemberAccess check;
  55 }
  56
  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 }