base/mac/objc_property_releaser_unittest.mm

   1 // Copyright (c) 2011 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 #import <Foundation/Foundation.h>
   6
   7 #import "base/mac/objc_property_releaser.h"
   8 #import "base/mac/scoped_nsautorelease_pool.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 // "When I'm alone, I count myself."
  12 //   --Count von Count, http://www.youtube.com/watch?v=FKzszqa9WA4
  13
  14 namespace {
  15
  16 // The number of CountVonCounts outstanding.
  17 int ah_ah_ah;
  18
  19 // NumberHolder exists to exercise the property attribute string parser by
  20 // providing a named struct and an anonymous union.
  21 struct NumberHolder {
  22   union {
  23     long long sixty_four;
  24     int thirty_two;
  25     short sixteen;
  26     char eight;
  27   } what;
  28   enum {
  29     SIXTY_FOUR,
  30     THIRTY_TWO,
  31     SIXTEEN,
  32     EIGHT
  33   } how;
  34 };
  35
  36 }  // namespace
  37
  38 @interface CountVonCount : NSObject<NSCopying>
  39
  40 + (CountVonCount*)countVonCount;
  41
  42 @end  // @interface CountVonCount
  43
  44 @implementation CountVonCount
  45
  46 + (CountVonCount*)countVonCount {
  47   return [[[CountVonCount alloc] init] autorelease];
  48 }
  49
  50 - (id)init {
  51   ++ah_ah_ah;
  52   return [super init];
  53 }
  54
  55 - (void)dealloc {
  56   --ah_ah_ah;
  57   [super dealloc];
  58 }
  59
  60 - (id)copyWithZone:(NSZone*)zone {
  61   return [[CountVonCount allocWithZone:zone] init];
  62 }
  63
  64 @end  // @implementation CountVonCount
  65
  66 @interface ObjCPropertyTestBase : NSObject {
  67  @private
  68   CountVonCount* baseCvcRetain_;
  69   CountVonCount* baseCvcCopy_;
  70   CountVonCount* baseCvcAssign_;
  71   CountVonCount* baseCvcNotProperty_;
  72   CountVonCount* baseCvcNil_;
  73   CountVonCount* baseCvcCustom_;
  74   int baseInt_;
  75   double baseDouble_;
  76   void* basePointer_;
  77   NumberHolder baseStruct_;
  78
  79   base::mac::ObjCPropertyReleaser propertyReleaser_ObjCPropertyTestBase_;
  80 }
  81
  82 @property(retain, nonatomic) CountVonCount* baseCvcRetain;
  83 @property(copy, nonatomic) CountVonCount* baseCvcCopy;
  84 @property(assign, nonatomic) CountVonCount* baseCvcAssign;
  85 @property(retain, nonatomic) CountVonCount* baseCvcNil;
  86 @property(retain, nonatomic, getter=baseCustom, setter=setBaseCustom:)
  87     CountVonCount* baseCvcCustom;
  88 @property(retain, nonatomic) CountVonCount* baseCvcDynamic;
  89 @property(assign, nonatomic) int baseInt;
  90 @property(assign, nonatomic) double baseDouble;
  91 @property(assign, nonatomic) void* basePointer;
  92 @property(assign, nonatomic) NumberHolder baseStruct;
  93
  94 - (void)setBaseCvcNotProperty:(CountVonCount*)cvc;
  95
  96 @end  // @interface ObjCPropertyTestBase
  97
  98 @implementation ObjCPropertyTestBase
  99
 100 @synthesize baseCvcRetain = baseCvcRetain_;
 101 @synthesize baseCvcCopy = baseCvcCopy_;
 102 @synthesize baseCvcAssign = baseCvcAssign_;
 103 @synthesize baseCvcNil = baseCvcNil_;
 104 @synthesize baseCvcCustom = baseCvcCustom_;
 105 @dynamic baseCvcDynamic;
 106 @synthesize baseInt = baseInt_;
 107 @synthesize baseDouble = baseDouble_;
 108 @synthesize basePointer = basePointer_;
 109 @synthesize baseStruct = baseStruct_;
 110
 111 - (id)init {
 112   if ((self = [super init])) {
 113     propertyReleaser_ObjCPropertyTestBase_.Init(
 114         self, [ObjCPropertyTestBase class]);
 115   }
 116   return self;
 117 }
 118
 119 - (void)dealloc {
 120   [baseCvcNotProperty_ release];
 121   [super dealloc];
 122 }
 123
 124 - (void)setBaseCvcNotProperty:(CountVonCount*)cvc {
 125   if (cvc != baseCvcNotProperty_) {
 126     [baseCvcNotProperty_ release];
 127     baseCvcNotProperty_ = [cvc retain];
 128   }
 129 }
 130
 131 @end  // @implementation ObjCPropertyTestBase
 132
 133 @protocol ObjCPropertyTestProtocol
 134
 135 @property(retain, nonatomic) CountVonCount* protoCvcRetain;
 136 @property(copy, nonatomic) CountVonCount* protoCvcCopy;
 137 @property(assign, nonatomic) CountVonCount* protoCvcAssign;
 138 @property(retain, nonatomic) CountVonCount* protoCvcNil;
 139 @property(retain, nonatomic, getter=protoCustom, setter=setProtoCustom:)
 140     CountVonCount* protoCvcCustom;
 141 @property(retain, nonatomic) CountVonCount* protoCvcDynamic;
 142 @property(assign, nonatomic) int protoInt;
 143 @property(assign, nonatomic) double protoDouble;
 144 @property(assign, nonatomic) void* protoPointer;
 145 @property(assign, nonatomic) NumberHolder protoStruct;
 146
 147 @end  // @protocol ObjCPropertyTestProtocol
 148
 149 @interface ObjCPropertyTestDerived
 150     : ObjCPropertyTestBase<ObjCPropertyTestProtocol> {
 151  @private
 152   CountVonCount* derivedCvcRetain_;
 153   CountVonCount* derivedCvcCopy_;
 154   CountVonCount* derivedCvcAssign_;
 155   CountVonCount* derivedCvcNotProperty_;
 156   CountVonCount* derivedCvcNil_;
 157   CountVonCount* derivedCvcCustom_;
 158   int derivedInt_;
 159   double derivedDouble_;
 160   void* derivedPointer_;
 161   NumberHolder derivedStruct_;
 162
 163   CountVonCount* protoCvcRetain_;
 164   CountVonCount* protoCvcCopy_;
 165   CountVonCount* protoCvcAssign_;
 166   CountVonCount* protoCvcNil_;
 167   CountVonCount* protoCvcCustom_;
 168   int protoInt_;
 169   double protoDouble_;
 170   void* protoPointer_;
 171   NumberHolder protoStruct_;
 172
 173   base::mac::ObjCPropertyReleaser propertyReleaser_ObjCPropertyTestDerived_;
 174 }
 175
 176 @property(retain, nonatomic) CountVonCount* derivedCvcRetain;
 177 @property(copy, nonatomic) CountVonCount* derivedCvcCopy;
 178 @property(assign, nonatomic) CountVonCount* derivedCvcAssign;
 179 @property(retain, nonatomic) CountVonCount* derivedCvcNil;
 180 @property(retain, nonatomic, getter=derivedCustom, setter=setDerivedCustom:)
 181     CountVonCount* derivedCvcCustom;
 182 @property(retain, nonatomic) CountVonCount* derivedCvcDynamic;
 183 @property(assign, nonatomic) int derivedInt;
 184 @property(assign, nonatomic) double derivedDouble;
 185 @property(assign, nonatomic) void* derivedPointer;
 186 @property(assign, nonatomic) NumberHolder derivedStruct;
 187
 188 - (void)setDerivedCvcNotProperty:(CountVonCount*)cvc;
 189
 190 @end  // @interface ObjCPropertyTestDerived
 191
 192 @implementation ObjCPropertyTestDerived
 193
 194 @synthesize derivedCvcRetain = derivedCvcRetain_;
 195 @synthesize derivedCvcCopy = derivedCvcCopy_;
 196 @synthesize derivedCvcAssign = derivedCvcAssign_;
 197 @synthesize derivedCvcNil = derivedCvcNil_;
 198 @synthesize derivedCvcCustom = derivedCvcCustom_;
 199 @dynamic derivedCvcDynamic;
 200 @synthesize derivedInt = derivedInt_;
 201 @synthesize derivedDouble = derivedDouble_;
 202 @synthesize derivedPointer = derivedPointer_;
 203 @synthesize derivedStruct = derivedStruct_;
 204
 205 @synthesize protoCvcRetain = protoCvcRetain_;
 206 @synthesize protoCvcCopy = protoCvcCopy_;
 207 @synthesize protoCvcAssign = protoCvcAssign_;
 208 @synthesize protoCvcNil = protoCvcNil_;
 209 @synthesize protoCvcCustom = protoCvcCustom_;
 210 @dynamic protoCvcDynamic;
 211 @synthesize protoInt = protoInt_;
 212 @synthesize protoDouble = protoDouble_;
 213 @synthesize protoPointer = protoPointer_;
 214 @synthesize protoStruct = protoStruct_;
 215
 216 - (id)init {
 217   if ((self = [super init])) {
 218     propertyReleaser_ObjCPropertyTestDerived_.Init(
 219         self, [ObjCPropertyTestDerived class]);
 220   }
 221   return self;
 222 }
 223
 224 - (void)dealloc {
 225   [derivedCvcNotProperty_ release];
 226   [super dealloc];
 227 }
 228
 229 - (void)setDerivedCvcNotProperty:(CountVonCount*)cvc {
 230   if (cvc != derivedCvcNotProperty_) {
 231     [derivedCvcNotProperty_ release];
 232     derivedCvcNotProperty_ = [cvc retain];
 233   }
 234 }
 235
 236 @end  // @implementation ObjCPropertyTestDerived
 237
 238 namespace {
 239
 240 TEST(ObjCPropertyReleaserTest, SesameStreet) {
 241   ObjCPropertyTestDerived* test_object = [[ObjCPropertyTestDerived alloc] init];
 242
 243   // Assure a clean slate.
 244   EXPECT_EQ(0, ah_ah_ah);
 245   EXPECT_EQ(1U, [test_object retainCount]);
 246
 247   CountVonCount* baseAssign = [[CountVonCount alloc] init];
 248   CountVonCount* derivedAssign = [[CountVonCount alloc] init];
 249   CountVonCount* protoAssign = [[CountVonCount alloc] init];
 250
 251   // Make sure that worked before things get more involved.
 252   EXPECT_EQ(3, ah_ah_ah);
 253
 254   {
 255     base::mac::ScopedNSAutoreleasePool pool;
 256
 257     test_object.baseCvcRetain = [CountVonCount countVonCount];
 258     test_object.baseCvcCopy = [CountVonCount countVonCount];
 259     test_object.baseCvcAssign = baseAssign;
 260     test_object.baseCvcCustom = [CountVonCount countVonCount];
 261     [test_object setBaseCvcNotProperty:[CountVonCount countVonCount]];
 262
 263     // That added 4 objects, plus 1 more that was copied.
 264     EXPECT_EQ(8, ah_ah_ah);
 265
 266     test_object.derivedCvcRetain = [CountVonCount countVonCount];
 267     test_object.derivedCvcCopy = [CountVonCount countVonCount];
 268     test_object.derivedCvcAssign = derivedAssign;
 269     test_object.derivedCvcCustom = [CountVonCount countVonCount];
 270     [test_object setDerivedCvcNotProperty:[CountVonCount countVonCount]];
 271
 272     // That added 4 objects, plus 1 more that was copied.
 273     EXPECT_EQ(13, ah_ah_ah);
 274
 275     test_object.protoCvcRetain = [CountVonCount countVonCount];
 276     test_object.protoCvcCopy = [CountVonCount countVonCount];
 277     test_object.protoCvcAssign = protoAssign;
 278     test_object.protoCvcCustom = [CountVonCount countVonCount];
 279
 280     // That added 3 objects, plus 1 more that was copied.
 281     EXPECT_EQ(17, ah_ah_ah);
 282   }
 283
 284   // Now that the autorelease pool has been popped, the 3 objects that were
 285   // copied when placed into the test object will have been deallocated.
 286   EXPECT_EQ(14, ah_ah_ah);
 287
 288   // Make sure that the setters work and have the expected semantics.
 289   test_object.baseCvcRetain = nil;
 290   test_object.baseCvcCopy = nil;
 291   test_object.baseCvcAssign = nil;
 292   test_object.baseCvcCustom = nil;
 293   test_object.derivedCvcRetain = nil;
 294   test_object.derivedCvcCopy = nil;
 295   test_object.derivedCvcAssign = nil;
 296   test_object.derivedCvcCustom = nil;
 297   test_object.protoCvcRetain = nil;
 298   test_object.protoCvcCopy = nil;
 299   test_object.protoCvcAssign = nil;
 300   test_object.protoCvcCustom = nil;
 301
 302   // The CountVonCounts marked "retain" and "copy" should have been
 303   // deallocated. Those marked assign should not have been. The only ones that
 304   // should exist now are the ones marked "assign" and the ones held in
 305   // non-property instance variables.
 306   EXPECT_EQ(5, ah_ah_ah);
 307
 308   {
 309     base::mac::ScopedNSAutoreleasePool pool;
 310
 311     // Put things back to how they were.
 312     test_object.baseCvcRetain = [CountVonCount countVonCount];
 313     test_object.baseCvcCopy = [CountVonCount countVonCount];
 314     test_object.baseCvcAssign = baseAssign;
 315     test_object.baseCvcCustom = [CountVonCount countVonCount];
 316     test_object.derivedCvcRetain = [CountVonCount countVonCount];
 317     test_object.derivedCvcCopy = [CountVonCount countVonCount];
 318     test_object.derivedCvcAssign = derivedAssign;
 319     test_object.derivedCvcCustom = [CountVonCount countVonCount];
 320     test_object.protoCvcRetain = [CountVonCount countVonCount];
 321     test_object.protoCvcCopy = [CountVonCount countVonCount];
 322     test_object.protoCvcAssign = protoAssign;
 323     test_object.protoCvcCustom = [CountVonCount countVonCount];
 324
 325     // 9 more CountVonCounts, 3 of which were copied.
 326     EXPECT_EQ(17, ah_ah_ah);
 327   }
 328
 329   // Now that the autorelease pool has been popped, the 3 copies are gone.
 330   EXPECT_EQ(14, ah_ah_ah);
 331
 332   // Releasing the test object should get rid of everything that it owns.
 333   [test_object release];
 334
 335   // The property releaser should have released all of the CountVonCounts
 336   // associated with properties marked "retain" or "copy". The -dealloc
 337   // methods in each should have released the single non-property objects in
 338   // each. Only the CountVonCounts assigned to the properties marked "assign"
 339   // should remain.
 340   EXPECT_EQ(3, ah_ah_ah);
 341
 342   [baseAssign release];
 343   [derivedAssign release];
 344   [protoAssign release];
 345
 346   // Zero! Zero counts! Ah, ah, ah.
 347   EXPECT_EQ(0, ah_ah_ah);
 348 }
 349
 350 }  // namespace