lib/resque_spec/scheduler.rb in resque_spec-0.6.2 vs lib/resque_spec/scheduler.rb in resque_spec-0.6.3

@@ -1,20 +1,25 @@
 require 'resque_spec'
 
 module ResqueSpec
-  def schedule_for(klass)
-    name = "#{queue_name(klass)}_scheduled"
-    queues[name]
-  end
-
   module SchedulerExt
     def enqueue_at(time, klass, *args)
-      if ResqueSpec.inline
-        klass.send(:perform, *args)
-      else
-        ResqueSpec.schedule_for(klass) << {:klass => klass.to_s, :time  => time, :args => args}
-      end
+      ResqueSpec.enqueue_at(time, klass, *args)
     end
+  end
+
+  def enqueue_at(time, klass, *args)
+    store(schedule_queue_name(klass), klass, { :class => klass.to_s, :time  => time, :args => args })
+  end
+
+  def schedule_for(klass)
+    queues[schedule_queue_name(klass)]
+  end
+
+  private
+
+  def schedule_queue_name(klass)
+    "#{queue_name(klass)}_scheduled"
   end
 end
 
 Resque.extend(ResqueSpec::SchedulerExt)