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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::BackgroundMigration::BackfillPreparedAtMergeRequests, :migration,
feature_category: :code_review_workflow, schema: 20230202135758 do
let(:namespaces) { table(:namespaces) }
let(:projects) { table(:projects) }
let(:mr_table) { table(:merge_requests) }
let(:namespace) { namespaces.create!(name: 'batchtest1', type: 'Group', path: 'space1') }
let(:proj_namespace) { namespaces.create!(name: 'proj1', path: 'proj1', type: 'Project', parent_id: namespace.id) }
let(:project) do
projects.create!(name: 'proj1', path: 'proj1', namespace_id: namespace.id, project_namespace_id: proj_namespace.id)
end
it 'updates merge requests with prepared_at nil' do
time = Time.current
mr_1 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
prepared_at: nil, merge_status: 'checking')
mr_2 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
prepared_at: nil, merge_status: 'preparing')
mr_3 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
prepared_at: time)
mr_4 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
prepared_at: time, merge_status: 'checking')
mr_5 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
prepared_at: time, merge_status: 'preparing')
test_worker = described_class.new(
start_id: mr_1.id,
end_id: [(mr_5.id + 1), 100].max,
batch_table: :merge_requests,
batch_column: :id,
sub_batch_size: 10,
pause_ms: 0,
connection: ApplicationRecord.connection
)
expect(mr_1.prepared_at).to be_nil
expect(mr_2.prepared_at).to be_nil
expect(mr_3.prepared_at.to_i).to eq(time.to_i)
expect(mr_4.prepared_at.to_i).to eq(time.to_i)
expect(mr_5.prepared_at.to_i).to eq(time.to_i)
test_worker.perform
expect(mr_1.reload.prepared_at.to_i).to eq(mr_1.created_at.to_i)
expect(mr_2.reload.prepared_at).to be_nil
expect(mr_3.reload.prepared_at.to_i).to eq(time.to_i)
expect(mr_4.reload.prepared_at.to_i).to eq(time.to_i)
expect(mr_5.reload.prepared_at.to_i).to eq(time.to_i)
end
end
|