activejob/test/cases/queue_priority_test.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

require 'helper'
require 'jobs/hello_job'

class QueuePriorityTest < ActiveSupport::TestCase
  test 'priority unset by default' do
    assert_equal nil, HelloJob.priority
  end

  test 'uses given priority' do
    original_priority = HelloJob.priority

    begin
      HelloJob.queue_with_priority 90
      assert_equal 90, HelloJob.new.priority
    ensure
      HelloJob.priority = original_priority
    end
  end

  test 'evals block given to priority to determine priority' do
    original_priority = HelloJob.priority

    begin
      HelloJob.queue_with_priority { 25 }
      assert_equal 25, HelloJob.new.priority
    ensure
      HelloJob.priority = original_priority
    end
  end

  test 'can use arguments to determine priority in priority block' do
    original_priority = HelloJob.priority

    begin
      HelloJob.queue_with_priority { self.arguments.first=='1' ? 99 : 11 }
      assert_equal 99, HelloJob.new('1').priority
      assert_equal 11, HelloJob.new('3').priority
    ensure
      HelloJob.priority = original_priority
    end
  end

  test 'uses priority passed to #set' do
    job = HelloJob.set(priority: 123).perform_later
    assert_equal 123, job.priority
  end
end