Corrected spec: occurrence_count does not error when not set

Author: avit

spec/examples/ice_cube_spec.rb

@@ -605,11 +605,18 @@
 
   it 'should be able to remove a count validation from a rule' do
     rule = IceCube::Rule.daily.count(5)
-    rule.count nil
-    rule.to_hash.should_not have_key(:count)
+    rule.occurrence_count.should == 5
+    rule.count(nil)
     rule.occurrence_count.should be_nil
   end
 
+  it 'should be able to remove a count validation from a rule' do
+    rule = IceCube::Rule.daily.count(5)
+    rule.to_hash[:count].should == 5
+    rule.count nil
+    rule.to_hash[:count].should be_nil
+  end
+
   it 'should be able to remove an until validation from a rule' do
     rule = IceCube::Rule.daily.until(Time.now + IceCube::ONE_DAY)
     rule.to_hash[:until].should_not be_nil