Use UIGraphicsImageRenderer instead of creating our own graphics cont…

…ext (#95)
uber · Jun 28, 2019 · d0ac08b · d0ac08b
1 parent 3571487
commit d0ac08b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/FBSnapshotTestCase/Categories/UIImage+Snapshot.m b/FBSnapshotTestCase/Categories/UIImage+Snapshot.m
@@ -56,11 +56,11 @@ + (UIImage *)fb_imageForView:(UIView *)view
     NSAssert1(CGRectGetWidth(bounds), @"Zero width for view %@", view);
     NSAssert1(CGRectGetHeight(bounds), @"Zero height for view %@", view);
 
-    UIGraphicsBeginImageContextWithOptions(bounds.size, NO, 0);
-    [view drawViewHierarchyInRect:view.bounds afterScreenUpdates:YES];
+    UIGraphicsImageRenderer *graphicsImageRenderer = [[UIGraphicsImageRenderer alloc] initWithSize:bounds.size];
 
-    UIImage *snapshot = UIGraphicsGetImageFromCurrentImageContext();
-    UIGraphicsEndImageContext();
+    UIImage *snapshot = [graphicsImageRenderer imageWithActions:^(UIGraphicsImageRendererContext * _Nonnull rendererContext) {
+        [view drawViewHierarchyInRect:bounds afterScreenUpdates:YES];
+    }];
 
     if (removeFromSuperview) {
         [view removeFromSuperview];