YYModel/YYModelTests/YYTestCustomClass.m

//
//  YYTestCustomClass.m
//  YYModel <https://github.com/ibireme/YYModel>
//
//  Created by ibireme on 15/11/29.
//  Copyright (c) 2015 ibireme.
//
//  This source code is licensed under the MIT-style license found in the
//  LICENSE file in the root directory of this source tree.
//

#import <XCTest/XCTest.h>
#import "YYModel.h"

@interface YYBaseUser : NSObject
@property uint64_t uid;
@property NSString *name;
@end


@interface YYLocalUser : YYBaseUser
@property NSString *localName;
@end
@implementation YYLocalUser
@end

@interface YYRemoteUser : YYBaseUser
@property NSString *remoteName;
@end
@implementation YYRemoteUser
@end


@implementation YYBaseUser
+ (Class)modelCustomClassForDictionary:(NSDictionary*)dictionary {
    if (dictionary[@"localName"]) {
        return [YYLocalUser class];
    } else if (dictionary[@"remoteName"]) {
        return [YYRemoteUser class];
    }
    return [YYBaseUser class];
}
@end

@interface YYTestCustomClassModel : NSObject
@property (nonatomic, strong) NSArray *users;
@property (nonatomic, strong) NSSet *userSet;
@property (nonatomic, strong) YYBaseUser *user;
@end

@implementation YYTestCustomClassModel

+ (NSDictionary *)modelContainerPropertyGenericClass {
    return @{@"users" : YYBaseUser.class,
             @"userSet" : YYBaseUser.class};
}
+ (Class)modelCustomClassForDictionary:(NSDictionary*)dictionary {
    if (dictionary[@"localName"]) {
        return [YYLocalUser class];
    } else if (dictionary[@"remoteName"]) {
        return [YYRemoteUser class];
    }
    return nil;
}
@end


@interface YYTestCustomClass : XCTestCase

@end

@implementation YYTestCustomClass

- (void)test {
    YYTestCustomClassModel *model;
    YYBaseUser *user;
    
    NSDictionary *jsonUserBase = @{@"uid" : @123, @"name" : @"Harry"};
    NSDictionary *jsonUserLocal = @{@"uid" : @123, @"name" : @"Harry", @"localName" : @"HarryLocal"};
    NSDictionary *jsonUserRemote = @{@"uid" : @123, @"name" : @"Harry", @"remoteName" : @"HarryRemote"};
    
    user = [YYBaseUser yy_modelWithDictionary:jsonUserBase];
    XCTAssert([user isMemberOfClass:[YYBaseUser class]]);
    
    user = [YYBaseUser yy_modelWithDictionary:jsonUserLocal];
    XCTAssert([user isMemberOfClass:[YYLocalUser class]]);
    
    user = [YYBaseUser yy_modelWithDictionary:jsonUserRemote];
    XCTAssert([user isMemberOfClass:[YYRemoteUser class]]);
    
    
    model = [YYTestCustomClassModel yy_modelWithJSON:@{@"user" : jsonUserLocal}];
    XCTAssert([model.user isMemberOfClass:[YYLocalUser class]]);
    
    model = [YYTestCustomClassModel yy_modelWithJSON:@{@"users" : @[jsonUserBase, jsonUserLocal, jsonUserRemote]}];
    XCTAssert([model.users[0] isMemberOfClass:[YYBaseUser class]]);
    XCTAssert([model.users[1] isMemberOfClass:[YYLocalUser class]]);
    XCTAssert([model.users[2] isMemberOfClass:[YYRemoteUser class]]);
    
    model = [YYTestCustomClassModel yy_modelWithJSON:@{@"userSet" : @[jsonUserBase, jsonUserLocal, jsonUserRemote]}];
    XCTAssert([model.userSet.anyObject isKindOfClass:[YYBaseUser class]]);
}

@end