Contents

require 'sidekiq/testing'

module Sidekiq
  module Worker
    module ClassMethods
      # Drain and run all jobs for this worker
      def drain
        while (job = jobs.shift)
          worker = new
          worker.jid = job['jid']
          worker.bid = job['bid'] if worker.respond_to?(:bid=)
          execute_job(worker, job['args'])
          unlock(job) if Sidekiq::Testing.fake?
        end
      end unless Sidekiq::Testing.respond_to?(:server_middleware)

      # 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'])
        unlock(job) if Sidekiq::Testing.fake?
      end unless Sidekiq::Testing.respond_to?(:server_middleware)

      # Clear all jobs for this worker
      def clear
        jobs.each do |job|
          unlock(job) if Sidekiq::Testing.fake?
        end
        # if Sidekiq::VERSION >= '4'
        #   Queues.jobs[queue].clear
        # else
        jobs.clear
        # end
      end

      def execute_job(worker, args)
        worker.perform(*args)
      end unless respond_to?(:execute_job)

      def unlock(job)
        SidekiqUniqueJobs::Unlockable.unlock(job)
      end
    end

    module Overrides
      def self.included(base)
        base.extend Testing
        base.class_eval do
          class << self
            alias_method :clear_all_orig, :clear_all
            alias_method :clear_all, :clear_all_ext
          end
        end
      end

      module Testing
        def clear_all_ext
          SidekiqUniqueJobs::Util.del('*', 1000, false) unless SidekiqUniqueJobs.mocked?
          clear_all_orig
        end
      end
    end

    include Overrides
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
sidekiq-unique-jobs-4.0.18	lib/sidekiq_unique_jobs/testing/sidekiq_overrides.rb