added 2 tests for custom button with actionBlock

velga · Oct 22, 2014 · ab80aff · ab80aff
1 parent f176272
commit ab80aff
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/ObjC-Example/ActionSheetPicker Tests/AbstractActionSheetPickerTestCase.m b/ObjC-Example/ActionSheetPicker Tests/AbstractActionSheetPickerTestCase.m
@@ -74,5 +74,26 @@ - (void)testCustomFontNameWithAttributedTextAndTextAttributes
     XCTAssertNotNil(_sheetStringPicker);
 }
 
+- (void)testPickerWithCustomActionBlockOnButton
+{
+    NSString *title = @"Custom label:";
+
+    [_sheetStringPicker addCustomButtonWithTitle:title actionBlock:^{
+        NSLog(@"Test block invoked");
+    }];
+
+    [_sheetStringPicker showActionSheetPicker];
+    XCTAssertNotNil(_sheetStringPicker);
+}
+
+- (void)testPickerWithCustomActionBlockOnButtonAndNilString
+{
+    [_sheetStringPicker addCustomButtonWithTitle:nil actionBlock:^{
+        NSLog(@"Test block invoked");
+    }];
+
+    [_sheetStringPicker showActionSheetPicker];
+    XCTAssertNotNil(_sheetStringPicker);
+}
 
 @end