__blockObjectAssign.m   [plain text]

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

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

//  TEST_CONFIG
//  TEST_CFLAGS -framework Foundation

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


#import <Foundation/Foundation.h>
#import <stdio.h>
#import <Block.h>
#import "test.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() {
    NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
    testFunction(false, true);
    testFunction(true, true);
    testFunction(false, false);
    testFunction(true, true);
    [pool drain];
    if (! objc_collectingEnabled()) {
        if (DellocationCounter != AllocationCounter) {
            fail("only recovered %d of %d objects", DellocationCounter, AllocationCounter);
        }
    }

    succeed(__FILE__);
}