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

Question

我有一个非常标准的 Xcode 生成的 Core Data 对象接口(interface)，即我的应用程序委托(delegate)上的这些属性:

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

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

• 我在应用委托(delegate)类中有一个静态变量 storeType。
• -persistentStoreCoordinator 如果它是 nil，则将其设置为 NSSQLiteStoreType。这将是默认值，并且是生产中的唯一值，以确保运行应用程序时一切正常。
• 我确保为所有调试版本(包括我的应用测试目标)设置了 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 的答案是正确的，我只需要花一段时间就能弄清楚细节。