apply feedback

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

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