Fixed missing StringIO exception in download! [Toby Bryans, Delano Mandelbaum]

Author: delano

CHANGELOG.rdoc

@@ -1,3 +1,9 @@
+
+=== 2.0.5 / 19 Aug 2010
+
+* Fixed missing StringIO exception in download! [Toby Bryans, Delano Mandelbaum]
+
+
 === 2.0.4 / 23 Nov 2009
 
 * Fixed frozen string issue in StatusException.message [appoxy]

lib/net/sftp/session.rb

@@ -120,6 +120,7 @@ def download(remote, local, options={}, &block)
       # and returns the result as a string; otherwise, returns the
       # Net::SFTP::Operations::Download instance.
       def download!(remote, local=nil, options={}, &block)
+        require 'stringio' unless defined?(StringIO)
         destination = local || StringIO.new
         result = download(remote, destination, options, &block).wait
         local ? result : destination.string

test/common.rb

@@ -1,5 +1,6 @@
 require 'test/unit'
 require 'mocha'
+require 'stringio'
 
 begin
   require 'net/ssh'

test/test_download.rb

@@ -141,7 +141,7 @@ def prepare_large_file_download(local, remote, text)
         channel.sends_packet(FXP_CLOSE, :long, 12, :string, "handle")
         channel.gets_packet(FXP_STATUS, :long, 12, :long, 0)
       end
-
+      
       file = StringIO.new
       File.stubs(:open).with(local, "wb").returns(file)