lib/resque_spec/ext.rb in resque_spec-0.6.3 vs lib/resque_spec/ext.rb in resque_spec-0.6.4

@@ -17,6 +17,24 @@
       ResqueSpec.dequeue(queue, klass, *args)
 
       old_count - ResqueSpec.queues[queue].size
     end
   end
+
+  def enqueue(klass, *args)
+    if ResqueSpec.inline
+      run_after_enqueue(klass, *args)
+      Job.create(queue_from_class(klass), klass, *args)
+    else
+      Job.create(queue_from_class(klass), klass, *args)
+      run_after_enqueue(klass, *args)
+    end
+  end
+
+  private
+
+  def run_after_enqueue(klass, *args)
+    Plugin.after_enqueue_hooks(klass).each do |hook|
+      klass.send(hook, *args)
+    end
+  end
 end