lib/sidekiq/testing.rb in sidekiq-3.3.4 vs lib/sidekiq/testing.rb in sidekiq-3.4.0

@@ -150,19 +150,21 @@
       # Drain and run all jobs for this worker
       def drain
         while job = jobs.shift do
           worker = new
           worker.jid = job['jid']
+          worker.bid = job['bid'] if worker.respond_to?(:bid=)
           execute_job(worker, job['args'])
         end
       end
 
       # Pop out a single job and perform it
       def perform_one
         raise(EmptyQueueError, "perform_one called with empty job queue") if jobs.empty?
         job = jobs.shift
         worker = new
         worker.jid = job['jid']
+        worker.bid = job['bid'] if worker.respond_to?(:bid=)
         execute_job(worker, job['args'])
       end
 
       def execute_job(worker, args)
         worker.perform(*args)