module StubObjectStorage def stub_object_storage_uploader( config:, uploader:, remote_directory:, enabled: true, proxy_download: false, background_upload: false, direct_upload: false ) allow(config).to receive(:enabled) { enabled } allow(config).to receive(:proxy_download) { proxy_download } allow(config).to receive(:background_upload) { background_upload } allow(config).to receive(:direct_upload) { direct_upload } return unless enabled stub_object_storage(connection_params: uploader.object_store_credentials, remote_directory: remote_directory) end def stub_object_storage(connection_params:, remote_directory:) Fog.mock! ::Fog::Storage.new(connection_params).tap do |connection| connection.directories.create(key: remote_directory) # Cleanup remaining files connection.directories.each do |directory| directory.files.map(&:destroy) end rescue Excon::Error::Conflict end end def stub_artifacts_object_storage(**params) stub_object_storage_uploader(config: Gitlab.config.artifacts.object_store, uploader: JobArtifactUploader, remote_directory: 'artifacts', **params) end def stub_external_diffs_object_storage(uploader = described_class, **params) stub_object_storage_uploader(config: Gitlab.config.external_diffs.object_store, uploader: uploader, remote_directory: 'external_diffs', **params) end def stub_lfs_object_storage(**params) stub_object_storage_uploader(config: Gitlab.config.lfs.object_store, uploader: LfsObjectUploader, remote_directory: 'lfs-objects', **params) end def stub_uploads_object_storage(uploader = described_class, **params) stub_object_storage_uploader(config: Gitlab.config.uploads.object_store, uploader: uploader, remote_directory: 'uploads', **params) 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 example-bucket example-object #{upload_id} EOS end end require_relative '../../../ee/spec/support/helpers/ee/stub_object_storage' if Dir.exist?("#{__dir__}/../../../ee")