35 lines
1.4 KiB
Ruby
35 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
require_migration!
|
|
|
|
RSpec.describe FinalizeBackfillNullNoteDiscussionIds, :migration do
|
|
subject(:migration) { described_class.new }
|
|
|
|
let(:notes) { table(:notes) }
|
|
let(:bg_migration_class) { Gitlab::BackgroundMigration::BackfillNoteDiscussionId }
|
|
let(:bg_migration) { instance_double(bg_migration_class) }
|
|
|
|
before do
|
|
stub_const("#{described_class.name}::BATCH_SIZE", 2)
|
|
end
|
|
|
|
it 'performs remaining background migrations', :aggregate_failures do
|
|
# Already migrated
|
|
notes.create!(noteable_type: 'Issue', noteable_id: 1, discussion_id: Digest::SHA1.hexdigest('note1'))
|
|
notes.create!(noteable_type: 'Issue', noteable_id: 1, discussion_id: Digest::SHA1.hexdigest('note2'))
|
|
# update required
|
|
record1 = notes.create!(noteable_type: 'Issue', noteable_id: 1, discussion_id: nil)
|
|
record2 = notes.create!(noteable_type: 'Issue', noteable_id: 1, discussion_id: nil)
|
|
record3 = notes.create!(noteable_type: 'Issue', noteable_id: 1, discussion_id: nil)
|
|
|
|
expect(Gitlab::BackgroundMigration).to receive(:steal).with(bg_migration_class.name.demodulize)
|
|
expect(bg_migration_class).to receive(:new).twice.and_return(bg_migration)
|
|
expect(bg_migration).to receive(:perform).with(record1.id, record2.id)
|
|
expect(bg_migration).to receive(:perform).with(record3.id, record3.id)
|
|
|
|
migrate!
|
|
end
|
|
end
|