active-job-enqueued-with-patch

active-job-enqueued-with-patch.rb

module ActiveJobTestHelper
  def prepare_args_for_assertion(args)
    args.dup.tap do |arguments|
      arguments[:at] = arguments[:at].to_f if arguments[:at]
    end
  end

  def assert_enqueued_with(job: nil, args: nil, at: nil, queue: nil)
    expected = { job: job, args: args, at: at, queue: queue }.compact
    expected_args = prepare_args_for_assertion(expected)

    if block_given?
      original_enqueued_jobs_count = enqueued_jobs.count

      yield

      jobs = enqueued_jobs.drop(original_enqueued_jobs_count)
    else
      jobs = enqueued_jobs
    end

    matching_job = jobs.find do |enqueued_job|
      deserialized_job = deserialize_args_for_assertion(enqueued_job)
      expected_args.all? { |key, value| value == deserialized_job[key] }
    end

    assert matching_job, "No enqueued job found with #{expected}"
    instantiate_job(matching_job)
  end

end

ActiveJob::TestHelper.include ActiveJobTestHelper