more reliable logic

EmergeTools · Aug 26, 2024 · 99f5c25 · 99f5c25
1 parent b646890
commit 99f5c25
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/Sources/SnapshotPreviewsCore/ExpandingViewController.swift b/Sources/SnapshotPreviewsCore/ExpandingViewController.swift
@@ -78,11 +78,7 @@ public final class ExpandingViewController: UIHostingController<EmergeModifierVi
 
   public override func viewDidLayoutSubviews() {
     super.viewDidLayoutSubviews()
-
-    // Kick off in next run loop cycle to let the layout pass complete first
-    DispatchQueue.main.async {
-      self.updateScrollViewHeight()
-    }
+    updateScrollViewHeight()
   }
 
   public func updateScrollViewHeight() {
@@ -91,6 +87,11 @@ public final class ExpandingViewController: UIHostingController<EmergeModifierVi
       return
     }
 
+    // Verify scrollview has no pending layout updates
+    if let firstScrollView = firstScrollView as? UIScrollView {
+        firstScrollView.layoutIfNeeded()
+    }
+
     updateHeight {
       runCallback()
     }