add guard against releasing slots that haven't been reserved

Author: sampatbadhe

lib/kredis/types/slots.rb

@@ -19,18 +19,23 @@ def reserve
           release
         end
       else
-        if incr <= available
+        if available?
+          incr
           true
         else
-          release
           false
         end
       end
     end
   end
 
   def release
-    decr
+    if get.to_i.between?(1, available)
+      decr
+      true
+    else
+      false
+    end
   end
 
   def available?

test/types/slots_test.rb

@@ -26,6 +26,20 @@ class SlotsTest < ActiveSupport::TestCase
     assert @slots.available?
   end
 
+  test 'release when slots are reserved' do
+    assert_not @slots.release
+
+    3.times do
+      assert @slots.reserve
+    end
+
+    3.times do
+      assert @slots.release
+    end
+
+    assert_not @slots.release
+  end
+
   test "reserve with block" do
     assert @slots.reserve
     assert @slots.reserve
@@ -70,6 +84,17 @@ class SlotsTest < ActiveSupport::TestCase
     assert_not slot.available?
   end
 
+  test "release single slot when reserved" do
+    slot = Kredis.slot "myslot"
+
+    assert_not slot.release
+
+    assert slot.reserve
+    assert slot.release
+
+    assert_not slot.release
+  end
+
   test "failing open" do
     stub_redis_down(@slots) do
       assert_not @slots.available?