1 files changed, 13 insertions, 0 deletions
diff --git a/spec/support/helpers/chunked_io_helpers.rb b/spec/support/helpers/chunked_io_helpers.rb
new file mode 100644
index 00000000000..278f577f3cb
--- /dev/null
+++ b/spec/support/helpers/chunked_io_helpers.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module ChunkedIOHelpers
+  def sample_trace_raw
+    @sample_trace_raw ||= File.read(expand_fixture_path('trace/sample_trace'))
+      .force_encoding(Encoding::BINARY)
+  end
+
+  def stub_buffer_size(size)
+    stub_const('Ci::BuildTraceChunk::CHUNK_SIZE', size)
+    stub_const('Gitlab::Ci::Trace::ChunkedIO::CHUNK_SIZE', size)
+  end
+end