61 lines
2.1 KiB
Ruby
61 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Users::MigrateRecordsToGhostUserInBatchesService do
|
|
let(:service) { described_class.new }
|
|
|
|
let_it_be(:ghost_user_migration) { create(:ghost_user_migration) }
|
|
|
|
describe '#execute' do
|
|
it 'stops when execution time limit reached' do
|
|
expect_next_instance_of(::Gitlab::Utils::ExecutionTracker) do |tracker|
|
|
expect(tracker).to receive(:over_limit?).and_return(true)
|
|
end
|
|
|
|
expect(Users::MigrateRecordsToGhostUserService).not_to receive(:new)
|
|
|
|
service.execute
|
|
end
|
|
|
|
it 'calls Users::MigrateRecordsToGhostUserService' do
|
|
expect_next_instance_of(Users::MigrateRecordsToGhostUserService) do |service|
|
|
expect(service).to(
|
|
receive(:execute)
|
|
.with(hard_delete: ghost_user_migration.hard_delete))
|
|
end
|
|
|
|
service.execute
|
|
end
|
|
|
|
it 'process jobs ordered by the consume_after timestamp' do
|
|
older_ghost_user_migration = create(:ghost_user_migration, user: create(:user),
|
|
consume_after: 5.minutes.ago)
|
|
|
|
# setup execution tracker to only allow a single job to be processed
|
|
allow_next_instance_of(::Gitlab::Utils::ExecutionTracker) do |tracker|
|
|
allow(tracker).to receive(:over_limit?).and_return(false, true)
|
|
end
|
|
|
|
expect(Users::MigrateRecordsToGhostUserService).to(
|
|
receive(:new).with(older_ghost_user_migration.user,
|
|
older_ghost_user_migration.initiator_user,
|
|
any_args)
|
|
).and_call_original
|
|
|
|
service.execute
|
|
end
|
|
|
|
it 'reschedules job in case of an error', :freeze_time do
|
|
expect_next_instance_of(Users::MigrateRecordsToGhostUserService) do |service|
|
|
expect(service).to(receive(:execute)).and_raise(ActiveRecord::QueryCanceled)
|
|
end
|
|
expect(Gitlab::ErrorTracking).to receive(:track_exception)
|
|
|
|
expect { service.execute }.to(
|
|
change { ghost_user_migration.reload.consume_after }
|
|
.to(30.minutes.from_now))
|
|
end
|
|
end
|
|
end
|