55 lines
1.2 KiB
Ruby
55 lines
1.2 KiB
Ruby
|
# rubocop:disable Style/FrozenStringLiteralComment
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
RSpec.describe Gitlab::BufferedIo do
|
||
|
|
describe '#readuntil' do
|
||
|
|
let(:never_ending_tcp_socket) do
|
||
|
|
Class.new do
|
||
|
|
def initialize(*_)
|
||
|
|
@read_counter = 0
|
||
|
|
end
|
||
|
|
|
||
|
|
def setsockopt(*_); end
|
||
|
|
|
||
|
|
def closed?
|
||
|
|
false
|
||
|
|
end
|
||
|
|
|
||
|
|
def close
|
||
|
|
true
|
||
|
|
end
|
||
|
|
|
||
|
|
def to_io
|
||
|
|
StringIO.new('Hello World!')
|
||
|
|
end
|
||
|
|
|
||
|
|
def write_nonblock(data, *_)
|
||
|
|
data.size
|
||
|
|
end
|
||
|
|
|
||
|
|
def read_nonblock(buffer_size, *_)
|
||
|
|
sleep 0.01
|
||
|
|
@read_counter += 1
|
||
|
|
|
||
|
|
raise 'Test did not raise HeaderReadTimeout' if @read_counter > 10
|
||
|
|
|
||
|
|
'H' * buffer_size
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
before do
|
||
|
|
stub_const('Gitlab::BufferedIo::HEADER_READ_TIMEOUT', 0.1)
|
||
|
|
end
|
||
|
|
|
||
|
|
subject(:readuntil) do
|
||
|
|
Gitlab::BufferedIo.new(never_ending_tcp_socket.new).readuntil('a')
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'raises a timeout error' do
|
||
|
|
expect { readuntil }.to raise_error(Gitlab::HTTP::HeaderReadTimeout, /Request timed out after reading headers for 0\.[0-9]+ seconds/)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
# rubocop:enable Style/FrozenStringLiteralComment
|