Added check for nil handler before calling

foulkesjohn · Apr 15, 2016 · 30a4613 · 30a4613
1 parent 07ca463
commit 30a4613
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Source/MockUIAlertController/QCOMockAlertVerifier.m b/Source/MockUIAlertController/QCOMockAlertVerifier.m
@@ -70,7 +70,10 @@ - (UIAlertActionStyle)styleForButtonWithTitle:(NSString *)title
 - (void)executeActionForButtonWithTitle:(NSString *)title
 {
     UIAlertAction *action = [self actionWithTitle:title];
-    [action qcoMock_handler](action);
+    void (^handler)(UIAlertAction *action) = [action qcoMock_handler];
+    if (handler) {
+        handler(action);
+    }
 }
 
 - (UIAlertAction *)actionWithTitle:(NSString *)title