1 files changed, 101 insertions, 0 deletions
diff --git a/spec/lib/gitlab/ci/trace/archive_spec.rb b/spec/lib/gitlab/ci/trace/archive_spec.rb
new file mode 100644
index 00000000000..c9fc4e720c4
--- /dev/null
+++ b/spec/lib/gitlab/ci/trace/archive_spec.rb
@@ -0,0 +1,101 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Ci::Trace::Archive do
+  let_it_be(:job) { create(:ci_build, :success, :trace_live) }
+  let_it_be_with_reload(:trace_metadata) { create(:ci_build_trace_metadata, build: job) }
+  let_it_be(:src_checksum) do
+    job.trace.read { |stream| Digest::MD5.hexdigest(stream.raw) }
+  end
+
+  let(:metrics) { spy('metrics') }
+
+  describe '#execute' do
+    subject { described_class.new(job, trace_metadata, metrics) }
+
+    it 'computes and assigns checksum' do
+      Gitlab::Ci::Trace::ChunkedIO.new(job) do |stream|
+        expect { subject.execute!(stream) }.to change { Ci::JobArtifact.count }.by(1)
+      end
+
+      expect(trace_metadata.checksum).to eq(src_checksum)
+      expect(trace_metadata.trace_artifact).to eq(job.job_artifacts_trace)
+    end
+
+    context 'validating artifact checksum' do
+      let(:trace) { 'abc' }
+      let(:stream) { StringIO.new(trace, 'rb') }
+      let(:src_checksum) { Digest::MD5.hexdigest(trace) }
+
+      context 'when the object store is disabled' do
+        before do
+          stub_artifacts_object_storage(enabled: false)
+        end
+
+        it 'skips validation' do
+          subject.execute!(stream)
+
+          expect(trace_metadata.checksum).to eq(src_checksum)
+          expect(trace_metadata.remote_checksum).to be_nil
+          expect(metrics)
+            .not_to have_received(:increment_error_counter)
+            .with(type: :archive_invalid_checksum)
+        end
+      end
+
+      context 'with background_upload enabled' do
+        before do
+          stub_artifacts_object_storage(background_upload: true)
+        end
+
+        it 'skips validation' do
+          subject.execute!(stream)
+
+          expect(trace_metadata.checksum).to eq(src_checksum)
+          expect(trace_metadata.remote_checksum).to be_nil
+          expect(metrics)
+            .not_to have_received(:increment_error_counter)
+            .with(type: :archive_invalid_checksum)
+        end
+      end
+
+      context 'with direct_upload enabled' do
+        before do
+          stub_artifacts_object_storage(direct_upload: true)
+        end
+
+        it 'validates the archived trace' do
+          subject.execute!(stream)
+
+          expect(trace_metadata.checksum).to eq(src_checksum)
+          expect(trace_metadata.remote_checksum).to eq(src_checksum)
+          expect(metrics)
+            .not_to have_received(:increment_error_counter)
+            .with(type: :archive_invalid_checksum)
+        end
+
+        context 'when the checksum does not match' do
+          let(:invalid_remote_checksum) { SecureRandom.hex }
+
+          before do
+            expect(::Gitlab::Ci::Trace::RemoteChecksum)
+              .to receive(:new)
+              .with(an_instance_of(Ci::JobArtifact))
+              .and_return(double(md5_checksum: invalid_remote_checksum))
+          end
+
+          it 'validates the archived trace' do
+            subject.execute!(stream)
+
+            expect(trace_metadata.checksum).to eq(src_checksum)
+            expect(trace_metadata.remote_checksum).to eq(invalid_remote_checksum)
+            expect(metrics)
+              .to have_received(:increment_error_counter)
+              .with(type: :archive_invalid_checksum)
+          end
+        end
+      end
+    end
+  end
+end