assert_queued_at_with_queue追加

ubiregiinc · Oct 23, 2024 · 5f227e2 · 5f227e2
1 parent eca03b2
commit 5f227e2
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/lib/resque_unit_without_mock/scheduler_assertions.rb b/lib/resque_unit_without_mock/scheduler_assertions.rb
@@ -1,15 +1,21 @@
 module ResqueUnitWithoutMock::SchedulerAssertions
   def assert_queued_at(expected_timestamp, klass)
-    queue = Resque.queue_for(klass)
+    assert_queued_at_with_queue(Resque.queue_for(klass), expected_timestamp, klass)
+  end
+
+  def assert_not_queued_at(expected_timestamp, klass)
+    assert_not_queued_at_with_queue(Resque.queue_for(klass), expected_timestamp, klass)
+  end
+
+  def assert_queued_at_with_queue(queue, expected_timestamp, klass)
     result = Resque.enqueue_ats(queue).detect { |hash| hash[:timestamp] <= expected_timestamp && hash[:klass] == klass }
     assert(
       result,
       "#{klass} should have been queued in #{Resque.enqueue_ats(queue)} before #{expected_timestamp}"
     )
   end
 
-  def assert_not_queued_at(expected_timestamp, klass)
-    queue = Resque.queue_for(klass)
+  def assert_not_queued_at_with_queue(queue, expected_timestamp, klass)
     result = Resque.enqueue_ats(queue).detect { |hash| hash[:timestamp] <= expected_timestamp && hash[:klass] == klass }
     assert(
       !result,

diff --git a/test/resque_unit_without_mock/scheduler_assertions_test.rb b/test/resque_unit_without_mock/scheduler_assertions_test.rb
@@ -30,7 +30,7 @@ def test_assert_queued_at
     assert_queued_at(Time.new(2011,11,11,0,0,3), PrintJob)
   end
 
-  def test_assert_queued_at_with_queue
+  def test_assert_queued_at_with_class_defined_queue
     #      |=> queued!!
     #  |   |\\|\\\\\\\\
     #  1   2  3
@@ -41,4 +41,16 @@ def test_assert_queued_at_with_queue
     assert_queued_at(Time.new(2011,11,11,0,0,2), PrintJob)
     assert_queued_at(Time.new(2011,11,11,0,0,3), PrintJob)
   end
+
+  def test_assert_queued_at_with_queue
+    #      |=> queued!!
+    #  |   |\\|\\\\\\\\
+    #  1   2  3
+    assert_not_queued_at_with_queue(:urgent, Time.new(2011,11,11,0,0,2), PrintJob)
+    assert_not_queued_at_with_queue(:urgent, Time.new(2011,11,11,0,0,3), PrintJob)
+    Resque.enqueue_at_with_queue(:urgent, Time.new(2011,11,11,0,0,2), PrintJob)
+    assert_not_queued_at_with_queue(:urgent, Time.new(2011,11,11,0,0,1), PrintJob)
+    assert_queued_at_with_queue(:urgent, Time.new(2011,11,11,0,0,2), PrintJob)
+    assert_queued_at_with_queue(:urgent, Time.new(2011,11,11,0,0,3), PrintJob)
+  end
 end