test/token_bucket_test.rb in vines-0.4.5 vs test/token_bucket_test.rb in vines-0.4.6

@@ -1,24 +1,44 @@
 # encoding: UTF-8
 
-require 'vines'
-require 'minitest/autorun'
+require 'test_helper'
 
-class TokenBucketTest < MiniTest::Unit::TestCase
-  def test_init
-    assert_raises(ArgumentError) { Vines::TokenBucket.new(0, 1) }
-    assert_raises(ArgumentError) { Vines::TokenBucket.new(1, 0) }
-    assert_raises(ArgumentError) { Vines::TokenBucket.new(-1, 1) }
-    assert_raises(ArgumentError) { Vines::TokenBucket.new(1, -1) }
+describe Vines::TokenBucket do
+  subject { Vines::TokenBucket.new(10, 1) }
+
+  it 'raises with invalid capacity and rate values' do
+    -> { Vines::TokenBucket.new(0, 1) }.must_raise ArgumentError
+    -> { Vines::TokenBucket.new(1, 0) }.must_raise ArgumentError
+    -> { Vines::TokenBucket.new(-1, 1) }.must_raise ArgumentError
+    -> { Vines::TokenBucket.new(1, -1) }.must_raise ArgumentError
   end
 
-  def test_take
-    bucket = Vines::TokenBucket.new(10, 1)
-    assert_raises(ArgumentError) { bucket.take(-1) }
-    assert !bucket.take(11)
-    assert bucket.take(10)
-    assert !bucket.take(1)
-    sleep(1)
-    assert bucket.take(1)
-    assert !bucket.take(1)
+  it 'does not allow taking a negative number of tokens' do
+    -> { subject.take(-1) }.must_raise ArgumentError
+  end
+
+  it 'does not allow taking more tokens than its capacity' do
+    refute subject.take(11)
+  end
+
+  it 'allows taking all tokens, but no more' do
+    assert subject.take(10)
+    refute subject.take(1)
+  end
+
+  it 'refills over time' do
+    assert subject.take(10)
+    refute subject.take(1)
+    Time.stub(:new, Time.now + 1) do
+      assert subject.take(1)
+      refute subject.take(1)
+    end
+  end
+
+  it 'does not refill over capacity' do
+    assert subject.take(10)
+    refute subject.take(1)
+    Time.stub(:new, Time.now + 15) do
+      refute subject.take(11)
+    end
   end
 end