Introduce a way to fine tune socket options: Bunny::Session#configure…

…_socket

lib/bunny/session.rb

@@ -171,6 +171,16 @@ def threaded?
       @threaded
     end
 
+    def configure_socket(&block)
+      raise ArgumentError, "No block provided!" if block.nil?
+
+      if @transport
+        @transport.configure_socket(&block)
+      else
+        @socket_configurator = block
+      end
+    end
+
     # Starts the connection process.
     #
     # @see http://rubybunny.info/articles/getting_started.html
@@ -188,6 +198,10 @@ def start
         self.maybe_close_transport
         self.initialize_transport
 
+        if @socket_configurator
+          @transport.configure_socket(&@socket_configurator)
+        end
+
         self.init_connection
         self.open_connection
 

lib/bunny/transport.rb

@@ -77,6 +77,10 @@ def connect
       end
     end
 
+    def configure_socket(&block)
+      block.call(@socket)
+    end
+
 
     # Writes data to the socket. If read/write timeout was specified, Bunny::ClientTimeout will be raised
     # if the operation times out.

spec/higher_level_api/integration/connection_spec.rb

@@ -62,6 +62,17 @@
       Bunny.new
     end
 
+    it "provides a way to fine tune socket options" do
+      subject.configure_socket do |sock|
+        sock.should respond_to(:setsockopt)
+      end
+
+      subject.start
+      subject.configure_socket do |sock|
+        sock.should respond_to(:setsockopt)
+      end
+    end
+
     it "successfully negotiates the connection" do
       subject.start
       subject.should be_connected