1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activesupport/test/queueing/test_queue_test.rb

147 lines
2.5 KiB
Ruby
Raw Normal View History

2012-04-27 00:43:12 -04:00
require 'abstract_unit'
2012-09-13 18:09:15 -04:00
require 'active_support/queueing'
2012-04-27 00:43:12 -04:00
class TestQueueTest < ActiveSupport::TestCase
def setup
2012-09-13 18:09:15 -04:00
@queue = ActiveSupport::TestQueue.new
end
2012-04-27 00:43:12 -04:00
class ExceptionRaisingJob
2012-04-27 00:43:12 -04:00
def run
raise
2012-04-27 00:43:12 -04:00
end
end
def test_drain_raises_exceptions_from_running_jobs
@queue.push ExceptionRaisingJob.new
assert_raises(RuntimeError) { @queue.drain }
end
def test_jobs
@queue.push 1
@queue.push 2
assert_equal [1,2], @queue.jobs
end
class EquivalentJob
def initialize
@initial_id = self.object_id
end
def run
end
def ==(other)
other.same_initial_id?(@initial_id)
end
def same_initial_id?(other_id)
other_id == @initial_id
end
end
2012-04-27 00:43:12 -04:00
def test_contents
job = EquivalentJob.new
assert @queue.empty?
2012-04-27 00:43:12 -04:00
@queue.push job
2012-04-27 17:26:26 -04:00
refute @queue.empty?
assert_equal job, @queue.pop
2012-04-27 00:43:12 -04:00
end
class ProcessingJob
def self.clear_processed
@processed = []
end
def self.processed
@processed
end
def initialize(object)
@object = object
end
def run
self.class.processed << @object
end
end
2012-04-27 16:21:51 -04:00
def test_order
ProcessingJob.clear_processed
job1 = ProcessingJob.new(1)
job2 = ProcessingJob.new(2)
2012-04-27 16:21:51 -04:00
@queue.push job1
@queue.push job2
@queue.drain
2012-04-27 16:21:51 -04:00
assert_equal [1,2], ProcessingJob.processed
end
class ThreadTrackingJob
attr_reader :thread_id
2012-04-27 00:43:12 -04:00
def run
@thread_id = Thread.current.object_id
2012-04-27 00:43:12 -04:00
end
def ran?
@thread_id
end
end
def test_drain
@queue.push ThreadTrackingJob.new
job = @queue.jobs.last
2012-04-27 00:43:12 -04:00
@queue.drain
2012-04-27 17:26:26 -04:00
assert @queue.empty?
assert job.ran?, "The job runs synchronously when the queue is drained"
assert_equal job.thread_id, Thread.current.object_id
2012-04-27 00:43:12 -04:00
end
class IdentifiableJob
def initialize(id)
@id = id
end
def ==(other)
other.same_id?(@id)
end
def same_id?(other_id)
other_id == @id
end
def run
end
end
def test_queue_can_be_observed
jobs = (1..10).map do |id|
IdentifiableJob.new(id)
end
jobs.each do |job|
@queue.push job
end
assert_equal jobs, @queue.jobs
end
def test_adding_an_unmarshallable_job
anonymous_class_instance = Struct.new(:run).new
assert_raises TypeError do
@queue.push anonymous_class_instance
end
end
def test_attempting_to_add_a_reference_to_itself
job = {reference: @queue}
assert_raises TypeError do
@queue.push job
end
end
2012-04-27 00:43:12 -04:00
end