Module: ActiveJob::TestHelper

Included in:
TestCase
Defined in:
lib/active_job/test_helper.rb

Overview

Provides helper methods for testing Active Job

Defined Under Namespace

Modules: TestQueueAdapter

Instance Method Summary collapse

Instance Method Details

#after_teardown ⇒ Object

:nodoc:



47
48
49
50
51
 
# File 'lib/active_job/test_helper.rb', line 47

def after_teardown # :nodoc:
  super

  queue_adapter_changed_jobs.each { |klass| klass.disable_test_adapter }
end

#assert_enqueued_jobs(number, only: nil, queue: nil) ⇒ Object

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 will 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

The number of times a job is enqueued to a specific queue can also be asserted.

def test_logging_job
  assert_enqueued_jobs 2, queue: 'default' do
    LoggingJob.perform_later
    HelloJob.perform_later('elfassy')
  end
end



107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/active_job/test_helper.rb', line 107

def assert_enqueued_jobs(number, only: nil, queue: nil)
  if block_given?
    original_count = enqueued_jobs_size(only: only, queue: queue)
    yield
    new_count = enqueued_jobs_size(only: only, queue: queue)
    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, queue: queue)
    assert_equal number, actual_count, "#{number} jobs expected, but #{actual_count} were enqueued"
  end
end

#assert_enqueued_with(job: nil, args: nil, at: nil, queue: nil) ⇒ Object

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



262
263
264
265
266
267
268
269
270
271
272
273
 
# File 'lib/active_job/test_helper.rb', line 262

def assert_enqueued_with(job: nil, args: nil, at: nil, queue: nil)
  original_enqueued_jobs_count = enqueued_jobs.count
  expected = { job: job, args: args, at: at, queue: queue }.compact
  serialized_args = serialize_args_for_assertion(expected)
  yield
  in_block_jobs = enqueued_jobs.drop(original_enqueued_jobs_count)
  matching_job = in_block_jobs.find do |in_block_job|
    serialized_args.all? { |key, value| value == in_block_job[key] }
  end
  assert matching_job, "No enqueued job found with #{expected}"
  instantiate_job(matching_job)
end

#assert_no_enqueued_jobs(only: nil, &block) ⇒ Object

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



146
147
148
 
# File 'lib/active_job/test_helper.rb', line 146

def assert_no_enqueued_jobs(only: nil, &block)
  assert_enqueued_jobs 0, only: only, &block
end

#assert_no_performed_jobs(only: nil, &block) ⇒ Object

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



247
248
249
 
# File 'lib/active_job/test_helper.rb', line 247

def assert_no_performed_jobs(only: nil, &block)
  assert_performed_jobs 0, only: only, &block
end

#assert_performed_jobs(number, only: nil) ⇒ Object

Asserts that the number of performed jobs matches the given number. If no block is passed, perform_enqueued_jobs 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



203
204
205
206
207
208
209
210
211
212
213
214
 
# File 'lib/active_job/test_helper.rb', line 203

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

#assert_performed_with(job: nil, args: nil, at: nil, queue: nil) ⇒ Object

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



286
287
288
289
290
291
292
293
294
295
296
297
 
# File 'lib/active_job/test_helper.rb', line 286

def assert_performed_with(job: nil, args: nil, at: nil, queue: nil)
  original_performed_jobs_count = performed_jobs.count
  expected = { job: job, args: args, at: at, queue: queue }.compact
  serialized_args = serialize_args_for_assertion(expected)
  perform_enqueued_jobs { yield }
  in_block_jobs = performed_jobs.drop(original_performed_jobs_count)
  matching_job = in_block_jobs.find do |in_block_job|
    serialized_args.all? { |key, value| value == in_block_job[key] }
  end
  assert matching_job, "No performed job found with #{expected}"
  instantiate_job(matching_job)
end

#before_setup ⇒ Object

:nodoc:



35
36
37
38
39
40
41
42
43
44
45
 
# File 'lib/active_job/test_helper.rb', line 35

def before_setup # :nodoc:
  test_adapter = queue_adapter_for_test

  queue_adapter_changed_jobs.each do |klass|
    klass.enable_test_adapter(test_adapter)
  end

  clear_enqueued_jobs
  clear_performed_jobs
  super
end

#perform_enqueued_jobs(only: nil) ⇒ Object

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 performed
  end
  assert_performed_jobs 1
end



318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
 
# File 'lib/active_job/test_helper.rb', line 318

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

#queue_adapter ⇒ Object

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



340
341
342
 
# File 'lib/active_job/test_helper.rb', line 340

def queue_adapter
  ActiveJob::Base.queue_adapter
end

#queue_adapter_for_test ⇒ Object

Specifies the queue adapter to use with all active job test helpers.

Returns an instance of the queue adapter and defaults to ActiveJob::QueueAdapters::TestAdapter.

Note: The adapter provided by this method must provide some additional methods from those expected of a standard ActiveJob::QueueAdapter in order to be used with the active job test helpers. Refer to ActiveJob::QueueAdapters::TestAdapter.



62
63
64
 
# File 'lib/active_job/test_helper.rb', line 62

def queue_adapter_for_test
  ActiveJob::QueueAdapters::TestAdapter.new
end