railties/test/queueing/test_queue_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
48
49
50
51
52
53
54
55
56
57

require 'abstract_unit'
require 'rails/queueing'

class TestQueueTest < ActiveSupport::TestCase
  class Job
    attr_reader :id
    def initialize(id, &block)
      @id = id
      @block = block
    end

    def run
      @block.call if @block
    end
  end

  def setup
    @queue = Rails::Queueing::TestQueue.new
  end

  def test_contents
    assert_equal [], @queue.contents
    job = Job.new(1)
    @queue.push job
    assert_equal [job], @queue.contents
  end

  def test_order
    time1 = time2 = nil

    job1 = Job.new(1) { time1 = Time.now }
    job2 = Job.new(2) { time2 = Time.now }

    @queue.push job1
    @queue.push job2
    @queue.drain

    assert time1 < time2, "Jobs run in the same order they were added"
  end

  def test_drain
    t = nil
    ran = false

    job = Job.new(1) do
      ran = true
      t = Thread.current
    end

    @queue.push job
    @queue.drain

    assert_equal [], @queue.contents
    assert ran, "The job runs synchronously when the queue is drained"
    assert_not_equal t, Thread.current
  end
end