Don’t crash if sample is called on @[]

Fixes #12 Tag 11.0.1
mxcl · Nov 23, 2014 · bbb5eb2 · bbb5eb2
1 parent 2b45ecd
commit bbb5eb2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/NSArray+sample.m b/NSArray+sample.m
@@ -3,8 +3,10 @@
 
 @implementation NSArray (YOLO)
 
-- (id)sample {   
-    return self[arc4random_uniform((u_int32_t)self.count)];
+- (id)sample {
+    return self.count == 0
+        ? nil
+        : self[arc4random_uniform((u_int32_t)self.count)];
 }
 
 @end
diff --git a/YOLOKit.podspec b/YOLOKit.podspec
@@ -1,6 +1,6 @@
 Pod::Spec.new do |s|
   s.name         = 'YOLOKit'
-  s.version      = '11'
+  s.version      = '11.0.1'
   s.source       = { :git => 'https://github.com/mxcl/YOLOKit.git', :tag => s.version }
   s.requires_arc = true
   s.summary      = 'A delightful library for enumerating Foundation objects.'

diff --git a/tests b/tests
@@ -592,6 +592,10 @@ int main() {
     });
 }
 
+- (void)testEmptySample {
+  XCTAssertNil(@[].sample);
+}
+
 @end
-Original file line number
+Diff line change
@@ Expand Up / @@ -592,6 +592,10 @@ int main() { @@
         });
     }
+    - (void)testEmptySample {
+      XCTAssertNil(@[].sample);
+    }
     @end
@@ Expand Down @@