diff options
author | Nobuyoshi Nakada <nobu@ruby-lang.org> | 2019-06-29 17:52:57 +0900 |
---|---|---|
committer | Nobuyoshi Nakada <nobu@ruby-lang.org> | 2019-06-29 17:52:57 +0900 |
commit | 7ba14b029c64c7f9ef080f5b22065b662145d5d5 (patch) | |
tree | a2fa6a1987d5322b06519d9f22ef2c8c428d428c /test | |
parent | a88107c44dd08b02ed1c1b5951882b070a1eab93 (diff) | |
download | ruby-7ba14b029c64c7f9ef080f5b22065b662145d5d5.tar.gz |
Add max option to TestProtocol#create_mockio
* test/net/protocol/test_protocol.rb (TestProtocol#create_mockio):
max option to limit maximum writable size at once, and unify
mock method in test_write0_multibyte.
Diffstat (limited to 'test')
-rw-r--r-- | test/net/protocol/test_protocol.rb | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/test/net/protocol/test_protocol.rb b/test/net/protocol/test_protocol.rb index 2ad3709ac5..d3dc2ccf4c 100644 --- a/test/net/protocol/test_protocol.rb +++ b/test/net/protocol/test_protocol.rb @@ -27,10 +27,11 @@ class TestProtocol < Test::Unit::TestCase end end - def create_mockio(capacity: 100) + def create_mockio(capacity: 100, max: nil) mockio = Object.new mockio.instance_variable_set(:@str, +'') mockio.instance_variable_set(:@capacity, capacity) + mockio.instance_variable_set(:@max, max) def mockio.string; @str; end def mockio.to_io; self; end def mockio.wait_writable(sec); sleep sec; false; end @@ -43,10 +44,11 @@ class TestProtocol < Test::Unit::TestCase end end len = 0 + max = @max ? [@capacity, @str.bytesize + @max].min : @capacity strs.each do |str| len1 = @str.bytesize - break if @capacity <= len1 - @str << str.byteslice(0, @capacity - @str.bytesize) + break if max <= len1 + @str << str.byteslice(0, max - @str.bytesize) len2 = @str.bytesize len += len2 - len1 end @@ -56,11 +58,7 @@ class TestProtocol < Test::Unit::TestCase end def test_write0_multibyte - mockio = create_mockio(capacity: 1) - def mockio.write_nonblock(str, *strs, **kw) - @str << str.byteslice(0, 1) - 1 - end + mockio = create_mockio(max: 1) io = Net::BufferedIO.new(mockio) assert_equal(3, io.write("\u3042")) end |