70 lines
2.2 KiB
Ruby
70 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Gitlab::BackgroundMigration::BackfillDesignInternalIds, :migration, schema: 20201030203854 do
|
|
subject { described_class.new(designs) }
|
|
|
|
let_it_be(:namespaces) { table(:namespaces) }
|
|
let_it_be(:projects) { table(:projects) }
|
|
let_it_be(:designs) { table(:design_management_designs) }
|
|
|
|
let(:namespace) { namespaces.create!(name: 'foo', path: 'foo') }
|
|
let(:project) { projects.create!(namespace_id: namespace.id) }
|
|
let(:project_2) { projects.create!(namespace_id: namespace.id) }
|
|
|
|
def create_design!(proj = project)
|
|
designs.create!(project_id: proj.id, filename: generate(:filename))
|
|
end
|
|
|
|
def migrate!
|
|
relation = designs.where(project_id: [project.id, project_2.id]).select(:project_id).distinct
|
|
|
|
subject.perform(relation)
|
|
end
|
|
|
|
it 'backfills the iid for designs' do
|
|
3.times { create_design! }
|
|
|
|
expect do
|
|
migrate!
|
|
end.to change { designs.pluck(:iid) }.from(contain_exactly(nil, nil, nil)).to(contain_exactly(1, 2, 3))
|
|
end
|
|
|
|
it 'scopes IIDs and handles range and starting-point correctly' do
|
|
create_design!.update!(iid: 10)
|
|
create_design!.update!(iid: 12)
|
|
create_design!(project_2).update!(iid: 7)
|
|
project_3 = projects.create!(namespace_id: namespace.id)
|
|
|
|
2.times { create_design! }
|
|
2.times { create_design!(project_2) }
|
|
2.times { create_design!(project_3) }
|
|
|
|
migrate!
|
|
|
|
expect(designs.where(project_id: project.id).pluck(:iid)).to contain_exactly(10, 12, 13, 14)
|
|
expect(designs.where(project_id: project_2.id).pluck(:iid)).to contain_exactly(7, 8, 9)
|
|
expect(designs.where(project_id: project_3.id).pluck(:iid)).to contain_exactly(nil, nil)
|
|
end
|
|
|
|
it 'updates the internal ID records' do
|
|
design = create_design!
|
|
2.times { create_design! }
|
|
design.update!(iid: 10)
|
|
scope = { project_id: project.id }
|
|
usage = :design_management_designs
|
|
init = ->(_d, _s) { 0 }
|
|
|
|
::InternalId.track_greatest(design, scope, usage, 10, init)
|
|
|
|
migrate!
|
|
|
|
next_iid = ::InternalId.generate_next(design, scope, usage, init)
|
|
|
|
expect(designs.pluck(:iid)).to contain_exactly(10, 11, 12)
|
|
expect(design.reload.iid).to eq(10)
|
|
expect(next_iid).to eq(13)
|
|
end
|
|
end
|