Fix accessing individual files when artifacts are on object storage

changelogs/unreleased-ee/fix-accessing-files-on-object-storage.yml

+---
+title: Fix accessing individual files on Object Storage
+merge_request:
+author:

lib/gitlab/workhorse.rb

@@ -123,8 +123,16 @@ module Gitlab
       end
 
       def send_artifacts_entry(build, entry)
+        file = build.artifacts_file
+        archive = 
+          if file.file_storage?
+            file.path
+          else
+            file.url
+          end
+
         params = {
-          'Archive' => build.artifacts_file.path,
+          'Archive' => archive,
           'Entry' => Base64.encode64(entry.path)
         }
 

spec/controllers/projects/artifacts_controller_spec.rb

@@ -65,26 +65,60 @@ describe Projects::ArtifactsController do
   end
 
   describe 'GET raw' do
+    subject { get(:raw, namespace_id: project.namespace, project_id: project, job_id: job, path: path) }
+
     context 'when the file exists' do
-      it 'serves the file using workhorse' do
-        get :raw, namespace_id: project.namespace, project_id: project, job_id: job, path: 'ci_artifacts.txt'
+      let(:path) { 'ci_artifacts.txt' }
+      let(:job) { create(:ci_build, :success, :artifacts, pipeline: pipeline, artifacts_file_store: store, artifacts_metadata_store: store) }
+
+      shared_examples 'a valid file' do
+        it 'serves the file using workhorse' do
+          subject
+
+          expect(send_data).to start_with('artifacts-entry:')
+
+          expect(params.keys).to eq(%w(Archive Entry))
+          expect(params['Archive']).to start_with(archive_path)
+          expect(params['Archive']).to end_with('build_artifacts.zip')
+          expect(params['Entry']).to eq(Base64.encode64('ci_artifacts.txt'))
+        end
+
+        def send_data
+          response.headers[Gitlab::Workhorse::SEND_DATA_HEADER]
+        end
 
-        send_data = response.headers[Gitlab::Workhorse::SEND_DATA_HEADER]
+        def params
+          @params ||= begin
+            base64_params = send_data.sub(/\Aartifacts\-entry:/, '')
+            JSON.parse(Base64.urlsafe_decode64(base64_params))
+          end
+        end
+      end
 
-        expect(send_data).to start_with('artifacts-entry:')
+      context 'when using local file storage' do
+        it_behaves_like 'a valid file' do
+          let(:store) { ObjectStoreUploader::LOCAL_STORE }
+          let(:archive_path) { ArtifactUploader.local_artifacts_store }
+        end
+      end
 
-        base64_params = send_data.sub(/\Aartifacts\-entry:/, '')
-        params = JSON.parse(Base64.urlsafe_decode64(base64_params))
+      context 'when using remote file storage' do
+        before do
+          stub_artifacts_object_storage
+        end
 
-        expect(params.keys).to eq(%w(Archive Entry))
-        expect(params['Archive']).to end_with('build_artifacts.zip')
-        expect(params['Entry']).to eq(Base64.encode64('ci_artifacts.txt'))
+        it_behaves_like 'a valid file' do
+          let(:store) { ObjectStoreUploader::REMOTE_STORE }
+          let(:archive_path) { 'https://' }
+        end
       end
     end
 
     context 'when the file does not exist' do
+      let(:path) { 'unknown' }
+    
       it 'responds Not Found' do
-        get :raw, namespace_id: project.namespace, project_id: project, job_id: job, path: 'unknown'
+        subject
 
         expect(response).to be_not_found
       end