53 lines
1.4 KiB
Ruby
53 lines
1.4 KiB
Ruby
|
require 'rake_helper'
|
||
|
|
|
||
|
|
describe 'gitlab:storage rake tasks' do
|
||
|
|
before do
|
||
|
|
Rake.application.rake_require 'tasks/gitlab/storage'
|
||
|
|
|
||
|
|
stub_warn_user_is_not_gitlab
|
||
|
|
end
|
||
|
|
|
||
|
|
describe 'migrate_to_hashed rake task' do
|
||
|
|
context '0 legacy projects' do
|
||
|
|
it 'does nothing' do
|
||
|
|
expect(StorageMigratorWorker).not_to receive(:perform_async)
|
||
|
|
|
||
|
|
run_rake_task('gitlab:storage:migrate_to_hashed')
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context '5 legacy projects' do
|
||
|
|
let(:projects) { create_list(:project, 5, storage_version: 0) }
|
||
|
|
|
||
|
|
context 'in batches of 1' do
|
||
|
|
before do
|
||
|
|
stub_env('BATCH' => 1)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'enqueues one StorageMigratorWorker per project' do
|
||
|
|
projects.each do |project|
|
||
|
|
expect(StorageMigratorWorker).to receive(:perform_async).with(project.id, project.id)
|
||
|
|
end
|
||
|
|
|
||
|
|
run_rake_task('gitlab:storage:migrate_to_hashed')
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'in batches of 2' do
|
||
|
|
before do
|
||
|
|
stub_env('BATCH' => 2)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'enqueues one StorageMigratorWorker per 2 projects' do
|
||
|
|
projects.map(&:id).sort.each_slice(2) do |first, last|
|
||
|
|
last ||= first
|
||
|
|
expect(StorageMigratorWorker).to receive(:perform_async).with(first, last)
|
||
|
|
end
|
||
|
|
|
||
|
|
run_rake_task('gitlab:storage:migrate_to_hashed')
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|