lib/sidekiq/testing.rb in sidekiq-2.12.4 vs lib/sidekiq/testing.rb in sidekiq-2.13.0

@@ -84,18 +84,22 @@
       end
 
       # Drain and run all jobs for this worker
       def drain
         while job = jobs.shift do
-          new.perform(*job['args'])
+          worker = new
+          worker.jid = job['jid']
+          worker.perform(*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
-        new.perform(*job['args'])
+        worker = new
+        worker.jid = job['jid']
+        worker.perform(*job['args'])
       end
     end
 
     class << self
       def jobs # :nodoc: