add tests

skywinder · Nov 11, 2014 · 032b72b · 032b72b
1 parent 0d5923f
commit 032b72b
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 0 deletions.
diff --git a/ObjC-Example/ActionSheetPicker Tests/ActionSheetDatePickerTestCase.m b/ObjC-Example/ActionSheetPicker Tests/ActionSheetDatePickerTestCase.m
@@ -9,6 +9,8 @@
 #import <XCTest/XCTest.h>
 #import <CoreActionSheetPicker/CoreActionSheetPicker.h>
 #import "AbstractActionSheetPicker+CustomButton.h"
+
+static const int countdownTestInt = 360;
 UIView *origin;
 
 @interface ActionSheetDatePickerTestCase : XCTestCase
@@ -134,6 +136,35 @@ - (void)testPickerWithNilCustomActionSelectorOnButtonNilTargetAndNilString
     XCTAssertNotNil(_actionSheetDatePicker);
 }
 
+- (void)testPickerCountDownTimerModeValueWithSelector
+{
+    _actionSheetDatePicker = [[ActionSheetDatePicker alloc] initWithTitle:@"Test title" datePickerMode:UIDatePickerModeCountDownTimer selectedDate:nil target:self action:@selector(countDownTest:) origin:origin cancelAction:nil];
+    _actionSheetDatePicker.countDownDuration = countdownTestInt;
+    [_actionSheetDatePicker showActionSheetPicker];
+    [_actionSheetDatePicker pressDoneButton];
+
+    XCTAssertNotNil(_actionSheetDatePicker);
+}
+
+- (void)testPickerCountDownTimerModeValueWithBlock
+{
+    _actionSheetDatePicker = [[ActionSheetDatePicker alloc] initWithTitle:@"Test" datePickerMode:UIDatePickerModeCountDownTimer selectedDate:nil doneBlock:^(ActionSheetDatePicker *picker, id selectedDate, id origin) {
+        XCTAssertEqualObjects(selectedDate, @(countdownTestInt));
+    } cancelBlock:nil origin:origin];
+
+    _actionSheetDatePicker.countDownDuration = countdownTestInt;
+    [_actionSheetDatePicker showActionSheetPicker];
+    UIDatePicker *picker = (UIDatePicker *)_actionSheetDatePicker.pickerView;
+    [_actionSheetDatePicker pressDoneButton];
+
+    XCTAssertNotNil(_actionSheetDatePicker);
+}
+
+- (void)countDownTest:(NSNumber *)number
+{
+    XCTAssertEqualObjects(number, @(countdownTestInt));
+}
+
 - (void)exampleSelector
 {
     NSLog(@"Test selector invoked");

diff --git a/ObjC-Example/Example/Classes/AbstractActionSheetPicker+CustomButton.h b/ObjC-Example/Example/Classes/AbstractActionSheetPicker+CustomButton.h
@@ -9,4 +9,5 @@
 
 - (void)pressFirstCustomButton;
 
+- (void)pressDoneButton;
 @end
diff --git a/ObjC-Example/Example/Classes/AbstractActionSheetPicker+CustomButton.m b/ObjC-Example/Example/Classes/AbstractActionSheetPicker+CustomButton.m
@@ -18,4 +18,14 @@ - (void)pressFirstCustomButton
     );
 }
 
+- (void)pressDoneButton
+{
+    //6 items in _actionSheetDatePicker.toolbar.items : [ cancel - custom - separator - title - separator - done ]
+    //So, check custom button:
+    UIBarButtonItem *customBarButton = self.toolbar.items[4];
+    SuppressPerformSelectorLeakWarning (
+            [customBarButton.target performSelector:customBarButton.action withObject:customBarButton];
+    );
+}
+
 @end
Original file line number	Diff line number	Diff line change
Expand Up		@@ -9,4 +9,5 @@

		- (void)pressFirstCustomButton;

		- (void)pressDoneButton;
		@end