do {

id userInfoValue = [[entity userInfo] valueForKey:RKEntityIdentificationAttributesUserInfoKey];

if (userInfoValue) {

NSArray *attributeNames = [userInfoValue isKindOfClass:[NSArray class]] ? userInfoValue : @[ userInfoValue ];

NSMutableArray *attributes = [NSMutableArray arrayWithCapacity:[attributeNames count]];

[attributeNames enumerateObjectsUsingBlock:^(NSString *attributeName, NSUInteger idx, BOOL *stop) {

if (! [attributeName isKindOfClass:[NSString class]]) {

[NSException raise:NSInvalidArgumentException format:@"Invalid value given in user info key '%@' of entity '%@': expected an `NSString` or `NSArray` of strings, instead got '%@' (%@)", RKEntityIdentificationAttributesUserInfoKey, [entity name], attributeName, [attributeName class]];

}

NSAttributeDescription *attribute = [[entity attributesByName] valueForKey:attributeName];

if (! attribute) {

[NSException raise:NSInvalidArgumentException format:@"Invalid identifier attribute specified in user info key '%@' of entity '%@': no attribue was found with the name '%@'", RKEntityIdentificationAttributesUserInfoKey, [entity name], attributeName];

}

[attributes addObject:attribute];

}];

return attributes;

}

entity = [entity superentity];

} while (entity);

- (void)testEntityIdentifierInferenceSearchesParentEntities

{

NSEntityDescription *entity = [[NSEntityDescription alloc] init];

[entity setName:@"Monkey"];

NSEntityDescription *parentEntity = [[NSEntityDescription alloc] init];

[parentEntity setName:@"Parent"];

[parentEntity setSubentities:@[ entity ]];

NSAttributeDescription *identifierAttribute = [NSAttributeDescription new];

[identifierAttribute setName:@"monkeyID"];

[parentEntity setProperties:@[ identifierAttribute ]];

NSArray *identificationAttributes = RKIdentificationAttributesInferredFromEntity(entity);

expect(identificationAttributes).notTo.beNil();

NSArray *attributeNames = @[ @"monkeyID" ];

expect([identificationAttributes valueForKey:@"name"]).to.equal(attributeNames);

}

- (void)testEntityIdentifierInferenceFromUserInfoSearchesParentEntities

{

NSEntityDescription *entity = [[NSEntityDescription alloc] init];

[entity setName:@"Monkey"];

NSAttributeDescription *identifierAttribute = [NSAttributeDescription new];

[identifierAttribute setName:@"monkeyID"]; // We ignore this by specifying the userInfo key

NSAttributeDescription *nameAttribute = [NSAttributeDescription new];

[nameAttribute setName:@"name"];

[entity setProperties:@[ identifierAttribute, nameAttribute ]];

[entity setUserInfo:@{ RKEntityIdentificationAttributesUserInfoKey: @"name" }];

NSEntityDescription *subentity = [NSEntityDescription new];

[subentity setName:@"SubMonkey"];

[entity setSubentities:@[ subentity ]];

NSArray *identificationAttributes = RKIdentificationAttributesInferredFromEntity(subentity);

expect(identificationAttributes).notTo.beNil();

NSArray *attributeNames = @[ @"name" ];

expect([identificationAttributes valueForKey:@"name"]).to.equal(attributeNames);

}