resque_unit 0.2.4 → 0.2.5

data/lib/resque_unit/resque.rb

@@ -5,23 +5,29 @@ module Resque
   # Resets all the queues to the empty state. This should be called in
   # your test's +setup+ method until I can figure out a way for it to
   # automatically be called.
-  def self.reset!
-    @queue = Hash.new { |h, k| h[k] = [] }
+  #
+  # If <tt>queue_name</tt> is given, then resets only that queue.
+  def self.reset!(queue_name = nil)
+    if @queue && queue_name
+      @queue[queue_name] = []
+    else
+      @queue = Hash.new { |h, k| h[k] = [] }
+    end
   end
 
   # Returns an array of all the jobs that have been queued. Each
   # element is of the form +{:klass => klass, :args => args}+ where
   # +klass+ is the job's class and +args+ is an array of the arguments
   # passed to the job.
-  def self.queue(queue)
+  def self.queue(queue_name)
     self.reset! unless @queue
-    @queue[queue]
+    @queue[queue_name]
   end
 
   # Executes all jobs in all queues in an undefined order.
   def self.run!
     old_queue = @queue.dup
-    reset!
+    self.reset!
 
     old_queue.each do |k, v|
       while job = v.shift
@@ -30,10 +36,20 @@ module Resque
     end
   end
 
+  # Executes all jobs in the given queue in an undefined order.
+  def self.run_for!(queue_name)
+    jobs = self.queue(queue_name)
+    self.reset!(queue_name)
+
+    while job = jobs.shift
+      job[:klass].perform(*job[:args])
+    end
+  end
+
   # Returns the size of the given queue
-  def self.size(queue)
+  def self.size(queue_name)
     self.reset! unless @queue
-    @queue[queue].length
+    @queue[queue_name].length
   end
 
   # :nodoc: 

data/test/resque_unit_test.rb

@@ -15,7 +15,7 @@ class ResqueUnitTest < Test::Unit::TestCase
       assert_equal 1, Resque.queue(MediumPriorityJob.queue).length
     end
   end
-  
+
   context "A task that schedules a resque job" do
     setup do 
       Resque.enqueue(LowPriorityJob)
@@ -137,6 +137,30 @@ class ResqueUnitTest < Test::Unit::TestCase
     end
   end
 
+  context "A task in a different queue" do
+    setup do
+      Resque.enqueue(LowPriorityJob)
+      Resque.enqueue(HighPriorityJob)
+    end
+
+    should "add a LowPriorityJob" do
+      assert_queued(LowPriorityJob)
+    end
+
+    should "add a HighPriorityJob" do
+      assert_queued(HighPriorityJob)
+    end
+
+    context ", when Resque.run_for! is called," do 
+      should "run only tasks in the high priority queue" do
+        Resque.run_for!(Resque.queue_for(HighPriorityJob))
+  
+        assert_queued(LowPriorityJob)
+        assert_not_queued(HighPriorityJob)
+      end
+    end
+  end
+
   context "An assertion message" do
     context "of assert_queued" do
       should "include job class and queue content" do

data/test/sample_jobs.rb

@@ -20,6 +20,13 @@ class MediumPriorityJob
   end
 end
 
+class HighPriorityJob
+  @queue = :high
+
+  def self.perform
+  end
+end
+
 class JobWithArguments
   @queue = :medium
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: resque_unit
 version: !ruby/object:Gem::Version 
-  hash: 31
+  hash: 29
   prerelease: false
   segments: 
   - 0
   - 2
-  - 4
-  version: 0.2.4
+  - 5
+  version: 0.2.5
 platform: ruby
 authors: 
 - Justin Weiss