138 lines
4.0 KiB
Ruby
138 lines
4.0 KiB
Ruby
require 'spec_helper'
|
|
|
|
describe Gitlab::HashedStorage::Migrator do
|
|
describe '#bulk_schedule' do
|
|
it 'schedules job to StorageMigratorWorker' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.bulk_schedule(start: 1, finish: 5) }.to change(HashedStorage::MigratorWorker.jobs, :size).by(1)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#bulk_migrate' do
|
|
let(:projects) { create_list(:project, 2, :legacy_storage) }
|
|
let(:ids) { projects.map(&:id) }
|
|
|
|
it 'enqueue jobs to ProjectMigrateHashedStorageWorker' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.bulk_migrate(start: ids.min, finish: ids.max) }.to change(ProjectMigrateHashedStorageWorker.jobs, :size).by(2)
|
|
end
|
|
end
|
|
|
|
it 'rescues and log exceptions' do
|
|
allow_any_instance_of(Project).to receive(:migrate_to_hashed_storage!).and_raise(StandardError)
|
|
expect { subject.bulk_migrate(start: ids.min, finish: ids.max) }.not_to raise_error
|
|
end
|
|
|
|
it 'delegates each project in specified range to #migrate' do
|
|
projects.each do |project|
|
|
expect(subject).to receive(:migrate).with(project)
|
|
end
|
|
|
|
subject.bulk_migrate(start: ids.min, finish: ids.max)
|
|
end
|
|
|
|
it 'has migrated projects set as writable' do
|
|
perform_enqueued_jobs do
|
|
subject.bulk_migrate(start: ids.min, finish: ids.max)
|
|
end
|
|
|
|
projects.each do |project|
|
|
expect(project.reload.repository_read_only?).to be_falsey
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#migrate' do
|
|
let(:project) { create(:project, :legacy_storage, :empty_repo) }
|
|
|
|
it 'enqueues project migration job' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.migrate(project) }.to change(ProjectMigrateHashedStorageWorker.jobs, :size).by(1)
|
|
end
|
|
end
|
|
|
|
it 'rescues and log exceptions' do
|
|
allow(project).to receive(:migrate_to_hashed_storage!).and_raise(StandardError)
|
|
|
|
expect { subject.migrate(project) }.not_to raise_error
|
|
end
|
|
|
|
it 'migrates project storage' do
|
|
perform_enqueued_jobs do
|
|
subject.migrate(project)
|
|
end
|
|
|
|
expect(project.reload.hashed_storage?(:attachments)).to be_truthy
|
|
end
|
|
|
|
it 'has migrated project set as writable' do
|
|
perform_enqueued_jobs do
|
|
subject.migrate(project)
|
|
end
|
|
|
|
expect(project.reload.repository_read_only?).to be_falsey
|
|
end
|
|
|
|
context 'when project is already on hashed storage' do
|
|
let(:project) { create(:project, :empty_repo) }
|
|
|
|
it 'doesnt enqueue any migration job' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.migrate(project) }.not_to change(ProjectMigrateHashedStorageWorker.jobs, :size)
|
|
end
|
|
end
|
|
|
|
it 'returns false' do
|
|
expect(subject.migrate(project)).to be_falsey
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#rollback' do
|
|
let(:project) { create(:project, :empty_repo) }
|
|
|
|
it 'enqueues project rollback job' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.rollback(project) }.to change(ProjectRollbackHashedStorageWorker.jobs, :size).by(1)
|
|
end
|
|
end
|
|
|
|
it 'rescues and log exceptions' do
|
|
allow(project).to receive(:rollback_to_hashed_storage!).and_raise(StandardError)
|
|
|
|
expect { subject.rollback(project) }.not_to raise_error
|
|
end
|
|
|
|
it 'rolls-back project storage' do
|
|
perform_enqueued_jobs do
|
|
subject.rollback(project)
|
|
end
|
|
|
|
expect(project.reload.legacy_storage?).to be_truthy
|
|
end
|
|
|
|
it 'has rolled-back project set as writable' do
|
|
perform_enqueued_jobs do
|
|
subject.rollback(project)
|
|
end
|
|
|
|
expect(project.reload.repository_read_only?).to be_falsey
|
|
end
|
|
|
|
context 'when project is already on legacy storage' do
|
|
let(:project) { create(:project, :legacy_storage, :empty_repo) }
|
|
|
|
it 'doesnt enqueue any rollback job' do
|
|
Sidekiq::Testing.fake! do
|
|
expect { subject.rollback(project) }.not_to change(ProjectRollbackHashedStorageWorker.jobs, :size)
|
|
end
|
|
end
|
|
|
|
it 'returns false' do
|
|
expect(subject.rollback(project)).to be_falsey
|
|
end
|
|
end
|
|
end
|
|
end
|