gitlab-org--gitlab-foss/spec/support/chunked_io/chunked_io_helpers.rb

module ChunkedIOHelpers
  def fill_trace_to_chunks(data)
    stream = Gitlab::Ci::Trace::ChunkedFile::ChunkedIO.new(job_id, data.length, 'wb')
    stream.write(data)
    stream.close
  end

  def sample_trace_raw
    if chunk_store == Gitlab::Ci::Trace::ChunkedFile::ChunkStore::Redis
      File.read(expand_fixture_path('trace/sample_trace'))
    else
      '01234567' * 32
    end
  end

  def sample_trace_size
    sample_trace_raw.length
  end

  def stub_chunk_store_get_failed
    allow_any_instance_of(chunk_store).to receive(:get).and_return(nil)
  end

  def set_smaller_buffer_size_than(file_size)
    blocks = (file_size / 128)
    new_size = (blocks / 2) * 128
    stub_const("Gitlab::Ci::Trace::ChunkedFile::ChunkedIO::BUFFER_SIZE", new_size)
  end

  def set_larger_buffer_size_than(file_size)
    blocks = (file_size / 128)
    new_size = (blocks * 2) * 128
    stub_const("Gitlab::Ci::Trace::ChunkedFile::ChunkedIO::BUFFER_SIZE", new_size)
  end
end