sidekiq-queue-throttled 1.0.0

data/lib/sidekiq/queue_throttled/middleware.rb

@@ -0,0 +1,92 @@
+# frozen_string_literal: true
+
+module Sidekiq
+  module QueueThrottled
+    class Middleware
+      def initialize
+        @queue_limiters = Concurrent::Map.new
+        @job_throttlers = Concurrent::Map.new
+      end
+
+      def call(worker, job, queue)
+        queue_name = job['queue'] || queue
+        job_class = worker.class.name
+
+        # Check queue-level limits
+        queue_limiter = get_queue_limiter(queue_name)
+        if queue_limiter
+          lock_id = queue_limiter.acquire_lock
+          unless lock_id
+            Sidekiq::QueueThrottled.logger.info "Queue limit reached for #{queue_name}, rescheduling job"
+            reschedule_job(job, queue_name)
+            return nil
+          end
+        end
+
+        # Check job-level throttling
+        job_throttler = get_job_throttler(job_class)
+        if job_throttler && !job_throttler.acquire_slot(job['args'])
+          Sidekiq::QueueThrottled.logger.info "Job throttling limit reached for #{job_class}, rescheduling job"
+          reschedule_job(job, queue_name)
+          return nil
+        end
+
+        # Process the job
+        begin
+          yield
+        ensure
+          # Release locks
+          queue_limiter&.release_lock(lock_id)
+          job_throttler&.release_slot(job['args'])
+        end
+      end
+
+      private
+
+      def get_queue_limiter(queue_name)
+        limit = Sidekiq::QueueThrottled.configuration.queue_limit(queue_name)
+        return nil unless limit
+
+        @queue_limiters.compute_if_absent(queue_name) do
+          QueueLimiter.new(queue_name, limit)
+        end
+      end
+
+      def get_job_throttler(job_class)
+        throttle_config = get_throttle_config(job_class)
+        return nil unless throttle_config
+
+        @job_throttlers.compute_if_absent(job_class) do
+          JobThrottler.new(job_class, throttle_config)
+        end
+      end
+
+      def get_throttle_config(job_class)
+        # Handle string class names
+        if job_class.is_a?(String)
+          begin
+            klass = Object.const_get(job_class)
+            return klass.sidekiq_throttle_config if klass.respond_to?(:sidekiq_throttle_config)
+          rescue NameError
+            # For test classes that don't have proper constant names
+            return nil
+          end
+        elsif job_class.respond_to?(:sidekiq_throttle_config)
+          # Handle actual class objects
+          return job_class.sidekiq_throttle_config
+        end
+        nil
+      end
+
+      def reschedule_job(job, queue_name)
+        delay = Sidekiq::QueueThrottled.configuration.retry_delay
+        job['at'] = Time.now.to_f + delay
+        job['queue'] = queue_name
+
+        Sidekiq.redis do |conn|
+          conn.zadd('schedule', job['at'], job.to_json)
+        end
+      end
+    end
+  end
+end

data/lib/sidekiq/queue_throttled/queue_limiter.rb

