Update validator

config/initializers/direct_upload_support.rb

-SUPPORTED_DIRECT_UPLOAD_PROVIDERS = %w(Google AWS).freeze
-def verify_provider_support!(object_store)
-  return unless object_store.enabled
-  return unless object_store.direct_upload
-  return if SUPPORTED_DIRECT_UPLOAD_PROVIDERS.include?(object_store.connection&.provider.to_s)
-  raise "Only #{SUPPORTED_DIRECT_UPLOAD_PROVIDERS.join(',')} are supported as a object storage provider when 'direct_upload' is used"
+class DirectUploadsValidator
+  SUPPORTED_DIRECT_UPLOAD_PROVIDERS = %w(Google AWS).freeze
+  ValidationError = Class.new(StandardError)
+  def verify!(object_store)
+    return unless object_store.enabled
+    return unless object_store.direct_upload
+    return if SUPPORTED_DIRECT_UPLOAD_PROVIDERS.include?(object_store.connection&.provider.to_s)
+
+    raise ValidationError, "Only #{SUPPORTED_DIRECT_UPLOAD_PROVIDERS.join(',')} are supported as a object storage provider when 'direct_upload' is used"
+  end
 end
-verify_provider_support!(Gitlab.config.artifacts.object_store)
-verify_provider_support!(Gitlab.config.uploads.object_store)
-verify_provider_support!(Gitlab.config.lfs.object_store)
+DirectUploadsValidator.new.tap do |validator|
+  [Gitlab.config.artifacts, Gitlab.config.uploads, Gitlab.config.lfs].each do |uploader|
+    validator.verify!(uploader.object_store)
+  end
+end

lib/object_storage/direct_upload.rb

@@ -2,7 +2,7 @@ module ObjectStorage
   #
   # The DirectUpload c;ass generates a set of presigned URLs
   # that can be used to upload data to object storage from untrusted component: Workhorse, Runner?
-  # 
+  #
   # For Google it assumes that the platform supports variable Content-Length.
   #
   # For AWS it initiates Multipart Upload and presignes a set of part uploads.

spec/initializers/direct_upload_support_spec.rb

@@ -6,7 +6,7 @@ describe 'Direct upload support' do
   end
   where(:config_name) do
-    ['lfs', 'artifacts', 'uploads']
+    %w(lfs artifacts uploads)
   end
   with_them do
@@ -23,8 +23,9 @@ describe 'Direct upload support' do
     end
     before do
-      allow(Gitlab.config).to receive_messages(to_settings(
-        config_name => { object_store: object_store }))
+      allow(Gitlab.config).to receive_messages(to_settings(config_name => {
+        object_store: object_store
+      }))
     end
     context 'when object storage is enabled' do

spec/lib/object_storage/direct_upload_spec.rb

@@ -161,28 +161,4 @@ describe ObjectStorage::DirectUpload do
       end
     end
   end
-
-  describe '#get_url' do
-    # this method can only be tested with integration tests
-  end
-
-  describe '#delete_url' do
-    # this method can only be tested with integration tests
-  end
-
-  describe '#store_url' do
-    # this method can only be tested with integration tests
-  end
-
-  describe '#multipart_part_upload_url' do
-    # this method can only be tested with integration tests
-  end
-
-  describe '#multipart_complete_url' do
-    # this method can only be tested with integration tests
-  end
-
-  describe '#multipart_abort_url' do
-    # this method can only be tested with integration tests
-  end
 end

spec/support/helpers/stub_object_storage.rb

@@ -47,8 +47,8 @@ module StubObjectStorage
   end
   def stub_object_storage_multipart_init(endpoint, upload_id = "upload_id")
-    stub_request(:post, %r{\A#{endpoint}tmp/uploads/[a-z0-9-]*\?uploads\z}).
-      to_return status: 200, body: <<-EOS.strip_heredoc
+    stub_request(:post, %r{\A#{endpoint}tmp/uploads/[a-z0-9-]*\?uploads\z})
+      .to_return status: 200, body: <<-EOS.strip_heredoc
         <?xml version="1.0" encoding="UTF-8"?>
         <InitiateMultipartUploadResult xmlns="http://s3.amazonaws.com/doc/2006-03-01/">
           <Bucket>example-bucket</Bucket>