Added support for easy setup with OCTest

pettair · Dec 29, 2011 · 2209c3e · 2209c3e
1 parent 6b4ef06
commit 2209c3e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/Source/MagicalRecordHelpers.h b/Source/MagicalRecordHelpers.h
@@ -29,6 +29,7 @@ typedef void (^CoreDataBlock)(NSManagedObjectContext *context);
 + (SEL) errorHandlerAction;
 + (id) errorHandlerTarget;
 
++ (void) setDefaultModelForTestCase:(Class)class;
 + (void) setDefaultModelNamed:(NSString *)modelName;
 + (NSString *) defaultStoreName;
 

diff --git a/Source/MagicalRecordHelpers.m b/Source/MagicalRecordHelpers.m
@@ -75,6 +75,7 @@ + (void) defaultErrorHandler:(NSError *)error
             MRLog(@"Error: %@", detailedError);
         }
     }
+    MRLog(@"Error Message: %@", [error localizedDescription]);
     MRLog(@"Error Domain: %@", [error domain]);
     MRLog(@"Recovery Suggestion: %@", [error localizedRecoverySuggestion]);
 }
@@ -126,6 +127,13 @@ + (void) setDefaultModelNamed:(NSString *)modelName;
     [NSManagedObjectModel MR_setDefaultManagedObjectModel:model];
 }
 
++ (void) setDefaultModelForTestCase:(Class)class;
+{
+    NSBundle *bundle = [NSBundle bundleForClass:class];
+    NSManagedObjectModel *model = [NSManagedObjectModel mergedModelFromBundles:[NSArray arrayWithObject:bundle]];
+    [NSManagedObjectModel MR_setDefaultManagedObjectModel:model];
+}
+
 + (NSString *) defaultStoreName;
 {
     NSString *defaultName = [[[NSBundle mainBundle] infoDictionary] valueForKey:(id)kCFBundleNameKey];