diff options
Diffstat (limited to 'spec/requests/admin/background_migrations_controller_spec.rb')
-rw-r--r-- | spec/requests/admin/background_migrations_controller_spec.rb | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/spec/requests/admin/background_migrations_controller_spec.rb b/spec/requests/admin/background_migrations_controller_spec.rb index 884448fdd95..fe2a2470511 100644 --- a/spec/requests/admin/background_migrations_controller_spec.rb +++ b/spec/requests/admin/background_migrations_controller_spec.rb @@ -97,6 +97,7 @@ RSpec.describe Admin::BackgroundMigrationsController, :enable_admin_mode do describe 'POST #retry' do let(:migration) { create(:batched_background_migration, :failed) } + let(:job_class) { Gitlab::BackgroundMigration::CopyColumnUsingBackgroundMigrationJob } before do create(:batched_background_migration_job, :failed, batched_migration: migration, batch_size: 10, min_value: 6, max_value: 15, attempts: 3) @@ -107,7 +108,8 @@ RSpec.describe Admin::BackgroundMigrationsController, :enable_admin_mode do anything, batch_min_value: 6, batch_size: 5, - job_arguments: migration.job_arguments + job_arguments: migration.job_arguments, + job_class: job_class ).and_return([6, 10]) end end |