Use fix from other PR pointfreeco#692 and inject `drawHierarchyInKeyW…

…indow` on UIViewController
nlamah · Aug 17, 2023 · 9b16c04 · 9b16c04
1 parent dc46eeb
commit 9b16c04
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/Sources/SnapshotTesting/Common/View.swift b/Sources/SnapshotTesting/Common/View.swift
@@ -853,7 +853,13 @@ extension View {
                 return
               }
               wkWebView.takeSnapshot(with: nil) { image, _ in
-                callback(image!)
+                  let configuaration = WKSnapshotConfiguration()
+                  if #available(iOS 13, macOS 10.15, *) {
+                      configuaration.afterScreenUpdates = false
+                  }
+                  wkWebView.takeSnapshot(with: configuaration) { image, _ in
+                      callback(image!)
+                  }
               }
             }
           } else {

diff --git a/Sources/SnapshotTesting/Snapshotting/UIViewController.swift b/Sources/SnapshotTesting/Snapshotting/UIViewController.swift
@@ -17,6 +17,7 @@ extension Snapshotting where Value == UIViewController, Format == UIImage {
   ///   - traits: A trait collection override.
   public static func image(
     on config: ViewImageConfig,
+    drawHierarchyInKeyWindow: Bool = false,
     precision: Float = 1,
     perceptualPrecision: Float = 1,
     size: CGSize? = nil,
@@ -27,7 +28,7 @@ extension Snapshotting where Value == UIViewController, Format == UIImage {
       return SimplySnapshotting.image(precision: precision, perceptualPrecision: perceptualPrecision, scale: traits.displayScale).asyncPullback { viewController in
         snapshotView(
           config: size.map { .init(safeArea: config.safeArea, size: $0, traits: config.traits) } ?? config,
-          drawHierarchyInKeyWindow: false,
+          drawHierarchyInKeyWindow: drawHierarchyInKeyWindow,
           traits: traits,
           view: viewController.view,
           viewController: viewController