Use ActiveRecord 5 batching to schedule bg migration

1 files changed, 26 insertions, 29 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 8b497656377..d3645ec0395 100644
--- a/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
+++ b/spec/migrations/migrate_stage_id_reference_in_background_spec.rb
@@ -1,44 +1,39 @@
 require 'spec_helper'
 require Rails.root.join('db', 'post_migrate', '20170628080858_migrate_stage_id_reference_in_background')
 
-RSpec::Matchers.define :have_migrated do |*expected|
-  match do |migration|
-    BackgroundMigrationWorker.jobs.any? do |job|
-      job['enqueued_at'].present? && job['args'] == [migration, expected]
+describe MigrateStageIdReferenceInBackground, :migration, :sidekiq do
+  matcher :have_migrated do |*expected|
+    match do |migration|
+      BackgroundMigrationWorker.jobs.any? do |job|
+        job['enqueued_at'].present? && job['args'] == [migration, expected]
+      end
     end
-  end
 
-  failure_message do |migration|
-    <<-EOS
-      Background migration `#{migration}` with args `#{expected.inspect}`
-      not migrated!
-    EOS
+    failure_message do |migration|
+      "Migration `#{migration}` with args `#{expected.inspect}` not executed!"
+    end
   end
-end
 
-RSpec::Matchers.define :have_scheduled_migration do |time, *expected|
-  match do |migration|
-    BackgroundMigrationWorker.jobs.any? do |job|
-      job['args'] == [migration, expected] && job['at'] >= time
+  matcher :have_scheduled_migration do |delay, *expected|
+    match do |migration|
+      BackgroundMigrationWorker.jobs.any? do |job|
+        job['args'] == [migration, expected] &&
+          job['at'].to_f == (delay.to_f + Time.now.to_f)
+      end
     end
-  end
 
-  failure_message do |migration|
-    <<-EOS
-      Background migration `#{migration}` with args `#{expected.inspect}`
-      not scheduled!
-    EOS
+    failure_message do |migration|
+      "Migration `#{migration}` with args `#{expected.inspect}` not scheduled!"
+    end
   end
-end
 
-describe MigrateStageIdReferenceInBackground, :migration do
   let(:jobs) { table(:ci_builds) }
   let(:stages) { table(:ci_stages) }
   let(:pipelines) { table(:ci_pipelines) }
   let(:projects) { table(:projects) }
 
   before do
-    stub_const('MigrateStageIdReferenceInBackground::BATCH_SIZE', 1)
+    stub_const('MigrateStageIdReferenceInBackground::BATCH_SIZE', 2)
 
     projects.create!(id: 123, name: 'gitlab1', path: 'gitlab1')
     pipelines.create!(id: 1, project_id: 123, ref: 'master', sha: 'adf43c3a')
@@ -55,12 +50,14 @@ describe MigrateStageIdReferenceInBackground, :migration do
 
   it 'correctly schedules background migrations' do
     Sidekiq::Testing.fake! do
-      migrate!
+      Timecop.freeze do
+        migrate!
 
-      expect(described_class::MIGRATION).to have_migrated(1)
-      expect(described_class::MIGRATION).to have_migrated(2)
-      expect(described_class::MIGRATION).to have_scheduled_migration(5.minutes, 3)
-      expect(described_class::MIGRATION).to have_scheduled_migration(5.minutes, 4)
+        expect(described_class::MIGRATION).to have_scheduled_migration(5.minutes, 1)
+        expect(described_class::MIGRATION).to have_scheduled_migration(5.minutes, 2)
+        expect(described_class::MIGRATION).to have_scheduled_migration(10.minutes, 3)
+        expect(described_class::MIGRATION).to have_scheduled_migration(10.minutes, 4)
+      end
     end
   end