1 files changed, 13 insertions, 1 deletions
diff --git a/spec/models/ci/job_artifact_spec.rb b/spec/models/ci/job_artifact_spec.rb
index b9cac6c3f99..b996bf84529 100644
--- a/spec/models/ci/job_artifact_spec.rb
+++ b/spec/models/ci/job_artifact_spec.rb
@@ -128,6 +128,18 @@ RSpec.describe Ci::JobArtifact do
     end
   end
 
+  describe '.file_types_for_report' do
+    it 'returns the report file types for the report type' do
+      expect(described_class.file_types_for_report(:test)).to match_array(%w[junit])
+    end
+
+    context 'when given an unrecognized report type' do
+      it 'raises error' do
+        expect { described_class.file_types_for_report(:blah) }.to raise_error(KeyError, /blah/)
+      end
+    end
+  end
+
   describe '.associated_file_types_for' do
     using RSpec::Parameterized::TableSyntax
 
@@ -193,7 +205,7 @@ RSpec.describe Ci::JobArtifact do
       it { is_expected.to be_truthy }
 
       context 'when the job does have archived trace' do
-        let!(:artifact) { }
+        let!(:artifact) {}
 
         it { is_expected.to be_falsy }
       end