Collection/CollectionTests/RVArrayHighOrderTest.m at master · BadChoice/Collection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
//
//  RVArrayHighOrderTest.m
//  CollectionTests
//
//  Created by Badchoice on 10/10/17.
//  Copyright © 2017 Revo. All rights reserved.
//

#import <XCTest/XCTest.h>
#import "NSArray+HighOrder.h"

@interface TestClass : NSObject
@property(nonatomic) BOOL available;
@end

@implementation TestClass : NSObject

+(TestClass*)make:(BOOL)available{
    TestClass * tc  = [TestClass new];
    tc.available    = available;
    return tc;
}

-(BOOL)isAvailable{
    return self.available;
}

-(NSString*)hello{
    return @"world";
}

-(NSString*)echo:(NSString*)echo{
    return echo;
}

@end


@interface RVArrayHighOrderTest : XCTestCase

@end

@implementation RVArrayHighOrderTest

- (void)setUp {
    [super setUp];
}

- (void)tearDown {
    [super tearDown];
}

-(void)test_map{
    NSArray* original  = @[[TestClass new], [TestClass new], [TestClass new]];
    NSArray * result   = [original map_:@selector(hello)];

    XCTAssertEqual(@"world", result.firstObject);
}

-(void)test_map_with_object{
    NSArray* original  = @[[TestClass new], [TestClass new], [TestClass new]];
    NSArray * result   = [original map_:@selector(echo:) withObject:@"baby"];

    XCTAssertEqual(@"baby", result.firstObject);
}

-(void)test_filter{
    NSArray* original  = @[[TestClass make:true], [TestClass make:true], [TestClass make:false]];
    NSArray * result   = [original filter_:@selector(isAvailable)];


    XCTAssertEqual(2, result.count);
}

-(void)test_reject{
    NSArray* original  = @[[TestClass make:true], [TestClass make:true], [TestClass make:false]];
    NSArray * result   = [original reject_:@selector(isAvailable)];


    XCTAssertEqual(1, result.count);
}

@end