Exposed setDeviceAgnostic globally

Author: John Foulkes

EXPMatchers+FBSnapshotTest.m

@@ -25,7 +25,8 @@ + (BOOL)compareSnapshotOfViewOrLayer:(id)viewOrLayer snapshot:(NSString *)snapsh
     snapshotController.recordMode = record;
     snapshotController.referenceImagesDirectory = referenceDirectory;
     snapshotController.usesDrawViewHierarchyInRect = [Expecta usesDrawViewHierarchyInRect];
-
+    snapshotController.deviceAgnostic = [Expecta isDeviceAgnostic];
+  
     if (! snapshotController.referenceImagesDirectory) {
         [NSException raise:@"Missing value for referenceImagesDirectory" format:@"Call [[EXPExpectFBSnapshotTest instance] setReferenceImagesDirectory"];
     }
@@ -91,6 +92,7 @@ - (NSString *)_getDefaultReferenceDirectory
     [NSException raise:@"Could not infer reference image folder" format:@"You should provide a reference dir using setGlobalReferenceImageDir(FB_REFERENCE_IMAGE_DIR);"];
     return nil;
 }
+
 @end
 
 
@@ -160,7 +162,6 @@ - (NSString *)_getDefaultReferenceDirectory
 
     match(^BOOL{
         NSString *referenceImageDir = [self _getDefaultReferenceDirectory];
-
         // For view controllers do the viewWill/viewDid dance, then pass view through
         if ([actual isKindOfClass:UIViewController.class]) {
 
@@ -244,7 +245,6 @@ - (NSString *)_getDefaultReferenceDirectory
 
     match(^BOOL{
         NSString *referenceImageDir = [self _getDefaultReferenceDirectory];
-
         // For view controllers do the viewWill/viewDid dance, then pass view through
         if ([actual isKindOfClass:UIViewController.class]) {
             [actual beginAppearanceTransition:YES animated:NO];

ExpectaObject+FBSnapshotTest.h

@@ -14,4 +14,8 @@
 
 + (BOOL)usesDrawViewHierarchyInRect;
 
++ (void)setDeviceAgnostic:(BOOL)deviceAgnostic;
+
++ (BOOL)isDeviceAgnostic;
+
 @end

ExpectaObject+FBSnapshotTest.m

@@ -22,4 +22,15 @@ + (BOOL)usesDrawViewHierarchyInRect {
     return usesDrawViewHierarchyInRect.boolValue;
 }
 
++ (void)setDeviceAgnostic:(BOOL)deviceAgnostic
+{
+    objc_setAssociatedObject(self, @selector(isDeviceAgnostic), @(deviceAgnostic), OBJC_ASSOCIATION_ASSIGN);
+}
+
++ (BOOL)isDeviceAgnostic
+{
+  NSNumber *isDeviceAgnostic = objc_getAssociatedObject(self, @selector(isDeviceAgnostic));
+  return isDeviceAgnostic.boolValue;
+}
+
 @end

FBSnapshotTestCaseDemo/FBSnapshotTestCaseDemoTests/FBSnapshotTestCaseDemoSpecs.m

@@ -10,6 +10,7 @@
 #include <Specta/Specta.h>
 #include <Expecta/Expecta.h>
 #include <Expecta+Snapshots/EXPMatchers+FBSnapshotTest.h>
+#import <FBSnapshotTestCase/FBSnapshotTestCasePlatform.h>
 
 #include "FBExampleView.h"
 #import "FBViewController.h"
@@ -191,6 +192,72 @@
             }).to.raise(@"Nil was passed into haveValidSnapshot. snapshots_with_a_view_controller_Raises_when_a_nil_is_passed_in");
         });
     });
+  
+  describe(@"device agnostic", ^{
+    
+    __block FBViewController *controller;
+    
+    before(^{
+      controller = [[FBRedViewController alloc] init];
+      controller.view.frame = frame;
+      [Expecta setDeviceAgnostic:TRUE];
+    });
+    
+    after(^{
+      [Expecta setDeviceAgnostic:FALSE];
+    });
+    
+    describe(@"recording", ^{
+      
+      it(@"named", ^{
+        expect(controller).toNot.recordSnapshotNamed(@"view controller 1");
+        expect(controller.viewWillAppearCalled).to.beTruthy();
+        expect(controller.viewDidAppearCalled).to.beTruthy();
+        NSString *imageName = [[fileManager contentsOfDirectoryAtPath:imagesDirectory error:nil] firstObject];
+        NSString *expectedImageName = FBDeviceAgnosticNormalizedFileName(@"view controller 1");
+        expect(imageName).to.contain(expectedImageName);
+      });
+      
+      it(@"unnamed", ^{
+        expect(controller).toNot.recordSnapshot();
+        expect(controller.viewWillAppearCalled).to.beTruthy();
+        expect(controller.viewDidAppearCalled).to.beTruthy();
+        NSString *imageName = [[fileManager contentsOfDirectoryAtPath:imagesDirectory error:nil] firstObject];
+        NSString *expectedImageName = FBDeviceAgnosticNormalizedFileName(@"snapshots_device_agnostic_recording_unnamed");
+        expect(imageName).to.contain(expectedImageName);
+      });
+      
+    });
+    
+    describe(@"unnamed", ^{
+      
+      describe(@"named", ^{
+        
+        it(@"matches view controller", ^{
+          expect(controller).toNot.recordSnapshotNamed(@"view controller 2");
+          
+          FBViewController *newVC = [[FBRedViewController alloc] init];
+          newVC.view.frame = frame;
+          expect(newVC).to.haveValidSnapshotNamed(@"view controller 2");
+          expect(newVC.viewWillAppearCalled).to.beTruthy();
+          expect(newVC.viewDidAppearCalled).to.beTruthy();
+        });
+        
+      });
+      
+      it(@"matches view controller", ^{
+        expect(controller).toNot.recordSnapshot();
+        
+        FBViewController *newVC = [[FBRedViewController alloc] init];
+        newVC.view.frame = frame;
+        expect(newVC).to.haveValidSnapshot();
+        expect(newVC.viewWillAppearCalled).to.beTruthy();
+        expect(newVC.viewDidAppearCalled).to.beTruthy();
+      });
+      
+    });
+    
+  });
 });
 
 SpecEnd