module Sidekiq::TestingExtensions

def jobs_for(klass)
  jobs.select do |job|
    marshalled = job["args"][0]
    marshalled.index(klass.to_s) && YAML.safe_load(marshalled)[0] == klass
  end
end