31 lines
1.1 KiB
Ruby
31 lines
1.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# Construct an `uploader` variable that is configured to `check_upload_type`
|
|
# with `mime_types` and `extensions`.
|
|
# @param uploader [CarrierWave::Uploader::Base] uploader with extension_whitelist method.
|
|
RSpec.shared_context 'ignore extension allowlist check' do
|
|
before do
|
|
allow(uploader).to receive(:extension_whitelist).and_return(nil)
|
|
end
|
|
end
|
|
|
|
# This works with a content_type_whitelist and content_type_blacklist type check.
|
|
# @param mime_type [String] mime type to forcibly detect.
|
|
RSpec.shared_context 'force content type detection to mime_type' do
|
|
before do
|
|
allow(Gitlab::Utils::MimeType).to receive(:from_io).and_return(mime_type)
|
|
end
|
|
end
|
|
|
|
def mock_upload(success = true)
|
|
allow(UploadService).to receive(:new).with(project, file).and_return(upload_service)
|
|
|
|
if success
|
|
allow(upload_service).to receive(:execute).and_return(uploader)
|
|
allow(uploader).to receive(:upload).and_return(upload)
|
|
allow(upload).to receive(:id).and_return(upload_id)
|
|
else
|
|
allow(upload_service).to receive(:execute).and_return(nil)
|
|
end
|
|
end
|