kotovalexarian-likes-github/ruby--ruby
1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
Code Releases Activity

Add max option to TestProtocol#create_mockio

Browse source 
* 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.
This commit is contained in:
Nobuyoshi Nakada 2019-06-29 17:52:57 +09:00
parent a88107c44d
commit 7ba14b029c
No known key found for this signature in database
GPG key ID: 4BC7D6DF58D8DF60
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
test/net/protocol/test_protocol.rb
View file

@ -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