summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/import_export/after_export_strategies/web_upload_strategy_spec.rb
blob: 7f2e0a4ee2c898e09d5c5e7b9fcea58abc3dc877 (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
require 'spec_helper'

describe Gitlab::ImportExport::AfterExportStrategies::WebUploadStrategy do
  let(:example_url) { 'http://www.example.com' }
  let(:strategy) { subject.new(url: example_url, http_method: 'post') }
  let!(:project) { create(:project, :with_export) }
  let!(:user) { build(:user) }

  subject { described_class }

  describe 'validations' do
    it 'only POST and PUT method allowed' do
      %w(POST post PUT put).each do |method|
        expect(subject.new(url: example_url, http_method: method)).to be_valid
      end

      expect(subject.new(url: example_url, http_method: 'whatever')).not_to be_valid
    end

    it 'onyl allow urls as upload urls' do
      expect(subject.new(url: example_url)).to be_valid
      expect(subject.new(url: 'whatever')).not_to be_valid
    end
  end

  describe '#execute' do
    context 'without object storage' do
      before do
        stub_feature_flags(import_export_object_storage: false)
      end

      it 'removes the exported project file after the upload' do
        allow(strategy).to receive(:send_file)
        allow(strategy).to receive(:handle_response_error)

        expect(project).to receive(:remove_exported_project_file)

        strategy.execute(user, project)
      end
    end

    context 'with object storage' do
      before do
        stub_feature_flags(import_export_object_storage: true)
      end

      it 'removes the exported project file after the upload' do
        allow(strategy).to receive(:send_file)
        allow(strategy).to receive(:handle_response_error)

        expect(project).to receive(:remove_exported_project_file)

        strategy.execute(user, project)
      end
    end
  end
end