diff options
Diffstat (limited to 'spec/tasks/gitlab/uploads')
-rw-r--r-- | spec/tasks/gitlab/uploads/migrate_rake_spec.rb | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/spec/tasks/gitlab/uploads/migrate_rake_spec.rb b/spec/tasks/gitlab/uploads/migrate_rake_spec.rb index b778d26060d..8f93ffa0edd 100644 --- a/spec/tasks/gitlab/uploads/migrate_rake_spec.rb +++ b/spec/tasks/gitlab/uploads/migrate_rake_spec.rb @@ -20,9 +20,20 @@ describe 'gitlab:uploads:migrate rake tasks' do run_rake_task("gitlab:uploads:migrate", *args) end - it 'enqueue jobs in batch' do - expect(ObjectStorage::MigrateUploadsWorker).to receive(:enqueue!).exactly(4).times + shared_examples 'enqueue jobs in batch' do |batch:| + it do + expect(ObjectStorage::MigrateUploadsWorker) + .to receive(:enqueue!).exactly(batch).times - run + run + end + end + + context 'Upload has store = nil' do + before do + Upload.where(model: projects.first(5)).update_all(store: nil) + end + + it_behaves_like 'enqueue jobs in batch', batch: 4 end end |