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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::BackgroundMigration::BackfillIssueSearchData, :migration, schema: 20220326161803 do
let(:namespaces_table) { table(:namespaces) }
let(:projects_table) { table(:projects) }
let(:issue_search_data_table) { table(:issue_search_data) }
let!(:namespace) { namespaces_table.create!(name: 'gitlab-org', path: 'gitlab-org') }
let!(:project) { projects_table.create!(name: 'gitlab', path: 'gitlab-org/gitlab-ce', namespace_id: namespace.id) }
let!(:issues) { Array.new(10) { table(:issues).create!(project_id: project.id, title: 'test title', description: 'test description') } }
let(:migration) { described_class.new }
before do
allow(migration).to receive(:sleep)
end
it 'backfills search data for the specified records' do
# sleeps for every sub-batch
expect(migration).to receive(:sleep).with(0.05).exactly(3).times
migration.perform(issues[0].id, issues[5].id, :issues, :id, 2, 50)
expect(issue_search_data_table.count).to eq(6)
end
it 'skips issues that already have search data' do
old_time = Time.new(2019, 1, 1).in_time_zone
issue_search_data_table.create!(project_id: project.id, issue_id: issues[0].id, updated_at: old_time)
migration.perform(issues[0].id, issues[5].id, :issues, :id, 2, 50)
expect(issue_search_data_table.count).to eq(6)
expect(issue_search_data_table.find_by_issue_id(issues[0].id).updated_at).to be_like_time(old_time)
end
it 'rescues batch with bad data and inserts other rows' do
issues[1].update!(description: Array.new(30_000) { SecureRandom.hex }.join(' '))
expect_next_instance_of(Gitlab::BackgroundMigration::Logger) do |logger|
expect(logger).to receive(:error).with(a_hash_including(message: /string is too long for tsvector/, model_id: issues[1].id))
end
expect { migration.perform(issues[0].id, issues[5].id, :issues, :id, 2, 50) }.not_to raise_error
expect(issue_search_data_table.count).to eq(5)
expect(issue_search_data_table.find_by_issue_id(issues[1].id)).to eq(nil)
end
it 're-raises other errors' do
allow(migration).to receive(:update_search_data).and_raise(ActiveRecord::StatementTimeout)
expect { migration.perform(issues[0].id, issues[5].id, :issues, :id, 2, 50) }.to raise_error(ActiveRecord::StatementTimeout)
end
end
|
