objective-c - 运行 iOS 应用程序测试时如何切换到内存存储？

Question

我有一个非常标准的 Xcode 为核心数据对象生成的接口(interface)，即我的应用委托(delegate)上的这些属性:

@property (readonly, strong, nonatomic) NSManagedObjectContext *managedObjectContext;
@property (readonly, strong, nonatomic) NSManagedObjectModel *managedObjectModel;
@property (readonly, strong, nonatomic) NSPersistentStoreCoordinator *persistentStoreCoordinator;

现在我正在编写应用程序测试，但我想使用内存数据库来存储每次测试运行时都会重置的核心数据。我已经找到了一种方法来做到这一点，但感觉非常古怪:

• 我在应用委托(delegate)类中有一个静态变量 storeType。
• -persistentStoreCoordinator 将它设置为 NSSQLiteStoreType 如果它是 nil。这将是默认值，并且在生产中是唯一值，以确保在运行应用程序时一切正常。
• 我确保为所有调试版本(包括我的应用测试目标)设置了 DEBUG 宏

• 如果设置了DEBUG，则在应用委托(delegate)中定义一个方法，-resetCoreData。该方法如下所示:

  #ifdef DEBUG
  - (void)resetCoreData {
      // Testing, we want to use the in memory store.
      storeType = NSInMemoryStoreType;
  
      // Disconnect core data.
      __persistentStoreCoordinator = nil;
      __managedObjectContext = nil;
  
      // Set up defaults.
      [self configureCoreDataDefaults];
  }
  #endif
  

  请注意，它将静态变量 storeType 设置为 NSInMemoryStoreType。 -configureCoreDataDefaults 方法创建了一些应该始终存在的托管对象。

• 在我的应用测试基类中，我有 -setup 调用 -resetCoreData:

  - (void)setUp {
      [super setUp];
      [[[UIApplication sharedApplication] delegate] resetCoreData];
  }
  

这给了我想要的东西:一个全新的核心数据存储，其中包含为每个测试方法创建的默认对象。

但是这很烦人。我基本上已经将测试环境的知识添加到我的应用程序委托(delegate)中，以使其在运行应用程序测试时表现不同。恶心!

那么，执行此操作的更好方法是什么？ 你是怎么做到的？

Answer 1

@eduardo-costa 的回答的后续，我已经接受了，以及我用来使其工作的代码。

首先，我创建了一个 DAO 类并将所有核心数据属性移到那里。 .h 文件如下所示:

@interface CollectionsDAO : NSObject

@property (readonly, strong, nonatomic) NSManagedObjectContext *managedObjectContext;
@property (readonly, strong, nonatomic) NSManagedObjectModel *managedObjectModel;
@property (readonly, strong, nonatomic) NSPersistentStoreCoordinator *persistentStoreCoordinator;

+ (CollectionsDAO *)defaultDAO;

@end

现在，只要我需要访问我的核心数据，我就可以使用这个类。 -defaultDAO 返回该类的一个静态实例，这样我就可以在任何地方使用一个。您看不到的是私有(private)实例方法 -storeType，它返回 NSSQLiteStoreType。这用于创建商店。我将在下面回到它。

接下来，我在这个类上创建了一个类别用于测试。头文件:

#import "CollectionsDAO.h"

@interface CollectionsDAO (Test)

+ (void)setupTestDAO;
+ (void)clearData;

@end

和实现:

#import "CollectionsDAO+Test.h"
#include <objc/runtime.h>

static CollectionsDAO *testDAO;
@implementation CollectionsDAO (Testing)

+ (CollectionsDAO *)testDAO {
    if (testDAO == nil) testDAO = [[self alloc] init];
    return testDAO;
}

+ (void)setupTestDAO {
    method_setImplementation(
        class_getClassMethod(self.class, @selector(defaultDAO)),
        method_getImplementation(class_getClassMethod(self.class, @selector(testDAO)))
    );
}

+ (void)clearData {
    testDAO = nil;
}

- (NSString *)storeType {
    return NSInMemoryStoreType;
}

@end

请注意，-storeType 替换了默认的同名私有(private)方法，返回 NSInMemoryStoreType，以便将数据存储在内存中。

现在，在我的测试基类中，我只是像这样使用这个类别:

#import "CollectionsDAO+Test.h"

@implementation AppTestBase
+ (void)initialize {
    if (self == AppTestBase.class) {
        [CollectionsDAO setupTestDAO];
    }
}
- (void)tearDown {
    [CollectionsDAO clearData];
}
@end

现在测试总是使用内存来存储核心数据数据，DAO 的测试实例总是由 +defaultDAO 返回(因为 +setupTestDAO 将其调配到位), 每次测试后清除核心数据。

我认为这比我之前使用的要干净很多。我花了一段时间才弄明白，但 Eduardo 的回答是正确的，我只需要花一些时间来弄清楚细节。