@@ -0,0 +1,110 @@
+# frozen_string_literal: true
+
+module Sidekiq
+  module QueueThrottled
+    class QueueLimiter
+      attr_reader :queue_name, :limit, :redis
+
+      def initialize(queue_name, limit, redis = nil)
+        @queue_name = queue_name.to_s
+        @limit = limit.to_i
+        @redis = redis || Sidekiq::QueueThrottled.redis
+        @lock_key = "#{Sidekiq::QueueThrottled.configuration.redis_key_prefix}:queue:#{@queue_name}:lock"
+        @counter_key = "#{Sidekiq::QueueThrottled.configuration.redis_key_prefix}:queue:#{@queue_name}:counter"
+        @mutex = Concurrent::ReentrantReadWriteLock.new
+      end
+
+      def acquire_lock(worker_id = nil)
+        worker_id ||= SecureRandom.uuid
+        lock_id = "#{worker_id}:#{Time.now.to_f}"
+
+        @mutex.with_write_lock do
+          current_count = get_current_count
+          puts "DEBUG: QueueLimiter - current_count: #{current_count}, limit: #{@limit}"
+          return false if current_count >= @limit
+
+          # Increment the counter first
+          increment_counter
+          puts "DEBUG: QueueLimiter - acquired lock: #{lock_id}"
+          return lock_id
+        end
+
+        false
+      end
+
+      def release_lock(lock_id)
+        return false unless lock_id
+
+        @mutex.with_write_lock do
+          # For time-based limiting, we don't immediately decrement the counter
+          # The counter will expire naturally after the TTL period
+          # This prevents immediate reuse of slots
+          true
+        end
+      rescue StandardError => e
+        Sidekiq::QueueThrottled.logger.error "Failed to release lock #{lock_id} for queue #{@queue_name}: #{e.message}"
+        false
+      end
+
+      def current_count
+        @mutex.with_read_lock do
+          get_current_count
+        end
+      end
+
+      def available_slots
+        [0, @limit - current_count].max
+      end
+
+      def reset!
+        @mutex.with_write_lock do
+          @redis.del(@counter_key)
+          # Clear all locks for this queue
+          pattern = "#{Sidekiq::QueueThrottled.configuration.redis_key_prefix}:queue:#{@queue_name}:lock:*"
+          keys = @redis.keys(pattern)
+          @redis.del(*keys) unless keys.empty?
+        end
+      end
+
+      private
+
+      def get_current_count
+        count = @redis.get(@counter_key)
+        result = count ? count.to_i : 0
+        puts "DEBUG: get_current_count - key: #{@counter_key}, count: #{result}"
+        result
+      end
+
+      def increment_counter
+        puts "DEBUG: increment_counter - key: #{@counter_key}"
+        @redis.multi do |multi|
+          multi.incr(@counter_key)
+          multi.expire(@counter_key, Sidekiq::QueueThrottled.configuration.throttle_ttl)
+        end
+        puts "DEBUG: increment_counter - after increment, count: #{get_current_count}"
+      end
+
+      def decrement_counter
+        @redis.multi do |multi|
+          multi.decr(@counter_key)
+          multi.expire(@counter_key, Sidekiq::QueueThrottled.configuration.throttle_ttl)
+        end
+        # Ensure counter doesn't go below 0
+        current = get_current_count
+        if current.negative?
+          @redis.set(@counter_key, 0)
+        end
+      end
+
+      def acquire_redis_lock(lock_id)
+        lock_key = "#{@lock_key}:#{lock_id}"
+        @redis.set(lock_key, '1', nx: true, ex: Sidekiq::QueueThrottled.configuration.lock_ttl)
+      end
+
+      def release_redis_lock(lock_id)
+        lock_key = "#{@lock_key}:#{lock_id}"
+        @redis.del(lock_key).positive?
+      end
+    end
+  end
+end

data/lib/sidekiq/queue_throttled/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Sidekiq
+  module QueueThrottled
+    VERSION = '1.0.0'
+  end
+end

data/lib/sidekiq/queue_throttled.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require 'sidekiq'
+require 'redis'
+require 'concurrent'
+require 'json'
+require 'logger'
+
+require_relative 'queue_throttled/version'
+require_relative 'queue_throttled/configuration'
+require_relative 'queue_throttled/queue_limiter'
+require_relative 'queue_throttled/job_throttler'
+require_relative 'queue_throttled/middleware'
+require_relative 'queue_throttled/job'
+
+module Sidekiq
+  module QueueThrottled
+    class << self
+      def configure
+        yield configuration
+      end
+
+      def configuration
+        @configuration ||= Configuration.new
+      end
+
+      attr_writer :configuration, :logger, :redis
+
+      def logger
+        @logger ||= begin
+          logger = Logger.new($stdout)
+          logger.level = Logger::INFO
+          logger
+        end
+      end
+
+      def redis
+        @redis ||= Sidekiq.redis { |conn| conn }
+      end
+    end
+  end
+end
+
+# Auto-load the middleware when the gem is required
+Sidekiq.configure_server do |config|
+  config.server_middleware do |chain|
+    chain.add Sidekiq::QueueThrottled::Middleware
+  end
+end

data/spec/examples.txt

