weakblockretain.m   [plain text]

/*
* Copyright (c) 2010 Apple Inc. All rights reserved.
*
* @APPLE_LLVM_LICENSE_HEADER@
*/

//
// weakblockretain.m
// testObjects
//
// Created by Blaine Garst on 11/3/08.
// Copyright 2008 __MyCompanyName__. All rights reserved.
//
// TEST_CFLAGS -framework Foundation

// rdar://5847976
// Test that weak block variables don't retain/release their contents

#import <Foundation/Foundation.h>
#import <Block.h>
#import "test.h"

int RetainCalled;
int ReleaseCalled;

@interface TestObject : NSObject
@end

@implementation TestObject

- (id)retain {
RetainCalled = 1;
return [super retain];
}
- (oneway void)release {
ReleaseCalled = 1;
[super release];
}

void testLocalScope(void) {
__block TestObject *__weak to __unused = [[TestObject alloc] init];
// when we leave the scope a byref release call is made
// this recovers the __block storage but leaves the contents alone
// XXX make 10^6 of these to make sure we collect 'em
}

@end

int main() {
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
testLocalScope();
if (RetainCalled || ReleaseCalled) {
fail("testLocalScope had some problems");
}

__block TestObject *__weak to = [[TestObject alloc] init];
void (^block)(void) = ^ { printf("is it still real? %p\n", to); };
(void)Block_copy(block);
if (RetainCalled) {
fail("Block_copy retain had some problems");
}
if (ReleaseCalled) {
fail("Block_copy release had some problems");
}
[pool drain];

succeed(__FILE__);
}