mixin_test.rb

# frozen_string_literal: true

require "test_helper"

module Limiter
  class MixinTest < Minitest::Test
    include AssertElapsed

    COUNT = 50
    RATE = 1
    INTERVAL = 1

    class MixinTestClass
      extend Limiter::Mixin

      @limited = 0

      class << self
        attr_reader :limited
      end

      limit_method :tick, rate: RATE, interval: INTERVAL
      limit_method :tock, rate: RATE, interval: INTERVAL
      limit_method :ticktock, rate: RATE, interval: INTERVAL do
        @limited += 1
      end

      attr_reader :ticks

      def initialize
        @ticks = 0
      end

      def tick(count = 1)
        @ticks += count
      end

      def tock(count: 1)
        @ticks += count
      end

      def ticktock
        @ticks += 2
      end
    end

    def setup
      super
      RateQueue.any_instance.stubs(:clock).returns(FakeClock)
      @object = MixinTestClass.new
    end

    def test_method_is_rate_limited
      assert_elapsed(COUNT.to_f / RATE - 1) do
        COUNT.times do
          @object.tick
        end
      end
    end

    def test_method_is_rate_limited_across_instances
      assert_elapsed(COUNT.to_f / RATE - 1) do
        COUNT.times do
          MixinTestClass.new.tick
        end
      end
    end

    def test_method_is_not_rate_limited
      assert_elapsed(0) do
        COUNT.times do
          @object.tick
          MixinTestClass.reset_tick_limit!
        end
      end
    end

    def test_method_is_not_rate_limited_across_instances
      assert_elapsed(0) do
        COUNT.times do
          MixinTestClass.new.tick
          MixinTestClass.reset_tick_limit!
        end
      end
    end

    def test_original_method_is_called
      COUNT.times do
        @object.tick
      end

      assert_equal(COUNT, @object.ticks)
    end

    def test_arguments_are_passed
      @object.tick(123)

      assert_equal(123, @object.ticks)
    end

    def test_default_keyword_arguments_are_passed
      COUNT.times do
        @object.tock
      end

      assert_equal(COUNT, @object.ticks)
    end

    def test_keyword_arguments_are_passed
      @object.tock(count: 321)

      assert_equal(321, @object.ticks)
    end

    def test_block_was_called_on_rate_limit
      @object.ticktock
      @object.ticktock
      @object.ticktock

      assert_equal(2, @object.class.limited)
    end
  end
end