summaryrefslogtreecommitdiff
path: root/spec/support/shared_contexts/upload_type_check_shared_context.rb
blob: d29c498fd15318518af9e17a63ff6b6bccdac022 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# frozen_string_literal: true

# Construct an `uploader` variable that is configured to `check_upload_type`
# with `mime_types` and `extensions`.
RSpec.shared_context 'uploader with type check' do
  let(:uploader_class) do
    Class.new(GitlabUploader) do
      include UploadTypeCheck::Concern
      storage :file
    end
  end

  let(:mime_types) { nil }
  let(:extensions) { nil }
  let(:uploader) do
    uploader_class.class_exec(mime_types, extensions) do |mime_types, extensions|
      check_upload_type mime_types: mime_types, extensions: extensions
    end
    uploader_class.new(build_stubbed(:user))
  end
end

RSpec.shared_context 'stubbed MimeMagic mime type detection' do
  let(:mime_type) { '' }
  let(:magic_mime) { mime_type }
  let(:ext_mime) { mime_type }
  before do
    magic_mime_obj = MimeMagic.new(magic_mime)
    ext_mime_obj = MimeMagic.new(ext_mime)
    allow(MimeMagic).to receive(:by_magic).with(anything).and_return(magic_mime_obj)
    allow(MimeMagic).to receive(:by_path).with(anything).and_return(ext_mime_obj)
  end
end