Extend stdlib queue for the test queue

1 files changed, 10 insertions, 10 deletions

diff --git a/railties/test/queueing/test_queue_test.rb b/railties/test/queueing/test_queue_test.rb
index 466af22351..fbb09c6fa5 100644
--- a/railties/test/queueing/test_queue_test.rb
+++ b/railties/test/queueing/test_queue_test.rb
@@ -3,9 +3,7 @@ require 'rails/queueing'
 
 class TestQueueTest < ActiveSupport::TestCase
   class Job
-    attr_reader :id
-    def initialize(id, &block)
-      @id = id
+    def initialize(&block)
       @block = block
     end
 
@@ -15,21 +13,23 @@ class TestQueueTest < ActiveSupport::TestCase
   end
 
   def setup
+    #@queue = Rails::Queueing::TestQueue.new
     @queue = Rails::Queueing::TestQueue.new
   end
 
   def test_contents
-    assert_equal [], @queue.contents
-    job = Job.new(1)
+    assert @queue.empty?
+    job = Job.new
     @queue.push job
-    assert_equal [job], @queue.contents
+    refute @queue.empty?
+    assert_equal job, @queue.pop
   end
 
   def test_order
     processed = []
 
-    job1 = Job.new(1) { processed << 1 }
-    job2 = Job.new(2) { processed << 2 }
+    job1 = Job.new { processed << 1 }
+    job2 = Job.new { processed << 2 }
 
     @queue.push job1
     @queue.push job2
@@ -42,7 +42,7 @@ class TestQueueTest < ActiveSupport::TestCase
     t = nil
     ran = false
 
-    job = Job.new(1) do
+    job = Job.new do
       ran = true
       t = Thread.current
     end
@@ -50,7 +50,7 @@ class TestQueueTest < ActiveSupport::TestCase
     @queue.push job
     @queue.drain
 
-    assert_equal [], @queue.contents
+    assert @queue.empty?
     assert ran, "The job runs synchronously when the queue is drained"
     assert_not_equal t, Thread.current
   end