Skip cells that can’t be found

kif-framework · Nov 27, 2024 · d8b6ef6 · d8b6ef6
1 parent be806f7
commit d8b6ef6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/KIF/Additions/UIView-KIFAdditions.m b/Sources/KIF/Additions/UIView-KIFAdditions.m
@@ -385,7 +385,10 @@ - (UIAccessibilityElement *)accessibilityElementMatchingBlock:(BOOL(^)(UIAccessi
                             [collectionView layoutIfNeeded];
                             cell = [collectionView cellForItemAtIndexPath:indexPath];
                         }
-                        NSAssert(cell, @"UICollectionViewCell returned from 'cellForItemAtIndexPath' is unexpectedly nil!");
+                        // Skip this cell if it can't be found
+                        if (!cell) {
+                            continue;
+                        }
                         UIAccessibilityElement *element = [cell accessibilityElementMatchingBlock:matchBlock notHidden:NO disableScroll:NO];
 
                         // Skip this cell if it isn't the one we're looking for