@@ -0,0 +1,110 @@
+example_id                                                    | status | run_time        |
+------------------------------------------------------------- | ------ | --------------- |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:1:1]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:1:2]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:1:3]   | passed | 0.00009 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:2:1]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:2:2]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:2:3]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:3:1]   | passed | 0.00019 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:3:2]   | passed | 0.0002 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:3:3]   | passed | 0.00354 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:4:1]   | passed | 0.00031 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:4:2]   | passed | 0.00437 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:4:3]   | passed | 0.01693 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:5:1]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:5:2]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:5:3]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:5:4]   | passed | 0.00011 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:6:1]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:6:2]   | passed | 0.00009 seconds |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:6:3]   | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/configuration_spec.rb[1:6:4]   | passed | 0.00009 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:1:1]             | passed | 0.00014 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:1:2]             | passed | 0.00013 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:1]             | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:2]             | passed | 0.00019 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:3]             | passed | 0.00012 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:4]             | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:5]             | passed | 0.00017 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:6]             | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:7]             | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:8]             | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:9]             | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:10]            | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:11]            | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:12]            | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:2:13]            | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:3:1]             | passed | 0.00072 seconds |
+./spec/sidekiq/queue_throttled/job_spec.rb[1:3:2]             | passed | 0.00139 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:1:1]   | passed | 0.00024 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:1:2]   | passed | 0.00018 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:1:1] | passed | 0.00011 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:2:1] | passed | 0.00022 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:2:2] | passed | 0.00065 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:2:3] | passed | 0.00053 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:3:1] | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:3:2] | passed | 0.00047 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:3:3] | passed | 0.00072 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:2:4:1] | passed | 0.0003 seconds  |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:1:1] | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:2:1] | passed | 0.00037 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:2:2] | passed | 0.00066 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:2:3] | passed | 0.00079 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:3:1] | passed | 0.00022 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:3:3:2] | passed | 0.00043 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:4:1:1] | passed | 0.0045 seconds  |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:4:2:1] | passed | 0.00214 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:4:2:2] | passed | 0.00612 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:4:3:1] | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:5:1:1] | passed | 0.00066 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:5:2:1] | passed | 0.00079 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:5:2:2] | passed | 0.00024 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:5:3:1] | passed | 0.00054 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:5:4:1] | passed | 0.00068 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:6:1]   | passed | 0.00155 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:6:2]   | passed | 0.00058 seconds |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:6:3]   | passed | 0.0007 seconds  |
+./spec/sidekiq/queue_throttled/job_throttler_spec.rb[1:7:1]   | passed | 0.0013 seconds  |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:1:1]    | passed | 0.00033 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:2:1]    | passed | 0.00045 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:2:2]    | passed | 0.00296 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:3:1]    | passed | 0.00073 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:3:2]    | passed | 0.00098 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:3:3]    | passed | 0.00104 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:4:1]    | passed | 0.00105 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:5:1]    | passed | 0.00068 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:1:5:2]    | passed | 0.00223 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:2:1]      | passed | 0.00009 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:2:2]      | passed | 0.00013 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:2:3]      | passed | 0.0001 seconds  |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:3:1]      | passed | 0.00015 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:3:2]      | passed | 0.00058 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:3:3]      | passed | 0.00023 seconds |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:4:1]      | passed | 0.0005 seconds  |
+./spec/sidekiq/queue_throttled/middleware_spec.rb[1:4:2]      | passed | 0.00098 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:1:1]   | passed | 0.00013 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:1:2]   | passed | 0.00014 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:1:3]   | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:2:1]   | passed | 0.0004 seconds  |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:2:2]   | passed | 0.0011 seconds  |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:2:3]   | passed | 0.00088 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:2:4]   | passed | 0.00046 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:2:5]   | passed | 0.00058 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:3:1]   | passed | 0.00058 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:3:2]   | passed | 0.00014 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:3:3]   | passed | 0.00016 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:3:4]   | passed | 0.00038 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:4:1]   | passed | 0.0002 seconds  |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:4:2]   | passed | 0.00098 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:4:3]   | passed | 0.00095 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:5:1]   | passed | 0.00021 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:5:2]   | passed | 0.00045 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:5:3]   | passed | 0.00112 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:5:4]   | passed | 0.00023 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:6:1]   | passed | 0.00081 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:6:2]   | passed | 0.00152 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:7:1]   | passed | 0.00151 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:7:2]   | passed | 0.00136 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:8:1]   | passed | 0.00066 seconds |
+./spec/sidekiq/queue_throttled/queue_limiter_spec.rb[1:8:2]   | passed | 0.00093 seconds |

data/spec/sidekiq/queue_throttled/configuration_spec.rb

