Compare and record new at the same time

FBSnapshotTestCase/FBSnapshotTestCase.h

@@ -123,6 +123,11 @@
  */
 @interface FBSnapshotTestCase : XCTestCase
 
+/**
+ Test mode. Can be record, compare or both combined.
+ */
+@property (readwrite, nonatomic, assign) FBSnapshotTestMode mode;
+
 /**
  When YES, the test macros will save reference images, rather than performing an actual test.
  */

FBSnapshotTestCase/FBSnapshotTestCase.m

@@ -32,13 +32,21 @@ - (void)tearDown
 
 - (BOOL)recordMode
 {
-  return _snapshotController.recordMode;
+  return _snapshotController.mode & FBSnapshotModeRecord;
 }
 
 - (void)setRecordMode:(BOOL)recordMode
 {
+  _snapshotController.mode = recordMode ? FBSnapshotModeRecord : FBSnapshotModeCompare;
+}
+
+- (FBSnapshotTestMode)mode{
+  return _snapshotController.mode;
+}
+
+- (void)setMode:(FBSnapshotTestMode)mode{
   NSAssert1(_snapshotController, @"%s cannot be called before [super setUp]", __FUNCTION__);
-  _snapshotController.recordMode = recordMode;
+  _snapshotController.mode = mode;
 }
 
 - (BOOL)isDeviceAgnostic

FBSnapshotTestCase/FBSnapshotTestController.h

@@ -18,6 +18,13 @@ typedef NS_ENUM(NSInteger, FBSnapshotTestControllerErrorCode) {
   FBSnapshotTestControllerErrorCodeImagesDifferentSizes,
   FBSnapshotTestControllerErrorCodeImagesDifferent,
 };
+
+typedef NS_OPTIONS(NSUInteger, FBSnapshotTestMode){
+  FBSnapshotModeNone = 0,
+  FBSnapshotModeCompare = 1 << 0,
+  FBSnapshotModeRecord = 1 << 1
+};
+
 /**
  Errors returned by the methods of FBSnapshotTestController use this domain.
  */
@@ -36,9 +43,9 @@ extern NSString *const FBReferenceImageFilePathKey;
 @interface FBSnapshotTestController : NSObject
 
 /**
- Record snapshots.
+ Record, Compare or both
  */
-@property (readwrite, nonatomic, assign) BOOL recordMode;
+@property (readwrite, nonatomic, assign) FBSnapshotTestMode mode;
 
 /**
  When @c YES appends the name of the device model and OS to the snapshot file name.

FBSnapshotTestCase/FBSnapshotTestController.m

@@ -44,6 +44,7 @@ - (instancetype)initWithTestName:(NSString *)testName
   if (self = [super init]) {
     _testName = [testName copy];
     _deviceAgnostic = NO;
+    _mode = FBSnapshotModeCompare;
 
     _fileManager = [[NSFileManager alloc] init];
   }
@@ -89,11 +90,14 @@ - (BOOL)compareSnapshotOfViewOrLayer:(id)viewOrLayer
                            tolerance:(CGFloat)tolerance
                                error:(NSError **)errorPtr
 {
-  if (self.recordMode) {
-    return [self _recordSnapshotOfViewOrLayer:viewOrLayer selector:selector identifier:identifier error:errorPtr];
-  } else {
-    return [self _performPixelComparisonWithViewOrLayer:viewOrLayer selector:selector identifier:identifier tolerance:tolerance error:errorPtr];
+  BOOL result = NO;
+  if (self.mode & FBSnapshotModeCompare) {
+    result = [self _performPixelComparisonWithViewOrLayer:viewOrLayer selector:selector identifier:identifier tolerance:tolerance error:errorPtr];
+  }
+  if (self.mode & FBSnapshotModeRecord) {
+    result = [self _recordSnapshotOfViewOrLayer:viewOrLayer selector:selector identifier:identifier error:errorPtr];
   }
+  return result;
 }
 
 - (UIImage *)referenceImageForSelector:(SEL)selector