1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# 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
|
