Support Net::HTTP::Persistent.write_timeout

lib/mechanize.rb

@@ -952,6 +952,21 @@ def read_timeout= read_timeout
     @agent.read_timeout = read_timeout
   end
 
+  ##
+  # Length of time to wait for data to be sent to the server
+
+  def write_timeout
+    @agent.write_timeout
+  end
+
+  ##
+  # Sets the timeout for each chunk of data to be sent to the server to
+  # +write_timeout+.  A single request may write many chunks of data.
+
+  def write_timeout= write_timeout
+    @agent.write_timeout = write_timeout
+  end
+
   ##
   # Controls how mechanize deals with redirects.  The following values are
   # allowed:

lib/mechanize/http/agent.rb

@@ -106,6 +106,9 @@ class Mechanize::HTTP::Agent
   # Length of time to attempt to read data from the server
   attr_accessor  :read_timeout
 
+  # Length of time to attempt to write data to the server
+  attr_accessor  :write_timeout
+
   # :section:
 
   # The cookies for this agent
@@ -161,6 +164,7 @@ def initialize(connection_name = 'mechanize')
     @robots_mutex             = Mutex.new
     @user_agent               = nil
     @webrobots                = nil
+    @write_timeout            = nil
 
     # HTTP Authentication
     @auth_store           = Mechanize::HTTP::AuthStore.new
@@ -274,6 +278,9 @@ def fetch uri, method = :get, headers = {}, params = [],
     if @read_timeout && connection.respond_to?(:read_timeout=)
       connection.read_timeout = @read_timeout
     end
+    if @write_timeout && connection.respond_to?(:write_timeout=)
+      connection.write_timeout = @write_timeout
+    end
 
     request_log request
 

test/test_mechanize.rb

@@ -1019,6 +1019,12 @@ def test_read_timeout_equals
     assert_equal 5, @mech.read_timeout
   end
 
+  def test_write_timeout_equals
+    @mech.write_timeout = 7
+
+    assert_equal 7, @mech.write_timeout
+  end
+
   def test_timeouts_for_file_connection
     uri = URI.parse "file://#{File.expand_path __FILE__}"
     @mech.read_timeout = 5