Update CoreDataMigrationTests

Author: kean

Sources/WordPressData/Swift/Bundle+WordPressData.swift

@@ -1,7 +1,7 @@
 import Foundation
 
 extension Bundle {
-    public static var wordPressData: Bundle {
+    @objc public class var wordPressData: Bundle {
         Bundle(for: BundleToken.self)
     }
 }

Tests/KeystoneTests/Tests/CoreData/CoreDataMigrationTests.m

@@ -217,13 +217,13 @@ - (void)testMigrationFrom103To104CustomTransformers
 - (NSURL *)urlForModelName:(NSString*)modelName
                inDirectory:(NSString*)directory
 {
-    NSBundle *bundle = [NSBundle mainBundle];
+    NSBundle *bundle = [NSBundle wordPressData];
     NSURL *url = [bundle URLForResource:modelName
                           withExtension:@"mom"
                            subdirectory:directory];
     if (nil == url) {
         // Get mom file paths from momd directories.
-        NSArray *momdPaths = [[NSBundle mainBundle] pathsForResourcesOfType:@"momd"
+        NSArray *momdPaths = [[NSBundle wordPressData] pathsForResourcesOfType:@"momd"
                                                                 inDirectory:directory];
         for (NSString *momdPath in momdPaths) {
             url = [bundle URLForResource:modelName