# encoding: UTF-8

require 'vines'
require 'minitest/autorun'

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) }
  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)
  end
end