@@ -0,0 +1,145 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Sidekiq::QueueThrottled::Configuration do
+  let(:config) { described_class.new }
+
+  describe '#set_queue_limit' do
+    it 'sets queue limit as integer' do
+      config.set_queue_limit(:test_queue, 100)
+      expect(config.queue_limit(:test_queue)).to eq(100)
+    end
+
+    it 'converts string limit to integer' do
+      config.set_queue_limit(:test_queue, '50')
+      expect(config.queue_limit(:test_queue)).to eq(50)
+    end
+
+    it 'handles string queue names' do
+      config.set_queue_limit('test_queue', 100)
+      expect(config.queue_limit('test_queue')).to eq(100)
+    end
+  end
+
+  describe '#queue_limit' do
+    it 'returns limit for string queue name' do
+      config.set_queue_limit('test_queue', 100)
+      expect(config.queue_limit('test_queue')).to eq(100)
+    end
+
+    it 'returns limit for symbol queue name' do
+      config.set_queue_limit(:test_queue, 100)
+      expect(config.queue_limit(:test_queue)).to eq(100)
+    end
+
+    it 'returns nil for non-existent queue' do
+      expect(config.queue_limit(:non_existent)).to be_nil
+    end
+  end
+
+  describe '#load_from_sidekiq_config!' do
+    let(:sidekiq_config) do
+      {
+        limits: {
+          'queue1' => 100,
+          'queue2' => 50
+        }
+      }
+    end
+
+    it 'loads limits from Sidekiq options' do
+      config.load_from_sidekiq_config!(sidekiq_config)
+      expect(config.queue_limit('queue1')).to eq(100)
+      expect(config.queue_limit('queue2')).to eq(50)
+    end
+
+    it 'handles string keys in limits' do
+      sidekiq_config_with_string_keys = {
+        'limits' => {
+          'queue1' => 100
+        }
+      }
+      config.load_from_sidekiq_config!(sidekiq_config_with_string_keys)
+      expect(config.queue_limit('queue1')).to eq(100)
+    end
+
+    it 'does nothing when limits are not defined' do
+      empty_config = {}
+      expect { config.load_from_sidekiq_config!(empty_config) }.not_to(change { config.queue_limits })
+    end
+  end
+
+  describe '#load_from_yaml!' do
+    let(:yaml_content) do
+      <<~YAML
+        limits:
+          queue1: 100
+          queue2: 50
+      YAML
+    end
+
+    it 'loads limits from YAML content' do
+      config.load_from_yaml!(yaml_content)
+      expect(config.queue_limit('queue1')).to eq(100)
+      expect(config.queue_limit('queue2')).to eq(50)
+    end
+
+    it 'handles string keys in YAML' do
+      yaml_with_string_keys = <<~YAML
+        "limits":
+          "queue1": 100
+      YAML
+      config.load_from_yaml!(yaml_with_string_keys)
+      expect(config.queue_limit('queue1')).to eq(100)
+    end
+
+    it 'does nothing when limits are not defined' do
+      yaml_without_limits = <<~YAML
+        other_config: value
+      YAML
+      expect { config.load_from_yaml!(yaml_without_limits) }.not_to(change { config.queue_limits })
+    end
+  end
+
+  describe '#validate!' do
+    it 'passes validation for valid limits' do
+      config.set_queue_limit(:queue1, 100)
+      config.set_queue_limit(:queue2, 50)
+      expect { config.validate! }.not_to raise_error
+    end
+
+    it 'raises error for non-positive limits' do
+      config.set_queue_limit(:queue1, 0)
+      expect { config.validate! }.to raise_error(ArgumentError, /positive integer/)
+    end
+
+    it 'raises error for negative limits' do
+      config.set_queue_limit(:queue1, -1)
+      expect { config.validate! }.to raise_error(ArgumentError, /positive integer/)
+    end
+
+    it 'raises error for non-integer limits' do
+      config.queue_limits[:queue1] = 'invalid'
+      expect { config.validate! }.to raise_error(ArgumentError, /positive integer/)
+    end
+  end
+
+  describe 'default values' do
+    it 'has correct default redis_key_prefix' do
+      expect(config.redis_key_prefix).to eq('sidekiq:queue_throttled')
+    end
+
+    it 'has correct default throttle_ttl' do
+      expect(config.throttle_ttl).to eq(3600)
+    end
+
+    it 'has correct default lock_ttl' do
+      expect(config.lock_ttl).to eq(300)
+    end
+
+    it 'has correct default retry_delay' do
+      expect(config.retry_delay).to eq(5)
+    end
+  end
+end