1 files changed, 143 insertions, 0 deletions
diff --git a/spec/migrations/adjust_task_note_rename_background_migration_values_spec.rb b/spec/migrations/adjust_task_note_rename_background_migration_values_spec.rb
new file mode 100644
index 00000000000..422d0655e36
--- /dev/null
+++ b/spec/migrations/adjust_task_note_rename_background_migration_values_spec.rb
@@ -0,0 +1,143 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require_migration!
+
+RSpec.describe AdjustTaskNoteRenameBackgroundMigrationValues, :migration do
+  let(:finished_status) { 3 }
+  let(:failed_status) { described_class::MIGRATION_FAILED_STATUS }
+  let(:active_status) { described_class::MIGRATION_ACTIVE_STATUS }
+
+  shared_examples 'task note migration with failing batches' do
+    it 'updates batch sizes and resets failed batches' do
+      migration = create_background_migration(status: initial_status)
+      batches = []
+
+      batches << create_failed_batched_job(migration)
+      batches << create_failed_batched_job(migration)
+
+      migrate!
+
+      expect(described_class::JOB_CLASS_NAME).to have_scheduled_batched_migration(
+        table_name: :system_note_metadata,
+        column_name: :id,
+        interval: 2.minutes,
+        batch_size: described_class::NEW_BATCH_SIZE,
+        max_batch_size: 20_000,
+        sub_batch_size: described_class::NEW_SUB_BATCH_SIZE
+      )
+      expect(migration.reload.status).to eq(active_status)
+
+      updated_batches = batches.map { |b| b.reload.attributes.slice('attempts', 'sub_batch_size') }
+      expect(updated_batches).to all(eq("attempts" => 0, "sub_batch_size" => 10))
+    end
+  end
+
+  describe '#up' do
+    context 'when migration was already finished' do
+      it 'does not update batch sizes' do
+        create_background_migration(status: finished_status)
+
+        migrate!
+
+        expect(described_class::JOB_CLASS_NAME).to have_scheduled_batched_migration(
+          table_name: :system_note_metadata,
+          column_name: :id,
+          interval: 2.minutes,
+          batch_size: described_class::OLD_BATCH_SIZE,
+          max_batch_size: 20_000,
+          sub_batch_size: described_class::OLD_SUB_BATCH_SIZE
+        )
+      end
+    end
+
+    context 'when the migration had failing batches' do
+      context 'when migration had a failed status' do
+        it_behaves_like 'task note migration with failing batches' do
+          let(:initial_status) { failed_status }
+        end
+
+        it 'updates started_at timestamp' do
+          migration = create_background_migration(status: failed_status)
+          now = Time.zone.now
+
+          travel_to now do
+            migrate!
+            migration.reload
+          end
+
+          expect(migration.started_at).to be_like_time(now)
+        end
+      end
+
+      context 'when migration had an active status' do
+        it_behaves_like 'task note migration with failing batches' do
+          let(:initial_status) { active_status }
+        end
+
+        it 'does not update started_at timestamp' do
+          migration = create_background_migration(status: active_status)
+          original_time = migration.started_at
+
+          migrate!
+          migration.reload
+
+          expect(migration.started_at).to be_like_time(original_time)
+        end
+      end
+    end
+  end
+
+  describe '#down' do
+    it 'reverts to old batch sizes' do
+      create_background_migration(status: finished_status)
+
+      migrate!
+      schema_migrate_down!
+
+      expect(described_class::JOB_CLASS_NAME).to have_scheduled_batched_migration(
+        table_name: :system_note_metadata,
+        column_name: :id,
+        interval: 2.minutes,
+        batch_size: described_class::OLD_BATCH_SIZE,
+        max_batch_size: 20_000,
+        sub_batch_size: described_class::OLD_SUB_BATCH_SIZE
+      )
+    end
+  end
+
+  def create_failed_batched_job(migration)
+    table(:batched_background_migration_jobs).create!(
+      batched_background_migration_id: migration.id,
+      status: described_class::JOB_FAILED_STATUS,
+      min_value: 1,
+      max_value: 10,
+      attempts: 3,
+      batch_size: described_class::OLD_BATCH_SIZE,
+      sub_batch_size: described_class::OLD_SUB_BATCH_SIZE
+    )
+  end
+
+  def create_background_migration(status:)
+    migrations_table = table(:batched_background_migrations)
+    # make sure we only have on migration with that job class name in the specs
+    migrations_table.where(job_class_name: described_class::JOB_CLASS_NAME).delete_all
+
+    migrations_table.create!(
+      job_class_name: described_class::JOB_CLASS_NAME,
+      status: status,
+      max_value: 10,
+      max_batch_size: 20_000,
+      batch_size: described_class::OLD_BATCH_SIZE,
+      sub_batch_size: described_class::OLD_SUB_BATCH_SIZE,
+      interval: 2.minutes,
+      table_name: :system_note_metadata,
+      column_name: :id,
+      total_tuple_count: 100_000,
+      pause_ms: 100,
+      gitlab_schema: :gitlab_main,
+      job_arguments: [],
+      started_at: 2.days.ago
+    )
+  end
+end