Allow count to be 0

ice-cube-ruby · Mar 26, 2010 · 3c0ff00 · 3c0ff00
1 parent 3983c95
commit 3c0ff00
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/lib/ice_cube/rule.rb b/lib/ice_cube/rule.rb
@@ -55,7 +55,7 @@ def until(until_date)
 
     # set the number of occurrences after which this rule is no longer effective
     def count(count)
-      raise ArgumentError.new('Argument must be a positive integer') unless Integer(count) && count > 0 #todo - maybe allow count to be 0
+      raise ArgumentError.new('Argument must be a positive integer') unless Integer(count) && count >= 0
       @occurrence_count = count
       self
     end

diff --git a/spec/examples/ice_cube_spec.rb b/spec/examples/ice_cube_spec.rb
@@ -107,4 +107,11 @@
     dates.each { |date| date.sec == 30 }
   end
 
+  it 'ensure that when count on a rule is set to 0, 0 occurrences come back' do
+    start_date = DAY
+    schedule = Schedule.new(start_date)
+    schedule.add_recurrence_rule Rule.daily.count(0)
+    schedule.all_occurrences.should == []
+  end
+
 end