Files
RestKit/Tests/Logic/CoreData/RKManagedObjectMappingTest.m
2012-05-17 18:00:16 -04:00

320 lines
16 KiB
Objective-C

//
// RKManagedObjectMappingTest.m
// RestKit
//
// Created by Blake Watters on 5/31/11.
// Copyright (c) 2009-2012 RestKit. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#import "RKTestEnvironment.h"
#import "RKManagedObjectMapping.h"
#import "RKHuman.h"
#import "RKMappableObject.h"
#import "RKChild.h"
#import "RKParent.h"
#import "NSEntityDescription+RKAdditions.h"
@interface RKManagedObjectMappingTest : RKTestCase
@end
@implementation RKManagedObjectMappingTest
- (void)setUp
{
[RKTestFactory setUp];
}
- (void)tearDown
{
[RKTestFactory tearDown];
}
- (void)testShouldReturnTheDefaultValueForACoreDataAttribute {
// Load Core Data
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForEntityWithName:@"RKCat" inManagedObjectStore:store];
id value = [mapping defaultValueForMissingAttribute:@"name"];
assertThat(value, is(equalTo(@"Kitty Cat!")));
}
- (void)testShouldCreateNewInstancesOfUnmanagedObjects {
[RKTestFactory managedObjectStore];
RKObjectMapping* mapping = [RKObjectMapping mappingForClass:[RKMappableObject class]];
id object = [mapping mappableObjectForData:[NSDictionary dictionary]];
assertThat(object, isNot(nilValue()));
assertThat([object class], is(equalTo([RKMappableObject class])));
}
- (void)testShouldCreateNewInstancesOfManagedObjectsWhenTheMappingIsAnRKObjectMapping {
[RKTestFactory managedObjectStore];
RKObjectMapping* mapping = [RKObjectMapping mappingForClass:[RKMappableObject class]];
id object = [mapping mappableObjectForData:[NSDictionary dictionary]];
assertThat(object, isNot(nilValue()));
assertThat([object class], is(equalTo([RKMappableObject class])));
}
- (void)testShouldCreateNewManagedObjectInstancesWhenThereIsNoPrimaryKeyInTheData {
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
NSDictionary* data = [NSDictionary dictionary];
id object = [mapping mappableObjectForData:data];
assertThat(object, isNot(nilValue()));
assertThat(object, is(instanceOf([RKHuman class])));
}
- (void)testShouldCreateNewManagedObjectInstancesWhenThereIsNoPrimaryKeyAttribute {
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
NSDictionary* data = [NSDictionary dictionary];
id object = [mapping mappableObjectForData:data];
assertThat(object, isNot(nilValue()));
assertThat(object, is(instanceOf([RKHuman class])));
}
- (void)testShouldCreateANewManagedObjectWhenThePrimaryKeyValueIsNSNull {
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"id" toKeyPath:@"railsID"]];
NSDictionary* data = [NSDictionary dictionaryWithObject:[NSNull null] forKey:@"id"];
id object = [mapping mappableObjectForData:data];
assertThat(object, isNot(nilValue()));
assertThat(object, is(instanceOf([RKHuman class])));
}
- (void)testShouldMapACollectionOfObjectsWithDynamicKeys {
RKManagedObjectStore *objectStore = [RKTestFactory managedObjectStore];
RKManagedObjectMapping *mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:objectStore];
mapping.forceCollectionMapping = YES;
mapping.primaryKeyAttribute = @"name";
[mapping mapKeyOfNestedDictionaryToAttribute:@"name"];
RKObjectAttributeMapping *idMapping = [RKObjectAttributeMapping mappingFromKeyPath:@"(name).id" toKeyPath:@"railsID"];
[mapping addAttributeMapping:idMapping];
RKObjectMappingProvider *provider = [[RKObjectMappingProvider new] autorelease];
[provider setMapping:mapping forKeyPath:@"users"];
id mockCacheStrategy = [OCMockObject partialMockForObject:objectStore.cacheStrategy];
[[[mockCacheStrategy expect] andForwardToRealObject] findInstanceOfEntity:OCMOCK_ANY
withPrimaryKeyAttribute:mapping.primaryKeyAttribute
value:@"blake"
inManagedObjectContext:objectStore.primaryManagedObjectContext];
[[[mockCacheStrategy expect] andForwardToRealObject] findInstanceOfEntity:mapping.entity
withPrimaryKeyAttribute:mapping.primaryKeyAttribute
value:@"rachit"
inManagedObjectContext:objectStore.primaryManagedObjectContext];
id userInfo = [RKTestFixture parsedObjectWithContentsOfFixture:@"DynamicKeys.json"];
RKObjectMapper* mapper = [RKObjectMapper mapperWithObject:userInfo mappingProvider:provider];
[mapper performMapping];
[mockCacheStrategy verify];
}
- (void)testShouldPickTheAppropriateMappingBasedOnAnAttributeValue {
RKManagedObjectStore *objectStore = [RKTestFactory managedObjectStore];
RKDynamicObjectMapping* dynamicMapping = [RKDynamicObjectMapping dynamicMapping];
RKManagedObjectMapping* childMapping = [RKManagedObjectMapping mappingForClass:[RKChild class] inManagedObjectStore:objectStore];
childMapping.primaryKeyAttribute = @"railsID";
[childMapping mapAttributes:@"name", nil];
RKManagedObjectMapping* parentMapping = [RKManagedObjectMapping mappingForClass:[RKParent class] inManagedObjectStore:objectStore];
parentMapping.primaryKeyAttribute = @"railsID";
[parentMapping mapAttributes:@"name", @"age", nil];
[dynamicMapping setObjectMapping:parentMapping whenValueOfKeyPath:@"type" isEqualTo:@"Parent"];
[dynamicMapping setObjectMapping:childMapping whenValueOfKeyPath:@"type" isEqualTo:@"Child"];
RKObjectMapping* mapping = [dynamicMapping objectMappingForDictionary:[RKTestFixture parsedObjectWithContentsOfFixture:@"parent.json"]];
assertThat(mapping, is(notNilValue()));
assertThatBool([mapping isKindOfClass:[RKManagedObjectMapping class]], is(equalToBool(YES)));
assertThat(NSStringFromClass(mapping.objectClass), is(equalTo(@"RKParent")));
mapping = [dynamicMapping objectMappingForDictionary:[RKTestFixture parsedObjectWithContentsOfFixture:@"child.json"]];
assertThat(mapping, is(notNilValue()));
assertThatBool([mapping isKindOfClass:[RKManagedObjectMapping class]], is(equalToBool(YES)));
assertThat(NSStringFromClass(mapping.objectClass), is(equalTo(@"RKChild")));
}
- (void)testShouldIncludeTransformableAttributesInPropertyNamesAndTypes {
[RKTestFactory managedObjectStore];
NSDictionary *attributesByName = [[RKHuman entity] attributesByName];
NSDictionary *propertiesByName = [[RKHuman entity] propertiesByName];
NSDictionary *relationshipsByName = [[RKHuman entity] relationshipsByName];
assertThat([attributesByName objectForKey:@"favoriteColors"], is(notNilValue()));
assertThat([propertiesByName objectForKey:@"favoriteColors"], is(notNilValue()));
assertThat([relationshipsByName objectForKey:@"favoriteColors"], is(nilValue()));
NSDictionary *propertyNamesAndTypes = [[RKObjectPropertyInspector sharedInspector] propertyNamesAndTypesForEntity:[RKHuman entity]];
assertThat([propertyNamesAndTypes objectForKey:@"favoriteColors"], is(notNilValue()));
}
- (void)testThatAssigningAnEntityWithANonNilPrimaryKeyAttributeSetsTheDefaultValueForTheMapping {
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
NSEntityDescription *entity = [NSEntityDescription entityForName:@"RKCat" inManagedObjectContext:store.primaryManagedObjectContext];
RKManagedObjectMapping *mapping = [RKManagedObjectMapping mappingForEntity:entity inManagedObjectStore:store];
assertThat(mapping.primaryKeyAttribute, is(equalTo(@"railsID")));
}
- (void)testThatAssigningAPrimaryKeyAttributeToAMappingWhoseEntityHasANilPrimaryKeyAttributeAssignsItToTheEntity {
RKManagedObjectStore *store = [RKTestFactory managedObjectStore];
NSEntityDescription *entity = [NSEntityDescription entityForName:@"RKCloud" inManagedObjectContext:store.primaryManagedObjectContext];
RKManagedObjectMapping *mapping = [RKManagedObjectMapping mappingForEntity:entity inManagedObjectStore:store];
assertThat(mapping.primaryKeyAttribute, is(nilValue()));
mapping.primaryKeyAttribute = @"cloudID";
assertThat(entity.primaryKeyAttribute, is(equalTo(@"cloudID")));
}
#pragma mark - Fetched Results Cache
- (void)testShouldFindExistingManagedObjectsByPrimaryKeyWithFetchedResultsCache {
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
store.cacheStrategy = [RKFetchRequestManagedObjectCache new];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"id" toKeyPath:@"railsID"]];
RKHuman* human = [RKHuman object];
human.railsID = [NSNumber numberWithInt:123];
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:[NSNumber numberWithInt:123] forKey:@"id"];
id object = [mapping mappableObjectForData:data];
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
}
- (void)testShouldFindExistingManagedObjectsByPrimaryKeyPathWithFetchedResultsCache {
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
store.cacheStrategy = [RKFetchRequestManagedObjectCache new];
[RKHuman truncateAll];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"monkey.id" toKeyPath:@"railsID"]];
[RKHuman truncateAll];
RKHuman* human = [RKHuman object];
human.railsID = [NSNumber numberWithInt:123];
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:[NSNumber numberWithInt:123] forKey:@"id"];
NSDictionary* nestedDictionary = [NSDictionary dictionaryWithObject:data forKey:@"monkey"];
id object = [mapping mappableObjectForData:nestedDictionary];
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
}
#pragma mark - In Memory Cache
- (void)testShouldFindExistingManagedObjectsByPrimaryKeyWithInMemoryCache {
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
[RKHuman truncateAllInContext:store.primaryManagedObjectContext];
store.cacheStrategy = [RKInMemoryManagedObjectCache new];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"id" toKeyPath:@"railsID"]];
RKHuman* human = [RKHuman createInContext:store.primaryManagedObjectContext];
human.railsID = [NSNumber numberWithInt:123];
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:[NSNumber numberWithInt:123] forKey:@"id"];
NSManagedObject *object = [mapping mappableObjectForData:data];
assertThat([object managedObjectContext], is(equalTo(store.primaryManagedObjectContext)));
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
}
- (void)testShouldFindExistingManagedObjectsByPrimaryKeyPathWithInMemoryCache {
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
[RKHuman truncateAllInContext:store.primaryManagedObjectContext];
store.cacheStrategy = [RKInMemoryManagedObjectCache new];
[RKHuman truncateAll];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"monkey.id" toKeyPath:@"railsID"]];
[RKHuman truncateAll];
RKHuman* human = [RKHuman object];
human.railsID = [NSNumber numberWithInt:123];
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:[NSNumber numberWithInt:123] forKey:@"id"];
NSDictionary* nestedDictionary = [NSDictionary dictionaryWithObject:data forKey:@"monkey"];
id object = [mapping mappableObjectForData:nestedDictionary];
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
}
- (void)testMappingWithFetchRequestCacheWherePrimaryKeyAttributeOfMappingDisagreesWithEntity
{
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
store.cacheStrategy = [RKFetchRequestManagedObjectCache new];
[RKHuman truncateAll];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"name";
[RKHuman entity].primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"monkey.name" toKeyPath:@"name"]];
[RKHuman truncateAll];
RKHuman* human = [RKHuman object];
human.name = @"Testing";
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:@"Testing" forKey:@"name"];
NSDictionary* nestedDictionary = [NSDictionary dictionaryWithObject:data forKey:@"monkey"];
id object = [mapping mappableObjectForData:nestedDictionary];
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
id cachedObject = [store.cacheStrategy findInstanceOfEntity:[RKHuman entity] withPrimaryKeyAttribute:@"name" value:@"Testing" inManagedObjectContext:store.primaryManagedObjectContext];
assertThat(cachedObject, is(equalTo(human)));
}
- (void)testMappingWithInMemoryCacheWherePrimaryKeyAttributeOfMappingDisagreesWithEntity
{
RKManagedObjectStore* store = [RKTestFactory managedObjectStore];
store.cacheStrategy = [RKInMemoryManagedObjectCache new];
[RKHuman truncateAll];
RKManagedObjectMapping* mapping = [RKManagedObjectMapping mappingForClass:[RKHuman class] inManagedObjectStore:store];
mapping.primaryKeyAttribute = @"name";
[RKHuman entity].primaryKeyAttribute = @"railsID";
[mapping addAttributeMapping:[RKObjectAttributeMapping mappingFromKeyPath:@"monkey.name" toKeyPath:@"name"]];
[RKHuman truncateAll];
RKHuman* human = [RKHuman object];
human.name = @"Testing";
[store save:nil];
assertThatBool([RKHuman hasAtLeastOneEntity], is(equalToBool(YES)));
NSDictionary* data = [NSDictionary dictionaryWithObject:@"Testing" forKey:@"name"];
NSDictionary* nestedDictionary = [NSDictionary dictionaryWithObject:data forKey:@"monkey"];
id object = [mapping mappableObjectForData:nestedDictionary];
assertThat(object, isNot(nilValue()));
assertThat(object, is(equalTo(human)));
id cachedObject = [store.cacheStrategy findInstanceOfEntity:[RKHuman entity] withPrimaryKeyAttribute:@"name" value:@"Testing" inManagedObjectContext:store.primaryManagedObjectContext];
assertThat(cachedObject, is(equalTo(human)));
}
@end