mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
348 lines
11 KiB
Ruby
348 lines
11 KiB
Ruby
require "active_support/core_ext/class/subclasses"
|
|
require "active_support/core_ext/hash/keys"
|
|
|
|
module ActiveJob
|
|
# Provides helper methods for testing Active Job
|
|
module TestHelper
|
|
delegate :enqueued_jobs, :enqueued_jobs=,
|
|
:performed_jobs, :performed_jobs=,
|
|
to: :queue_adapter
|
|
|
|
def before_setup # :nodoc:
|
|
test_adapter = queue_adapter_for_test
|
|
|
|
@old_queue_adapters = (ActiveJob::Base.descendants << ActiveJob::Base).select do |klass|
|
|
# only override explicitly set adapters, a quirk of `class_attribute`
|
|
klass.singleton_class.public_instance_methods(false).include?(:_queue_adapter)
|
|
end.map do |klass|
|
|
[klass, klass.queue_adapter].tap do
|
|
klass.queue_adapter = test_adapter
|
|
end
|
|
end
|
|
|
|
clear_enqueued_jobs
|
|
clear_performed_jobs
|
|
super
|
|
end
|
|
|
|
def after_teardown # :nodoc:
|
|
super
|
|
@old_queue_adapters.each do |(klass, adapter)|
|
|
klass.queue_adapter = adapter
|
|
end
|
|
end
|
|
|
|
# Specifies the queue adapter to use with all active job test helpers.
|
|
#
|
|
# Returns an instance of the queue adapter and defaults to
|
|
# <tt>ActiveJob::QueueAdapters::TestAdapter</tt>.
|
|
#
|
|
# Note: The adapter provided by this method must provide some additional
|
|
# methods from those expected of a standard <tt>ActiveJob::QueueAdapter</tt>
|
|
# in order to be used with the active job test helpers. Refer to
|
|
# <tt>ActiveJob::QueueAdapters::TestAdapter</tt>.
|
|
def queue_adapter_for_test
|
|
ActiveJob::QueueAdapters::TestAdapter.new
|
|
end
|
|
|
|
# Asserts that the number of enqueued jobs matches the given number.
|
|
#
|
|
# def test_jobs
|
|
# assert_enqueued_jobs 0
|
|
# HelloJob.perform_later('david')
|
|
# assert_enqueued_jobs 1
|
|
# HelloJob.perform_later('abdelkader')
|
|
# assert_enqueued_jobs 2
|
|
# end
|
|
#
|
|
# If a block is passed, that block should cause the specified number of
|
|
# jobs to be enqueued.
|
|
#
|
|
# def test_jobs_again
|
|
# assert_enqueued_jobs 1 do
|
|
# HelloJob.perform_later('cristian')
|
|
# end
|
|
#
|
|
# assert_enqueued_jobs 2 do
|
|
# HelloJob.perform_later('aaron')
|
|
# HelloJob.perform_later('rafael')
|
|
# end
|
|
# end
|
|
#
|
|
# The number of times a specific job is enqueued can be asserted.
|
|
#
|
|
# def test_logging_job
|
|
# assert_enqueued_jobs 1, only: LoggingJob do
|
|
# LoggingJob.perform_later
|
|
# HelloJob.perform_later('jeremy')
|
|
# end
|
|
# end
|
|
def assert_enqueued_jobs(number, only: nil)
|
|
if block_given?
|
|
original_count = enqueued_jobs_size(only: only)
|
|
yield
|
|
new_count = enqueued_jobs_size(only: only)
|
|
assert_equal number, new_count - original_count, "#{number} jobs expected, but #{new_count - original_count} were enqueued"
|
|
else
|
|
actual_count = enqueued_jobs_size(only: only)
|
|
assert_equal number, actual_count, "#{number} jobs expected, but #{actual_count} were enqueued"
|
|
end
|
|
end
|
|
|
|
# Asserts that no jobs have been enqueued.
|
|
#
|
|
# def test_jobs
|
|
# assert_no_enqueued_jobs
|
|
# HelloJob.perform_later('jeremy')
|
|
# assert_enqueued_jobs 1
|
|
# end
|
|
#
|
|
# If a block is passed, that block should not cause any job to be enqueued.
|
|
#
|
|
# def test_jobs_again
|
|
# assert_no_enqueued_jobs do
|
|
# # No job should be enqueued from this block
|
|
# end
|
|
# end
|
|
#
|
|
# It can be asserted that no jobs of a specific kind are enqueued:
|
|
#
|
|
# def test_no_logging
|
|
# assert_no_enqueued_jobs only: LoggingJob do
|
|
# HelloJob.perform_later('jeremy')
|
|
# end
|
|
# end
|
|
#
|
|
# Note: This assertion is simply a shortcut for:
|
|
#
|
|
# assert_enqueued_jobs 0, &block
|
|
def assert_no_enqueued_jobs(only: nil, &block)
|
|
assert_enqueued_jobs 0, only: only, &block
|
|
end
|
|
|
|
# Asserts that the number of performed jobs matches the given number.
|
|
# If no block is passed, <tt>perform_enqueued_jobs</tt>
|
|
# must be called around the job call.
|
|
#
|
|
# def test_jobs
|
|
# assert_performed_jobs 0
|
|
#
|
|
# perform_enqueued_jobs do
|
|
# HelloJob.perform_later('xavier')
|
|
# end
|
|
# assert_performed_jobs 1
|
|
#
|
|
# perform_enqueued_jobs do
|
|
# HelloJob.perform_later('yves')
|
|
# assert_performed_jobs 2
|
|
# end
|
|
# end
|
|
#
|
|
# If a block is passed, that block should cause the specified number of
|
|
# jobs to be performed.
|
|
#
|
|
# def test_jobs_again
|
|
# assert_performed_jobs 1 do
|
|
# HelloJob.perform_later('robin')
|
|
# end
|
|
#
|
|
# assert_performed_jobs 2 do
|
|
# HelloJob.perform_later('carlos')
|
|
# HelloJob.perform_later('sean')
|
|
# end
|
|
# end
|
|
#
|
|
# The block form supports filtering. If the :only option is specified,
|
|
# then only the listed job(s) will be performed.
|
|
#
|
|
# def test_hello_job
|
|
# assert_performed_jobs 1, only: HelloJob do
|
|
# HelloJob.perform_later('jeremy')
|
|
# LoggingJob.perform_later
|
|
# end
|
|
# end
|
|
#
|
|
# An array may also be specified, to support testing multiple jobs.
|
|
#
|
|
# def test_hello_and_logging_jobs
|
|
# assert_nothing_raised do
|
|
# assert_performed_jobs 2, only: [HelloJob, LoggingJob] do
|
|
# HelloJob.perform_later('jeremy')
|
|
# LoggingJob.perform_later('stewie')
|
|
# RescueJob.perform_later('david')
|
|
# end
|
|
# end
|
|
# end
|
|
def assert_performed_jobs(number, only: nil)
|
|
if block_given?
|
|
original_count = performed_jobs.size
|
|
perform_enqueued_jobs(only: only) { yield }
|
|
new_count = performed_jobs.size
|
|
assert_equal number, new_count - original_count,
|
|
"#{number} jobs expected, but #{new_count - original_count} were performed"
|
|
else
|
|
performed_jobs_size = performed_jobs.size
|
|
assert_equal number, performed_jobs_size, "#{number} jobs expected, but #{performed_jobs_size} were performed"
|
|
end
|
|
end
|
|
|
|
# Asserts that no jobs have been performed.
|
|
#
|
|
# def test_jobs
|
|
# assert_no_performed_jobs
|
|
#
|
|
# perform_enqueued_jobs do
|
|
# HelloJob.perform_later('matthew')
|
|
# assert_performed_jobs 1
|
|
# end
|
|
# end
|
|
#
|
|
# If a block is passed, that block should not cause any job to be performed.
|
|
#
|
|
# def test_jobs_again
|
|
# assert_no_performed_jobs do
|
|
# # No job should be performed from this block
|
|
# end
|
|
# end
|
|
#
|
|
# The block form supports filtering. If the :only option is specified,
|
|
# then only the listed job(s) will not be performed.
|
|
#
|
|
# def test_no_logging
|
|
# assert_no_performed_jobs only: LoggingJob do
|
|
# HelloJob.perform_later('jeremy')
|
|
# end
|
|
# end
|
|
#
|
|
# Note: This assertion is simply a shortcut for:
|
|
#
|
|
# assert_performed_jobs 0, &block
|
|
def assert_no_performed_jobs(only: nil, &block)
|
|
assert_performed_jobs 0, only: only, &block
|
|
end
|
|
|
|
# Asserts that the job passed in the block has been enqueued with the given arguments.
|
|
#
|
|
# def test_assert_enqueued_with
|
|
# assert_enqueued_with(job: MyJob, args: [1,2,3], queue: 'low') do
|
|
# MyJob.perform_later(1,2,3)
|
|
# end
|
|
#
|
|
# assert_enqueued_with(job: MyJob, at: Date.tomorrow.noon) do
|
|
# MyJob.set(wait_until: Date.tomorrow.noon).perform_later
|
|
# end
|
|
# end
|
|
def assert_enqueued_with(args = {})
|
|
original_enqueued_jobs_count = enqueued_jobs.count
|
|
args.assert_valid_keys(:job, :args, :at, :queue)
|
|
serialized_args = serialize_args_for_assertion(args)
|
|
yield
|
|
in_block_jobs = enqueued_jobs.drop(original_enqueued_jobs_count)
|
|
matching_job = in_block_jobs.find do |job|
|
|
serialized_args.all? { |key, value| value == job[key] }
|
|
end
|
|
assert matching_job, "No enqueued job found with #{args}"
|
|
instantiate_job(matching_job)
|
|
end
|
|
|
|
# Asserts that the job passed in the block has been performed with the given arguments.
|
|
#
|
|
# def test_assert_performed_with
|
|
# assert_performed_with(job: MyJob, args: [1,2,3], queue: 'high') do
|
|
# MyJob.perform_later(1,2,3)
|
|
# end
|
|
#
|
|
# assert_performed_with(job: MyJob, at: Date.tomorrow.noon) do
|
|
# MyJob.set(wait_until: Date.tomorrow.noon).perform_later
|
|
# end
|
|
# end
|
|
def assert_performed_with(args = {})
|
|
original_performed_jobs_count = performed_jobs.count
|
|
args.assert_valid_keys(:job, :args, :at, :queue)
|
|
serialized_args = serialize_args_for_assertion(args)
|
|
perform_enqueued_jobs { yield }
|
|
in_block_jobs = performed_jobs.drop(original_performed_jobs_count)
|
|
matching_job = in_block_jobs.find do |job|
|
|
serialized_args.all? { |key, value| value == job[key] }
|
|
end
|
|
assert matching_job, "No performed job found with #{args}"
|
|
instantiate_job(matching_job)
|
|
end
|
|
|
|
# Performs all enqueued jobs in the duration of the block.
|
|
#
|
|
# def test_perform_enqueued_jobs
|
|
# perform_enqueued_jobs do
|
|
# MyJob.perform_later(1, 2, 3)
|
|
# end
|
|
# assert_performed_jobs 1
|
|
# end
|
|
#
|
|
# This method also supports filtering. If the +:only+ option is specified,
|
|
# then only the listed job(s) will be performed.
|
|
#
|
|
# def test_perform_enqueued_jobs_with_only
|
|
# perform_enqueued_jobs(only: MyJob) do
|
|
# MyJob.perform_later(1, 2, 3) # will be performed
|
|
# HelloJob.perform_later(1, 2, 3) # will not be perfomed
|
|
# end
|
|
# assert_performed_jobs 1
|
|
# end
|
|
def perform_enqueued_jobs(only: nil)
|
|
old_perform_enqueued_jobs = queue_adapter.perform_enqueued_jobs
|
|
old_perform_enqueued_at_jobs = queue_adapter.perform_enqueued_at_jobs
|
|
old_filter = queue_adapter.filter
|
|
|
|
begin
|
|
queue_adapter.perform_enqueued_jobs = true
|
|
queue_adapter.perform_enqueued_at_jobs = true
|
|
queue_adapter.filter = only
|
|
yield
|
|
ensure
|
|
queue_adapter.perform_enqueued_jobs = old_perform_enqueued_jobs
|
|
queue_adapter.perform_enqueued_at_jobs = old_perform_enqueued_at_jobs
|
|
queue_adapter.filter = old_filter
|
|
end
|
|
end
|
|
|
|
# Accesses the queue_adapter set by ActiveJob::Base.
|
|
#
|
|
# def test_assert_job_has_custom_queue_adapter_set
|
|
# assert_instance_of CustomQueueAdapter, HelloJob.queue_adapter
|
|
# end
|
|
def queue_adapter
|
|
ActiveJob::Base.queue_adapter
|
|
end
|
|
|
|
private
|
|
def clear_enqueued_jobs # :nodoc:
|
|
enqueued_jobs.clear
|
|
end
|
|
|
|
def clear_performed_jobs # :nodoc:
|
|
performed_jobs.clear
|
|
end
|
|
|
|
def enqueued_jobs_size(only: nil) # :nodoc:
|
|
if only
|
|
enqueued_jobs.count { |job| Array(only).include?(job.fetch(:job)) }
|
|
else
|
|
enqueued_jobs.count
|
|
end
|
|
end
|
|
|
|
def serialize_args_for_assertion(args) # :nodoc:
|
|
args.dup.tap do |serialized_args|
|
|
serialized_args[:args] = ActiveJob::Arguments.serialize(serialized_args[:args]) if serialized_args[:args]
|
|
serialized_args[:at] = serialized_args[:at].to_f if serialized_args[:at]
|
|
end
|
|
end
|
|
|
|
def instantiate_job(payload) # :nodoc:
|
|
job = payload[:job].new(*payload[:args])
|
|
job.scheduled_at = Time.at(payload[:at]) if payload.key?(:at)
|
|
job.queue_name = payload[:queue]
|
|
job
|
|
end
|
|
end
|
|
end
|