mperham#113 expose pool from Wrapper

lib/connection_pool.rb

@@ -109,12 +109,16 @@ def available
   private
 
   class Wrapper < ::BasicObject
-    METHODS = [:with, :pool_shutdown]
+    METHODS = [:with, :pool_shutdown, :pool]
 
     def initialize(options = {}, &block)
       @pool = options.fetch(:pool) { ::ConnectionPool.new(options, &block) }
     end
 
+    def pool
+      @pool
+    end
+
     def with(&block)
       @pool.with(&block)
     end

test/test_connection_pool.rb

@@ -474,6 +474,11 @@ def test_shutdown_is_executed_for_all_connections_in_wrapped_pool
     assert_equal [["shutdown"]] * 3, recorders.map { |r| r.calls }
   end
 
+  def test_wrapper_pool
+    wrapper = ConnectionPool::Wrapper.new { NetworkConnection.new }
+    assert_equal ConnectionPool, wrapper.pool.class
+  end
+
   def test_wrapper_method_missing
     wrapper = ConnectionPool::Wrapper.new { NetworkConnection.new }