activejob/test/support/integration/test_case_helpers.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

require "active_support/core_ext/string/inflections"
require "support/integration/jobs_manager"

module TestCaseHelpers
  extend ActiveSupport::Concern

  included do
    self.use_transactional_tests = false

    setup do
      clear_jobs
      @id = "AJ-#{SecureRandom.uuid}"
    end

    teardown do
      clear_jobs
    end
  end

  protected

    def jobs_manager
      JobsManager.current_manager
    end

    def clear_jobs
      jobs_manager.clear_jobs
    end

    def adapter_is?(*adapter_class_symbols)
      adapter = ActiveJob::Base.queue_adapter.class.name.demodulize.chomp("Adapter").underscore
      adapter_class_symbols.map(&:to_s).include? adapter
    end

    def wait_for_jobs_to_finish_for(seconds = 60)
      begin
        Timeout.timeout(seconds) do
          while !job_executed do
            sleep 0.25
          end
        end
      rescue Timeout::Error
      end
    end

    def job_file(id)
      Dummy::Application.root.join("tmp/#{id}")
    end

    def job_executed(id = @id)
      job_file(id).exist?
    end

    def job_data(id)
      Marshal.load(File.binread(job_file(id)))
    end

    def job_executed_at(id = @id)
      job_data(id)["executed_at"]
    end

    def job_executed_in_locale(id = @id)
      job_data(id)["locale"]
    end
end