support redis-namespace client

activejob-traffic_control.gemspec

@@ -28,4 +28,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "minitest", "~> 5.0"
   spec.add_development_dependency "codeclimate-test-reporter", "~> 0.4.7"
   spec.add_development_dependency "connection_pool"
+  spec.add_development_dependency "redis-namespace", "~> 1.6"
 end

lib/active_job/traffic_control.rb

@@ -58,7 +58,7 @@ def client=(cli)
       def client_class_type(client)
         if client.instance_of?(Dalli::Client)
           Suo::Client::Memcached
-        elsif client.instance_of?(::Redis)
+        elsif client.instance_of?(::Redis) || defined?(::Redis::Namespace) && client.instance_of?(::Redis::Namespace)
           Suo::Client::Redis
         else
           raise ArgumentError, "Unsupported client type: #{client}"

test/active_job/traffic_control_test.rb

@@ -235,3 +235,23 @@ def setup
     setup_globals
   end
 end
+
+class RedisNamespacedTrafficControlTest < Minitest::Test
+  include ActiveJob::TrafficControlTest
+
+  def setup
+    ActiveJob::TrafficControl.client = Redis::Namespace.new(:namespace, redis: Redis.new)
+    setup_globals
+  end
+end
+
+class RedisNamespacedPooldedTrafficControlTest < Minitest::Test
+  include ActiveJob::TrafficControlTest
+
+  def setup
+    ActiveJob::TrafficControl.client = ConnectionPool.new(size: 5, timeout: 5) do
+      Redis::Namespace.new(:namespace, redis: Redis.new)
+    end
+    setup_globals
+  end
+end

test/test_helper.rb

@@ -14,6 +14,7 @@
 require "activejob/traffic_control"
 require "active_job"
 require "minitest/autorun"
+require "redis/namespace"
 
 test_logger = begin
   l = Logger.new(STDOUT)