Use new `each_batch` helper instead of custom one

In stage_id backgrond migration.

1 files changed, 3 insertions, 3 deletions

diff --git a/spec/migrations/migrate_stage_id_reference_in_background_spec.rb b/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
index f3dde8b59c0..90ad5c39089 100644
--- a/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
+++ b/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
@@ -47,9 +47,9 @@ describe MigrateStageIdReferenceInBackground, :migration, :sidekiq do
       Timecop.freeze do
         migrate!
 
-        expect(described_class::MIGRATION).to be_scheduled_migration(2.minutes, 1, 3)
-        expect(described_class::MIGRATION).to be_scheduled_migration(4.minutes, 3, 5)
-        expect(described_class::MIGRATION).to be_scheduled_migration(6.minutes, 5, 0)
+        expect(described_class::MIGRATION).to be_scheduled_migration(2.minutes, 1, 2)
+        expect(described_class::MIGRATION).to be_scheduled_migration(4.minutes, 3, 4)
+        expect(described_class::MIGRATION).to be_scheduled_migration(6.minutes, 5, 6)
         expect(BackgroundMigrationWorker.jobs.size).to eq 3
       end
     end