diff options
-rw-r--r-- | lib/tasks/gitlab/uploads/migrate.rake | 2 | ||||
-rw-r--r-- | spec/tasks/gitlab/uploads/migrate_rake_spec.rb | 4 |
2 files changed, 4 insertions, 2 deletions
diff --git a/lib/tasks/gitlab/uploads/migrate.rake b/lib/tasks/gitlab/uploads/migrate.rake index f0f4acd6140..e7e656a911b 100644 --- a/lib/tasks/gitlab/uploads/migrate.rake +++ b/lib/tasks/gitlab/uploads/migrate.rake @@ -25,7 +25,7 @@ namespace :gitlab do Upload.class_eval { include EachBatch } unless Upload < EachBatch Upload - .where(store: [nil, 1], + .where(store: [nil, ObjectStorage::Store::LOCAL], uploader: @uploader_class.to_s, model_type: @model_class.base_class.sti_name) end diff --git a/spec/tasks/gitlab/uploads/migrate_rake_spec.rb b/spec/tasks/gitlab/uploads/migrate_rake_spec.rb index f273992008e..53b1e45ba54 100644 --- a/spec/tasks/gitlab/uploads/migrate_rake_spec.rb +++ b/spec/tasks/gitlab/uploads/migrate_rake_spec.rb @@ -30,9 +30,11 @@ describe 'gitlab:uploads:migrate rake tasks' do end end + it_behaves_like 'enqueue jobs in batch', batch: 4 + context 'Upload has store = nil' do before do - Upload.where(model: projects.first(5)).update_all(store: nil) + Upload.where(model: projects).update_all(store: nil) end it_behaves_like 'enqueue jobs in batch', batch: 4 |