Merge pull request #1242 from kif-framework/derko/fix-screenshot-xc13

Remove use of XCUI for taking screenshots

KIF.podspec

@@ -6,7 +6,7 @@ Pod::Spec.new do |s|
   s.license                 = 'Apache 2.0'
   s.authors                 = 'Michael Thole', 'Eric Firestone', 'Jim Puls', 'Brian Nickel'
   s.source                  = { :git => "https://github.com/kif-framework/KIF.git", :tag => "v#{ s.version.to_s }" }
-  s.platform                = :ios, '9.0'
+  s.platform                = :ios, '10.0'
   s.frameworks              = 'CoreGraphics', 'QuartzCore', 'IOKit', 'WebKit', 'XCTest'
   s.default_subspec         = 'Core'
   s.requires_arc            = true

Sources/KIF/Additions/XCTestCase-KIFAdditions.m

@@ -86,8 +86,8 @@ - (void)writeScreenshotForException:(NSException *)exception;
         dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
 
         [XCTContext runActivityNamed:(@"screenshot") block:^(id<XCTActivity>  _Nonnull activity) {
-            XCUIScreenshot *screenShot = [[XCUIScreen mainScreen] screenshot];
-            XCTAttachment *attachment = [XCTAttachment attachmentWithScreenshot:screenShot];
+            XCTAttachment *attachment = [XCTAttachment attachmentWithImage:[self _snapshotScreen]];
+            attachment.name = @"Screenshot";
             [activity addAttachment:(attachment)];
             dispatch_semaphore_signal(semaphore);
         }];
@@ -96,6 +96,22 @@ - (void)writeScreenshotForException:(NSException *)exception;
     }   
 }
 
+- (UIImage *)_snapshotScreen
+{
+    UIView *view = [[UIScreen mainScreen] snapshotViewAfterScreenUpdates:YES];
+    [view layoutIfNeeded];
+
+    CGRect bounds = view.bounds;
+    NSAssert1(CGRectGetWidth(bounds), @"Zero width for view %@", view);
+    NSAssert1(CGRectGetHeight(bounds), @"Zero height for view %@", view);
+
+    UIGraphicsImageRenderer *graphicsImageRenderer = [[UIGraphicsImageRenderer alloc] initWithSize:bounds.size];
+
+    return [graphicsImageRenderer imageWithActions:^(UIGraphicsImageRendererContext * _Nonnull rendererContext) {
+        [view drawViewHierarchyInRect:bounds afterScreenUpdates:YES];
+    }];
+}
+
 - (void)printViewHierarchyIfOptedIn;
 {
     static BOOL shouldPrint;