2018-03-29 12:44:41 -04:00
|
|
|
module ChunkedIOHelpers
|
|
|
|
def fill_trace_to_chunks(data)
|
2018-04-02 03:52:18 -04:00
|
|
|
stream = described_class.new(job_id, nil, 'a+b')
|
2018-03-29 12:44:41 -04:00
|
|
|
stream.write(data)
|
|
|
|
stream.close
|
|
|
|
end
|
|
|
|
|
|
|
|
def sample_trace_raw
|
2018-04-02 10:11:48 -04:00
|
|
|
# ChunkStore::Database doesn't support appending, so the test data size has to be least common multiple
|
|
|
|
if chunk_stores.first == Gitlab::Ci::Trace::ChunkedFile::ChunkStore::Database
|
|
|
|
'01234567' * 32 # 256 bytes
|
2018-03-29 15:13:09 -04:00
|
|
|
else
|
2018-04-02 10:11:48 -04:00
|
|
|
File.read(expand_fixture_path('trace/sample_trace'))
|
2018-03-29 15:13:09 -04:00
|
|
|
end
|
2018-03-29 12:44:41 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def set_smaller_buffer_size_than(file_size)
|
|
|
|
blocks = (file_size / 128)
|
|
|
|
new_size = (blocks / 2) * 128
|
2018-04-01 00:34:29 -04:00
|
|
|
allow_any_instance_of(described_class).to receive(:buffer_size).and_return(new_size)
|
2018-03-29 12:44:41 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def set_larger_buffer_size_than(file_size)
|
|
|
|
blocks = (file_size / 128)
|
|
|
|
new_size = (blocks * 2) * 128
|
2018-04-01 00:34:29 -04:00
|
|
|
allow_any_instance_of(described_class).to receive(:buffer_size).and_return(new_size)
|
2018-03-29 12:44:41 -04:00
|
|
|
end
|
2018-04-02 10:11:48 -04:00
|
|
|
|
|
|
|
def set_half_buffer_size_of(file_size)
|
|
|
|
allow_any_instance_of(described_class).to receive(:buffer_size).and_return(file_size / 2)
|
|
|
|
end
|
2018-03-29 12:44:41 -04:00
|
|
|
end
|