diff options
Diffstat (limited to 'spec/lib/gitlab/background_migration/backfill_imported_issue_search_data_spec.rb')
-rw-r--r-- | spec/lib/gitlab/background_migration/backfill_imported_issue_search_data_spec.rb | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/spec/lib/gitlab/background_migration/backfill_imported_issue_search_data_spec.rb b/spec/lib/gitlab/background_migration/backfill_imported_issue_search_data_spec.rb new file mode 100644 index 00000000000..e363a5a6b20 --- /dev/null +++ b/spec/lib/gitlab/background_migration/backfill_imported_issue_search_data_spec.rb @@ -0,0 +1,106 @@ +# frozen_string_literal: true + +require 'spec_helper' +require_migration! + +RSpec.describe Gitlab::BackgroundMigration::BackfillImportedIssueSearchData, + :migration, + schema: 20220707075300 do + let!(:namespace) { table(:namespaces).create!(name: 'user', path: 'user') } + let!(:issue_search_data_table) { table(:issue_search_data) } + + let!(:user) { table(:users).create!(email: 'author@example.com', username: 'author', projects_limit: 10) } + let!(:project) do + table(:projects) + .create!( + namespace_id: namespace.id, + creator_id: user.id, + name: 'projecty', + path: 'path', + project_namespace_id: namespace.id) + end + + let!(:issue) do + table(:issues).create!( + project_id: project.id, + title: 'Patterson', + description: FFaker::HipsterIpsum.paragraph + ) + end + + let(:migration) do + described_class.new(start_id: issue.id, + end_id: issue.id + 30, + batch_table: :issues, + batch_column: :id, + sub_batch_size: 2, + pause_ms: 0, + connection: ApplicationRecord.connection) + end + + let(:perform_migration) { migration.perform } + + context 'when issue has search data record' do + let!(:issue_search_data) { issue_search_data_table.create!(project_id: project.id, issue_id: issue.id) } + + it 'does not create or update any search data records' do + expect { perform_migration } + .to not_change { issue_search_data_table.count } + .and not_change { issue_search_data } + + expect(issue_search_data_table.count).to eq(1) + end + end + + context 'when issue has no search data record' do + let(:title_node) { "'#{issue.title.downcase}':1A" } + + it 'creates search data records' do + expect { perform_migration } + .to change { issue_search_data_table.count }.from(0).to(1) + + expect(issue_search_data_table.find_by(project_id: project.id).issue_id) + .to eq(issue.id) + + expect(issue_search_data_table.find_by(project_id: project.id).search_vector) + .to include(title_node) + end + end + + context 'error handling' do + let!(:issue2) do + table(:issues).create!( + project_id: project.id, + title: 'Chatterton', + description: FFaker::HipsterIpsum.paragraph + ) + end + + before do + issue.update!(description: Array.new(30_000) { SecureRandom.hex }.join(' ')) + end + + let(:title_node2) { "'#{issue2.title.downcase}':1A" } + + it 'skips insertion for that issue but continues with migration' do + 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: issue.id)) + end + + expect { perform_migration }.to change { issue_search_data_table.count }.from(0).to(1) + expect(issue_search_data_table.find_by(issue_id: issue.id)).to eq(nil) + expect(issue_search_data_table.find_by(issue_id: issue2.id).search_vector) + .to include(title_node2) + end + + it 're-raises exceptions' do + allow(migration) + .to receive(:update_search_data_individually) + .and_raise(ActiveRecord::StatementTimeout) + + expect { perform_migration }.to raise_error(ActiveRecord::StatementTimeout) + end + end +end |