Skip text field subviews

EmergeTools · Aug 27, 2024 · 594396f · 594396f
1 parent 53775ed
commit 594396f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/Sources/SnapshotPreviewsCore/ScrollExpansion.swift b/Sources/SnapshotPreviewsCore/ScrollExpansion.swift
@@ -123,11 +123,15 @@ extension NSView: FirstScrollViewProviding {
     var subviews = subviews
     while !subviews.isEmpty {
       let subview = subviews.removeFirst()
-      subviews.append(contentsOf: subview.subviews)
-      // Don’t expand NSTextView, it can cause flakes
-      if let scrollView = subview as? NSScrollView, !(scrollView.documentView is NSTextView) {
+      if let scrollView = subview as? NSScrollView {
+        // Don’t expand NSTextView, it can cause flakes
+        guard !(scrollView.documentView is NSTextView) else {
+          continue
+        }
+
         return scrollView
       }
+      subviews.append(contentsOf: subview.subviews)
     }
     return nil
   }