base/ios/weak_nsobject_unittest.mm

   1 // Copyright 2013 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/basictypes.h"
   6 #include "base/bind.h"
   7 #include "base/ios/weak_nsobject.h"
   8 #include "base/mac/scoped_nsobject.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/single_thread_task_runner.h"
  11 #include "base/threading/thread.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace base {
  15 namespace {
  16
  17 TEST(WeakNSObjectTest, WeakNSObject) {
  18   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  19   WeakNSObject<NSObject> w1(p1);
  20   EXPECT_TRUE(w1);
  21   p1.reset();
  22   EXPECT_FALSE(w1);
  23 }
  24
  25 TEST(WeakNSObjectTest, MultipleWeakNSObject) {
  26   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  27   WeakNSObject<NSObject> w1(p1);
  28   WeakNSObject<NSObject> w2(w1);
  29   EXPECT_TRUE(w1);
  30   EXPECT_TRUE(w2);
  31   EXPECT_TRUE(w1.get() == w2.get());
  32   p1.reset();
  33   EXPECT_FALSE(w1);
  34   EXPECT_FALSE(w2);
  35 }
  36
  37 TEST(WeakNSObjectTest, WeakNSObjectDies) {
  38   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  39   {
  40     WeakNSObject<NSObject> w1(p1);
  41     EXPECT_TRUE(w1);
  42   }
  43 }
  44
  45 TEST(WeakNSObjectTest, WeakNSObjectReset) {
  46   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  47   WeakNSObject<NSObject> w1(p1);
  48   EXPECT_TRUE(w1);
  49   w1.reset();
  50   EXPECT_FALSE(w1);
  51   EXPECT_TRUE(p1);
  52   EXPECT_TRUE([p1 description]);
  53 }
  54
  55 TEST(WeakNSObjectTest, WeakNSObjectResetWithObject) {
  56   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  57   scoped_nsobject<NSObject> p2([[NSObject alloc] init]);
  58   WeakNSObject<NSObject> w1(p1);
  59   EXPECT_TRUE(w1);
  60   w1.reset(p2);
  61   EXPECT_TRUE(w1);
  62   EXPECT_TRUE([p1 description]);
  63   EXPECT_TRUE([p2 description]);
  64 }
  65
  66 TEST(WeakNSObjectTest, WeakNSObjectEmpty) {
  67   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  68   WeakNSObject<NSObject> w1;
  69   EXPECT_FALSE(w1);
  70   w1.reset(p1);
  71   EXPECT_TRUE(w1);
  72   p1.reset();
  73   EXPECT_FALSE(w1);
  74 }
  75
  76 TEST(WeakNSObjectTest, WeakNSObjectCopy) {
  77   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  78   WeakNSObject<NSObject> w1(p1);
  79   WeakNSObject<NSObject> w2(w1);
  80   EXPECT_TRUE(w1);
  81   EXPECT_TRUE(w2);
  82   p1.reset();
  83   EXPECT_FALSE(w1);
  84   EXPECT_FALSE(w2);
  85 }
  86
  87 TEST(WeakNSObjectTest, WeakNSObjectAssignment) {
  88   scoped_nsobject<NSObject> p1([[NSObject alloc] init]);
  89   WeakNSObject<NSObject> w1(p1);
  90   WeakNSObject<NSObject> w2;
  91   EXPECT_FALSE(w2);
  92   w2 = w1;
  93   EXPECT_TRUE(w1);
  94   EXPECT_TRUE(w2);
  95   p1.reset();
  96   EXPECT_FALSE(w1);
  97   EXPECT_FALSE(w2);
  98 }
  99
 100 // Touches |weak_data| by increasing its length by 1. Used to check that the
 101 // weak object can be dereferenced.
 102 void TouchWeakData(const WeakNSObject<NSMutableData>& weak_data) {
 103   if (!weak_data)
 104     return;
 105   [weak_data increaseLengthBy:1];
 106 }
 107
 108 // Makes a copy of |weak_object| on the current thread and posts a task to touch
 109 // the weak object on its original thread.
 110 void CopyWeakNSObjectAndPost(const WeakNSObject<NSMutableData>& weak_object,
 111                              scoped_refptr<SingleThreadTaskRunner> runner) {
 112   // Copy using constructor.
 113   WeakNSObject<NSMutableData> weak_copy1(weak_object);
 114   runner->PostTask(FROM_HERE, Bind(&TouchWeakData, weak_copy1));
 115   // Copy using assignment operator.
 116   WeakNSObject<NSMutableData> weak_copy2 = weak_object;
 117   runner->PostTask(FROM_HERE, Bind(&TouchWeakData, weak_copy2));
 118 }
 119
 120 // Tests that the weak object can be copied on a different thread.
 121 TEST(WeakNSObjectTest, WeakNSObjectCopyOnOtherThread) {
 122   MessageLoop loop;
 123   Thread other_thread("WeakNSObjectCopyOnOtherThread");
 124   other_thread.Start();
 125
 126   scoped_nsobject<NSMutableData> data([[NSMutableData alloc] init]);
 127   WeakNSObject<NSMutableData> weak(data);
 128
 129   scoped_refptr<SingleThreadTaskRunner> runner = loop.task_runner();
 130   other_thread.task_runner()->PostTask(
 131       FROM_HERE, Bind(&CopyWeakNSObjectAndPost, weak, runner));
 132   other_thread.Stop();
 133   loop.RunUntilIdle();
 134
 135   // Check that TouchWeakData was called and the object touched twice.
 136   EXPECT_EQ(2u, [data length]);
 137 }
 138
 139 }  // namespace
 140 }  // namespace base