Rejigger raw_push for Pro's atomic batch fix

lib/sidekiq/client.rb

@@ -176,23 +176,26 @@ def enqueue_in(interval, klass, *args)
     private
 
     def raw_push(payloads)
-      pushed = false
       @redis_pool.with do |conn|
-        if payloads.first['at']
-          pushed = conn.zadd('schedule', payloads.map do |hash|
-            at = hash.delete('at').to_s
-            [at, Sidekiq.dump_json(hash)]
-          end)
-        else
-          q = payloads.first['queue']
-          to_push = payloads.map { |entry| Sidekiq.dump_json(entry) }
-          _, pushed = conn.multi do
-            conn.sadd('queues', q)
-            conn.lpush("queue:#{q}", to_push)
-          end
+        conn.multi do
+          atomic_push(conn, payloads)
         end
       end
-      pushed
+      true
+    end
+
+    def atomic_push(conn, payloads)
+      if payloads.first['at']
+        conn.zadd('schedule', payloads.map do |hash|
+          at = hash.delete('at').to_s
+          [at, Sidekiq.dump_json(hash)]
+        end)
+      else
+        q = payloads.first['queue']
+        to_push = payloads.map { |entry| Sidekiq.dump_json(entry) }
+        conn.sadd('queues', q)
+        conn.lpush("queue:#{q}", to_push)
+      end
     end
 
     def process_single(worker_class, item)