Merge branch '327405-fix-setting-of-status-for-background-migration-jobs' into 'master'

Ensure that migration job status is updated

See merge request gitlab-org/gitlab!59864

lib/gitlab/database/background_migration/batched_migration_wrapper.rb

@@ -19,10 +19,10 @@ module Gitlab
           execute_batch(batch_tracking_record)
 
           batch_tracking_record.status = :succeeded
-        rescue => e
+        rescue Exception # rubocop:disable Lint/RescueException
           batch_tracking_record.status = :failed
 
-          raise e
+          raise
         ensure
           finish_tracking_execution(batch_tracking_record)
           track_prometheus_metrics(batch_tracking_record)

spec/lib/gitlab/database/background_migration/batched_migration_wrapper_spec.rb

@@ -112,13 +112,14 @@ RSpec.describe Gitlab::Database::BackgroundMigration::BatchedMigrationWrapper, '
   end
 
   context 'when the migration job raises an error' do
-    it 'marks the tracking record as failed before raising the error' do
+    shared_examples 'an error is raised' do |error_class|
+      it 'marks the tracking record as failed' do
         expect(job_instance).to receive(:perform)
           .with(1, 10, 'events', 'id', 1, 'id', 'other_id')
-        .and_raise(RuntimeError, 'Something broke!')
+          .and_raise(error_class)
 
         freeze_time do
-        expect { subject }.to raise_error(RuntimeError, 'Something broke!')
+          expect { subject }.to raise_error(error_class)
 
           reloaded_job_record = job_record.reload
 
@@ -127,4 +128,8 @@ RSpec.describe Gitlab::Database::BackgroundMigration::BatchedMigrationWrapper, '
         end
       end
     end
+
+    it_behaves_like 'an error is raised', RuntimeError.new('Something broke!')
+    it_behaves_like 'an error is raised', SignalException.new('SIGTERM')
+  end
 end