require File.expand_path(File.dirname(__FILE__)) + '/unit_helper'

require 'lhm/throttler'

describe Lhm::Throttler do
  include UnitHelper

  before :each do
    @mock = Class.new do
      extend Lhm::Throttler
    end
  end

  describe '#setup_throttler' do
    describe 'when passing a key' do
      before do
        @mock.setup_throttler(:time_throttler, :delay => 2)
      end

      it 'instantiates the time throttle' do
        @mock.throttler.class.must_equal Lhm::Throttler::Time
      end

      it 'returns 2 seconds as time' do
        @mock.throttler.timeout_seconds.must_equal 2
      end
    end

    describe 'when passing an instance' do

      before do
        @instance = Class.new(Lhm::Throttler::Time) do
          def timeout_seconds
            0
          end
        end.new

        @mock.setup_throttler(@instance)
      end

      it 'returns the instace given' do
        @mock.throttler.must_equal @instance
      end

      it 'returns 0 seconds as time' do
        @mock.throttler.timeout_seconds.must_equal 0
      end
    end

    describe 'when passing a class' do

      before do
        @klass = Class.new(Lhm::Throttler::Time)
        @mock.setup_throttler(@klass)
      end

      it 'has the same class as given' do
        @mock.throttler.class.must_equal @klass
      end
    end
  end

  describe '#throttler' do

    it 'returns the default Time based' do
      @mock.throttler.class.must_equal Lhm::Throttler::Time
    end

    it 'should default to 100 milliseconds' do
      @mock.throttler.timeout_seconds.must_equal 0.1
    end
  end
end