NSArray+Map

gistfile1.m

#import "NSArray+Map.h"

@implementation NSArray (Map)

- (NSArray *)map:(id(^)(id))block {
    NSMutableArray * newArray = [NSMutableArray array];
    for (id item in self) {
        id obj = block(item);
        [newArray addObject:obj];
    }
    return newArray;
}

- (id)find:(BOOL(^)(id))block {
    for (id item in self) {
        if(block(item))return item;
    }
    return nil;
}

- (NSArray *)findAll:(BOOL(^)(id))block {
    NSMutableArray * newArray = [NSMutableArray array];
    for (id item in self) {
        if(block(item)) [newArray addObject:item];
    }
    return newArray;
}

@end





#import "Kiwi.h"
#import "NSArray+Map.h"

SPEC_BEGIN(NSArray_MapSpec)

describe(@"Map", ^{
    it(@"works", ^{
        
        NSArray * array = [NSArray arrayWithObjects:@"A", @"B", @"C",nil];
        
        NSArray * result = [array map:^(id item) {
            return [NSString stringWithFormat:@"mapped%@", item];
        }];
        
        [[[result objectAtIndex:0] should] equal:@"mappedA"];
        [[[result objectAtIndex:1] should] equal:@"mappedB"];
        [[[result objectAtIndex:2] should] equal:@"mappedC"];
    });
});

describe(@"FindAll", ^{
    it(@"works", ^{
        NSArray * array = [NSArray arrayWithObjects:@"ABC", @"BCD", @"CDE",nil];
        
        NSArray * result = [array findAll:^(id item) {
            NSRange match = [item rangeOfString:@"CD"];
            return (BOOL)(match.location != NSNotFound);
        }];
        
        [[[result objectAtIndex:0] should] equal:@"BCD"];
        [[[result objectAtIndex:1] should] equal:@"CDE"];
    });
});

describe(@"Find", ^{
    it(@"works", ^{
        NSArray * array = [NSArray arrayWithObjects:@"ABC", @"BCD", @"CDE",nil];
        
        NSString * result = [array find:^(id item) {
            NSRange match = [item rangeOfString:@"CD"];
            return (BOOL)(match.location != NSNotFound);
        }];
        
        [[result should] equal:@"BCD"];
    });
});

SPEC_END