blob: 1746f480c9be79787cb7cf85f8fce96514d2bd69 (
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe ObjectStorage::MigrateUploadsWorker do
let(:project) { create(:project, :with_avatar) }
let(:uploads) { Upload.all }
def perform(uploads, store = ObjectStorage::Store::REMOTE)
described_class.new.perform(uploads.ids, store)
rescue ObjectStorage::MigrateUploadsWorker::Report::MigrationFailures
# swallow
end
before do
stub_uploads_object_storage(AvatarUploader)
stub_uploads_object_storage(FileUploader)
FileUploader.new(project).store!(fixture_file_upload('spec/fixtures/doc_sample.txt'))
end
describe '#perform' do
it 'migrates files to remote storage' do
expect(Gitlab::AppLogger).to receive(:info).with(%r{Migrated 2/2 files})
perform(uploads)
expect(Upload.where(store: ObjectStorage::Store::LOCAL).count).to eq(0)
expect(Upload.where(store: ObjectStorage::Store::REMOTE).count).to eq(2)
end
context 'reversed' do
before do
perform(uploads)
end
it 'migrates files to local storage' do
expect(Upload.where(store: ObjectStorage::Store::REMOTE).count).to eq(2)
perform(uploads, ObjectStorage::Store::LOCAL)
expect(Upload.where(store: ObjectStorage::Store::LOCAL).count).to eq(2)
expect(Upload.where(store: ObjectStorage::Store::REMOTE).count).to eq(0)
end
end
context 'migration is unsuccessful' do
before do
allow_any_instance_of(ObjectStorage::Concern)
.to receive(:migrate!).and_raise(CarrierWave::UploadError, 'I am a teapot.')
end
it 'does not migrate files to remote storage' do
expect(Gitlab::AppLogger).to receive(:warn).with(/Error .* I am a teapot/)
perform(uploads)
expect(Upload.where(store: ObjectStorage::Store::LOCAL).count).to eq(2)
expect(Upload.where(store: ObjectStorage::Store::REMOTE).count).to eq(0)
end
end
describe "limits N+1 queries" do
it "to N*5" do
query_count = ActiveRecord::QueryRecorder.new { perform(uploads) }
create(:project, :with_avatar)
expect { perform(Upload.all) }.not_to exceed_query_limit(query_count).with_threshold(5)
end
end
it 'handles legacy argument format' do
described_class.new.perform(uploads.ids, 'Project', :avatar, ObjectStorage::Store::REMOTE)
expect(Upload.where(store: ObjectStorage::Store::LOCAL).count).to eq(0)
expect(Upload.where(store: ObjectStorage::Store::REMOTE).count).to eq(2)
end
it 'logs an error when number of arguments is incorrect' do
expect(Gitlab::AppLogger).to receive(:warn).with(/Job has wrong arguments format/)
described_class.new.perform(uploads.ids, 'Project', ObjectStorage::Store::REMOTE)
end
end
end
|