alloc8 0.0.2 → 0.0.3

data/lib/alloc8.rb

@@ -60,9 +60,12 @@ module Alloc8
     # block to allocate a resource
     def self.with_resource(klass, host, port, db=nil)
       a = Alloc8::Tor.new host, port, db
-      res = a.acquire(klass)
-      yield res
-      a.return(klass, res)
+      begin
+        res = a.acquire(klass)
+        yield res
+      ensure
+        a.return(klass, res)
+      end
     end
 
     private

data/lib/alloc8/version.rb

@@ -1,3 +1,3 @@
 module Alloc8
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/alloc8_spec.rb

@@ -52,10 +52,19 @@ describe Alloc8 do
       lambda { @allocator.return(KIND, "foo") }.should raise_error
     end
 
-    it "can acquire with a block" do
+    it "can acquire with a block, and returns correctly" do
       Alloc8::Tor.with_resource(KIND, HOST, PORT) do |res|
         res.should == "a"
+        @allocator.available(KIND).should eq([])
       end
+      @allocator.list(KIND).should eq(["a"])
+      @allocator.available(KIND).should eq(["a"])
+    end
+
+    it "can acquire with a block, and handle an exception" do
+      lambda { Alloc8::Tor.with_resource(KIND, HOST, PORT) { raise Exception.new("test") } }.should raise_error
+      @allocator.list(KIND).should eq(["a"])
+      @allocator.available(KIND).should eq(["a"])
     end
 
     it "can't acquire a second resource" do

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: alloc8
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Ariel Salomon
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-05-22 00:00:00 -07:00
+date: 2011-05-23 00:00:00 -07:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency