- update

Example/Pods/FBSnapshotTestCase/FBSnapshotTestCase.modulemap

@@ -1,15 +0,0 @@
-framework module FBSnapshotTestCase {
-  umbrella header "FBSnapshotTestCase.h"
-
-  export *
-  module * { export * }
-
-  header "FBSnapshotTestCase.h"
-  header "FBSnapshotTestCasePlatform.h"
-  header "FBSnapshotTestController.h"
-
-  private header "UIImage+Compare.h"
-  private header "UIImage+Diff.h"
-  private header "UIImage+Snapshot.h"
-}
-

Example/Pods/FBSnapshotTestCase/FBSnapshotTestCase/FBSnapshotTestCase.h

@@ -62,43 +62,9 @@
 
 #define FBSnapshotVerifyViewOrLayerWithOptions(what__, viewOrLayer__, identifier__, suffixes__, tolerance__) \
 { \
-  NSString *referenceImageDirectory = [self getReferenceImageDirectoryWithDefault:(@ FB_REFERENCE_IMAGE_DIR)]; \
-  XCTAssertNotNil(referenceImageDirectory, @"Missing value for referenceImagesDirectory - Set FB_REFERENCE_IMAGE_DIR as Environment variable in your scheme.");\
-  XCTAssertTrue((suffixes__.count > 0), @"Suffixes set cannot be empty %@", suffixes__); \
-  \
-  BOOL testSuccess__ = NO; \
-  NSError *error__ = nil; \
-  NSMutableArray *errors__ = [NSMutableArray array]; \
-  \
-  if (self.recordMode) { \
-    \
-    NSString *referenceImagesDirectory__ = [NSString stringWithFormat:@"%@%@", referenceImageDirectory, suffixes__.firstObject]; \
-    BOOL referenceImageSaved__ = [self compareSnapshotOf ## what__ :(viewOrLayer__) referenceImagesDirectory:referenceImagesDirectory__ identifier:(identifier__) tolerance:(tolerance__) error:&error__]; \
-    if (!referenceImageSaved__) { \
-      [errors__ addObject:error__]; \
-    } \
-  } else { \
-    \
-    for (NSString *suffix__ in suffixes__) { \
-      NSString *referenceImagesDirectory__ = [NSString stringWithFormat:@"%@%@", referenceImageDirectory, suffix__]; \
-      BOOL referenceImageAvailable = [self referenceImageRecordedInDirectory:referenceImagesDirectory__ identifier:(identifier__) error:&error__]; \
-      \
-      if (referenceImageAvailable) { \
-        BOOL comparisonSuccess__ = [self compareSnapshotOf ## what__ :(viewOrLayer__) referenceImagesDirectory:referenceImagesDirectory__ identifier:(identifier__) tolerance:(tolerance__) error:&error__]; \
-        [errors__ removeAllObjects]; \
-        if (comparisonSuccess__) { \
-          testSuccess__ = YES; \
-          break; \
-        } else { \
-          [errors__ addObject:error__]; \
-        } \
-      } else { \
-        [errors__ addObject:error__]; \
-      } \
-    } \
-  } \
-  XCTAssertTrue(testSuccess__, @"Snapshot comparison failed: %@", errors__.firstObject); \
-  XCTAssertFalse(self.recordMode, @"Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!"); \
+  NSString *errorDescription = [self snapshotVerifyViewOrLayer:viewOrLayer__ identifier:identifier__ suffixes:suffixes__ tolerance:tolerance__]; \
+  BOOL noErrors = (errorDescription == nil); \
+  XCTAssertTrue(noErrors, @"%@", errorDescription); \
 }
 
 
@@ -148,6 +114,19 @@
 - (void)setUp NS_REQUIRES_SUPER;
 - (void)tearDown NS_REQUIRES_SUPER;
 
+/**
+ Performs the comparison or records a snapshot of the layer if recordMode is YES.
+ @param viewOrLayer The UIView or CALayer to snapshot
+ @param identifier An optional identifier, used if there are multiple snapshot tests in a given -test method.
+ @param suffixes An NSOrderedSet of strings for the different suffixes
+ @param tolerance The percentage difference to still count as identical - 0 mean pixel perfect, 1 means I don't care
+ @returns nil if the comparison (or saving of the reference image) succeeded. Otherwise it contains an error description.
+ */
+- (NSString *)snapshotVerifyViewOrLayer:(id)viewOrLayer
+                             identifier:(NSString *)identifier
+                               suffixes:(NSOrderedSet *)suffixes
+                              tolerance:(CGFloat)tolerance;
+
 /**
  Performs the comparison or records a snapshot of the layer if recordMode is YES.
  @param layer The Layer to snapshot

Example/Pods/FBSnapshotTestCase/FBSnapshotTestCase/FBSnapshotTestCase.m

@@ -65,6 +65,62 @@
 
 #pragma mark - Public API
 
+- (NSString *)snapshotVerifyViewOrLayer:(id)viewOrLayer
+                             identifier:(NSString *)identifier
+                               suffixes:(NSOrderedSet *)suffixes
+                              tolerance:(CGFloat)tolerance
+{
+  if (nil == viewOrLayer) {
+    return @"Object to be snapshotted must not be nil";
+  }
+  NSString *referenceImageDirectory = [self getReferenceImageDirectoryWithDefault:(@ FB_REFERENCE_IMAGE_DIR)];
+  if (referenceImageDirectory == nil) {
+    return @"Missing value for referenceImagesDirectory - Set FB_REFERENCE_IMAGE_DIR as Environment variable in your scheme.";
+  }
+  if (suffixes.count == 0) {
+    return [NSString stringWithFormat:@"Suffixes set cannot be empty %@", suffixes];
+  }
+  
+  BOOL testSuccess = NO;
+  NSError *error = nil;
+  NSMutableArray *errors = [NSMutableArray array];
+  
+  if (self.recordMode) {
+    NSString *referenceImagesDirectory = [NSString stringWithFormat:@"%@%@", referenceImageDirectory, suffixes.firstObject];
+    BOOL referenceImageSaved = [self _compareSnapshotOfViewOrLayer:viewOrLayer referenceImagesDirectory:referenceImagesDirectory identifier:(identifier) tolerance:tolerance error:&error];
+    if (!referenceImageSaved) {
+      [errors addObject:error];
+    }
+  } else {
+    for (NSString *suffix in suffixes) {
+      NSString *referenceImagesDirectory = [NSString stringWithFormat:@"%@%@", referenceImageDirectory, suffix];
+      BOOL referenceImageAvailable = [self referenceImageRecordedInDirectory:referenceImagesDirectory identifier:(identifier) error:&error];
+     
+      if (referenceImageAvailable) {
+        BOOL comparisonSuccess = [self _compareSnapshotOfViewOrLayer:viewOrLayer referenceImagesDirectory:referenceImagesDirectory identifier:identifier tolerance:tolerance error:&error];
+        [errors removeAllObjects];
+        if (comparisonSuccess) {
+          testSuccess = YES;
+          break;
+        } else {
+          [errors addObject:error];
+        }
+      } else {
+        [errors addObject:error];
+      }
+    }
+  }
+  
+  if (!testSuccess) {
+    return [NSString stringWithFormat:@"Snapshot comparison failed: %@", errors.firstObject];
+  }
+  if (self.recordMode) {
+    return @"Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!";
+  }
+
+  return nil;
+}
+
 - (BOOL)compareSnapshotOfLayer:(CALayer *)layer
       referenceImagesDirectory:(NSString *)referenceImagesDirectory
                     identifier:(NSString *)identifier

Example/Pods/FBSnapshotTestCase/FBSnapshotTestCase/SwiftSupport.swift

@@ -8,6 +8,64 @@
 *
 */
 
+#if swift(>=3)
+  public extension FBSnapshotTestCase {
+    public func FBSnapshotVerifyView(_ view: UIView, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
+      FBSnapshotVerifyViewOrLayer(view, identifier: identifier, suffixes: suffixes, tolerance: tolerance, file: file, line: line)
+    }
+
+    public func FBSnapshotVerifyLayer(_ layer: CALayer, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
+      FBSnapshotVerifyViewOrLayer(layer, identifier: identifier, suffixes: suffixes, tolerance: tolerance, file: file, line: line)
+    }
+
+    private func FBSnapshotVerifyViewOrLayer(_ viewOrLayer: AnyObject, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
+      let envReferenceImageDirectory = self.getReferenceImageDirectory(withDefault: FB_REFERENCE_IMAGE_DIR)
+      var error: NSError?
+      var comparisonSuccess = false
+
+      if let envReferenceImageDirectory = envReferenceImageDirectory {
+        for suffix in suffixes {
+          let referenceImagesDirectory = "\(envReferenceImageDirectory)\(suffix)"
+          if viewOrLayer.isKind(of: UIView.self) {
+            do {
+              try compareSnapshot(of: viewOrLayer as! UIView, referenceImagesDirectory: referenceImagesDirectory, identifier: identifier, tolerance: tolerance)
+              comparisonSuccess = true
+            } catch let error1 as NSError {
+              error = error1
+              comparisonSuccess = false
+            }
+          } else if viewOrLayer.isKind(of: CALayer.self) {
+            do {
+              try compareSnapshot(of: viewOrLayer as! CALayer, referenceImagesDirectory: referenceImagesDirectory, identifier: identifier, tolerance: tolerance)
+              comparisonSuccess = true
+            } catch let error1 as NSError {
+              error = error1
+              comparisonSuccess = false
+            }
+          } else {
+            assertionFailure("Only UIView and CALayer classes can be snapshotted")
+          }
+
+          assert(recordMode == false, message: "Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!", file: file, line: line)
+
+          if comparisonSuccess || recordMode {
+            break
+          }
+
+          assert(comparisonSuccess, message: "Snapshot comparison failed: \(error)", file: file, line: line)
+        }
+      } else {
+        XCTFail("Missing value for referenceImagesDirectory - Set FB_REFERENCE_IMAGE_DIR as Environment variable in your scheme.")
+      }
+    }
+
+    func assert(_ assertion: Bool, message: String, file: StaticString, line: UInt) {
+      if !assertion {
+        XCTFail(message, file: file, line: line)
+      }
+    }
+  }
+#else
 public extension FBSnapshotTestCase {
   public func FBSnapshotVerifyView(view: UIView, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
     FBSnapshotVerifyViewOrLayer(view, identifier: identifier, suffixes: suffixes, tolerance: tolerance, file: file, line: line)
@@ -64,3 +122,4 @@ public extension FBSnapshotTestCase {
     }
   }
 }
+#endif