1 files changed, 82 insertions, 0 deletions
diff --git a/spec/models/ci/pipeline_artifact_spec.rb b/spec/models/ci/pipeline_artifact_spec.rb
new file mode 100644
index 00000000000..9d63d74a6cc
--- /dev/null
+++ b/spec/models/ci/pipeline_artifact_spec.rb
@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Ci::PipelineArtifact, type: :model do
+  let_it_be(:coverage_report) { create(:ci_pipeline_artifact) }
+
+  describe 'associations' do
+    it { is_expected.to belong_to(:pipeline) }
+    it { is_expected.to belong_to(:project) }
+  end
+
+  it_behaves_like 'having unique enum values'
+
+  describe 'validations' do
+    it { is_expected.to validate_presence_of(:pipeline) }
+    it { is_expected.to validate_presence_of(:project) }
+    it { is_expected.to validate_presence_of(:file_type) }
+    it { is_expected.to validate_presence_of(:file_format) }
+    it { is_expected.to validate_presence_of(:size) }
+    it { is_expected.to validate_presence_of(:file) }
+
+    context 'when attributes are valid' do
+      it 'returns no errors' do
+        expect(coverage_report).to be_valid
+      end
+    end
+
+    context 'when file_store is invalid' do
+      it 'returns errors' do
+        coverage_report.file_store = 0
+
+        expect(coverage_report).to be_invalid
+        expect(coverage_report.errors.full_messages).to eq(["File store is not included in the list"])
+      end
+    end
+
+    context 'when size is over 10 megabytes' do
+      it 'returns errors' do
+        coverage_report.size = 11.megabytes
+
+        expect(coverage_report).to be_invalid
+      end
+    end
+  end
+
+  describe '#set_size' do
+    subject { create(:ci_pipeline_artifact) }
+
+    context 'when file is being created' do
+      it 'sets the size' do
+        expect(subject.size).to eq(85)
+      end
+    end
+
+    context 'when file is being updated' do
+      it 'updates the size' do
+        subject.update!(file: fixture_file_upload('spec/fixtures/dk.png'))
+
+        expect(subject.size).to eq(1062)
+      end
+    end
+  end
+
+  describe 'file is being stored' do
+    subject { create(:ci_pipeline_artifact) }
+
+    context 'when existing object has local store' do
+      it_behaves_like 'mounted file in local store'
+    end
+
+    context 'when direct upload is enabled' do
+      before do
+        stub_artifacts_object_storage(Ci::PipelineArtifactUploader, direct_upload: true)
+      end
+
+      context 'when file is stored' do
+        it_behaves_like 'mounted file in object store'
+      end
+    end
+  end
+end