diff options
-rw-r--r-- | ChangeLog | 6 | ||||
-rw-r--r-- | ext/openssl/lib/openssl/buffering.rb | 2 | ||||
-rw-r--r-- | test/openssl/test_buffering.rb | 66 |
3 files changed, 74 insertions, 0 deletions
@@ -1,3 +1,9 @@ +Wed Mar 16 15:06:21 2011 Eric Hodel <drbrain@segment7.net> + + * ext/openssl/lib/openssl/buffering.rb (module OpenSSL): #flush should + not change sync mode on exception. + * test/openssl/test_buffering.rb: added + Wed Mar 16 13:45:28 2011 Eric Hodel <drbrain@segment7.net> * ext/openss/lib/openssl/bufering.rb: de-nest Buffering module diff --git a/ext/openssl/lib/openssl/buffering.rb b/ext/openssl/lib/openssl/buffering.rb index a9c60fb512..2fb33749e3 100644 --- a/ext/openssl/lib/openssl/buffering.rb +++ b/ext/openssl/lib/openssl/buffering.rb @@ -436,6 +436,8 @@ module OpenSSL::Buffering osync = @sync @sync = true do_write "" + return self + ensure @sync = osync end diff --git a/test/openssl/test_buffering.rb b/test/openssl/test_buffering.rb new file mode 100644 index 0000000000..49335de8d2 --- /dev/null +++ b/test/openssl/test_buffering.rb @@ -0,0 +1,66 @@ +require_relative 'utils' +require 'stringio' +require 'minitest/unit' + +class OpenSSL::TestBuffering < MiniTest::Unit::TestCase + + class IO + include OpenSSL::Buffering + + attr_accessor :sync + + def initialize + @io = StringIO.new + + super + + @sync = false + end + + def string + @io.string + end + + def sysread *a + @io.sysread *a + end + + def syswrite *a + @io.syswrite *a + end + end + + def setup + @io = IO.new + end + + def test_flush + @io.write 'a' + + refute @io.sync + assert_empty @io.string + + assert_equal @io, @io.flush + + refute @io.sync + assert_equal 'a', @io.string + end + + def test_flush_error + @io.write 'a' + + refute @io.sync + assert_empty @io.string + + def @io.syswrite *a + raise SystemCallError, 'fail' + end + + assert_raises SystemCallError do + @io.flush + end + + refute @io.sync, 'sync must not change' + end + +end if defined?(OpenSSL) |