Specs FTW.

freels · Jul 22, 2010 · 2e885cb · 2e885cb
1 parent c31cffd
commit 2e885cb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/kestrel/client/reliable.rb b/lib/kestrel/client/reliable.rb
@@ -38,7 +38,7 @@ def get(key, opts = false)
       end
 
       def current_try
-        @job ? 1 : @job.retries + 1
+        @job ? @job.retries + 1 : 1
       end
 
       # Enqueues the current job on the error queue for later

diff --git a/spec/kestrel/client/reliable_spec.rb b/spec/kestrel/client/reliable_spec.rb
@@ -53,6 +53,16 @@
         @kestrel.get(@queue).should be_nil
       end
 
+      it "returns the payload of a RetryableJob" do
+        stub(@kestrel).rand { 0 }
+        mock(@raw_kestrel_client).get(@queue + "_errors", anything) do
+          Kestrel::Client::Reliable::RetryableJob.new(1, :mcmuffin)
+        end
+
+        @kestrel.get(@queue).should == :mcmuffin
+        @kestrel.current_try.should == 2
+      end
+
     end
 
     describe "#abort" do