1 files changed, 8 insertions, 7 deletions
diff --git a/spec/lib/gitlab/database/migrations/test_batched_background_runner_spec.rb b/spec/lib/gitlab/database/migrations/test_batched_background_runner_spec.rb
index f1f72d71e1a..9451a6bd34a 100644
--- a/spec/lib/gitlab/database/migrations/test_batched_background_runner_spec.rb
+++ b/spec/lib/gitlab/database/migrations/test_batched_background_runner_spec.rb
@@ -18,7 +18,7 @@ RSpec.describe Gitlab::Database::Migrations::TestBatchedBackgroundRunner, :freez
 
   let(:connection) { ApplicationRecord.connection }
 
-  let(:table_name) { "_test_column_copying"}
+  let(:table_name) { "_test_column_copying" }
 
   before do
     connection.execute(<<~SQL)
@@ -50,18 +50,16 @@ RSpec.describe Gitlab::Database::Migrations::TestBatchedBackgroundRunner, :freez
   context 'with jobs to run' do
     let(:migration_name) { 'TestBackgroundMigration' }
 
-    before do
-      migration.queue_batched_background_migration(
-        migration_name, table_name, :id, job_interval: 5.minutes, batch_size: 100
-      )
-    end
-
     it 'samples jobs' do
       calls = []
       define_background_migration(migration_name) do |*args|
         calls << args
       end
 
+      migration.queue_batched_background_migration(migration_name, table_name, :id,
+                                                   job_interval: 5.minutes,
+                                                   batch_size: 100)
+
       described_class.new(result_dir: result_dir, connection: connection).run_jobs(for_duration: 3.minutes)
 
       expect(calls.count).to eq(10) # 1000 rows / batch size 100 = 10
@@ -70,6 +68,9 @@ RSpec.describe Gitlab::Database::Migrations::TestBatchedBackgroundRunner, :freez
     context 'with multiple jobs to run' do
       it 'runs all jobs created within the last 3 hours' do
         old_migration = define_background_migration(migration_name)
+        migration.queue_batched_background_migration(migration_name, table_name, :id,
+                                             job_interval: 5.minutes,
+                                             batch_size: 100)
 
         travel 4.hours