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 351
