__blockObjectAssign.m   [plain text]

//
//  __blockObjectAssign.m
//  testObjects
//
//  Created by Blaine Garst on 2/5/09.
//  Copyright 2009 Apple. All rights reserved.
//

//  CONFIG RR -C99

// tests whether assigning to a __block id variable works in a reasonable way


#import <Foundation/Foundation.h>
#include <stdio.h>
#include <Block.h>


@interface TestObject : NSObject {
    int version;
}
- (void)hi;
@end

int AllocationCounter = 0;
int DellocationCounter = 0;

@implementation TestObject


- init {
    version = AllocationCounter++;
    return self;
}

- (void)hi {
    printf("hi from %p, #%d\n", self, version);
}

- (void)dealloc {
    printf("dealloc %p, #%d called\n", self, version);
    ++DellocationCounter;
    [super dealloc];
}


@end

void testFunction(bool doExecute, bool doCopy) {
    __block id a = [[TestObject alloc] init];
    printf("testing - will execute? %d\n", doExecute);
    void (^changeA)(void) = ^{
        [a hi];
        [a release];
        a = [[TestObject alloc] init];
        [a hi];
    };
    if (doCopy) changeA = [changeA copy];
    if (doExecute) changeA();
    if (doCopy) [changeA release];
    [a release];
    printf("done with explict releasing, implicit to follow\n");
}

int main(int argc, char *argv[]) {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    testFunction(false, true);
    testFunction(true, true);
    testFunction(false, false);
    testFunction(true, true);
    [pool drain];
    if (DellocationCounter != AllocationCounter) {
        printf("only recovered %d of %d objects\n", DellocationCounter, AllocationCounter);
        return 1;
    }
    printf("%s: success\n", argv[0]);
    return 0;
}