Commit 099f9fcd authored by Stan Hu's avatar Stan Hu

Fallback to creating a note if DiffNote fails to import

parent da02df04
...@@ -216,31 +216,49 @@ module Gitlab ...@@ -216,31 +216,49 @@ module Gitlab
def import_inline_comments(inline_comments, merge_request) def import_inline_comments(inline_comments, merge_request)
inline_comments.each do |comment| inline_comments.each do |comment|
position = build_position(merge_request, comment) position = build_position(merge_request, comment)
parent = build_diff_note(merge_request, comment, position) parent = create_diff_note(merge_request, comment, position)
next unless parent&.persisted? next unless parent&.persisted?
discussion_id = parent.discussion_id discussion_id = parent.discussion_id
comment.comments.each do |reply| comment.comments.each do |reply|
build_diff_note(merge_request, reply, position, discussion_id) create_diff_note(merge_request, reply, position, discussion_id)
end end
end end
end end
def build_diff_note(merge_request, comment, position, discussion_id = nil) def create_diff_note(merge_request, comment, position, discussion_id = nil)
attributes = pull_request_comment_attributes(comment) attributes = pull_request_comment_attributes(comment)
attributes.merge!( attributes.merge!(
position: position, position: position,
type: 'DiffNote') type: 'DiffNote')
attributes[:discussion_id] = discussion_id if discussion_id attributes[:discussion_id] = discussion_id if discussion_id
merge_request.notes.create!(attributes) note = merge_request.notes.build(attributes)
if note.valid?
note.save
return note
end
# Fallback to a regular comment
create_fallback_diff_note(merge_request, comment)
rescue StandardError => e rescue StandardError => e
errors << { type: :pull_request, id: comment.id, errors: e.message } errors << { type: :pull_request, id: comment.id, errors: e.message }
nil nil
end end
# Bitbucket Server supports the ability to comment on any line, not just the
# line in the diff. If we can't add the note as a DiffNote, fallback to creating
# a regular note.
def create_fallback_diff_note(merge_request, comment)
attributes = pull_request_comment_attributes(comment)
attributes[:note] = "Comment on file: #{comment.file_path}, old position: #{comment.old_pos}, new_position: #{comment.new_pos}\n\n" + attributes[:note]
merge_request.notes.create!(attributes)
end
def build_position(merge_request, pr_comment) def build_position(merge_request, pr_comment)
params = { params = {
diff_refs: merge_request.diff_refs, diff_refs: merge_request.diff_refs,
......
...@@ -168,6 +168,40 @@ describe Gitlab::BitbucketServerImport::Importer do ...@@ -168,6 +168,40 @@ describe Gitlab::BitbucketServerImport::Importer do
expect(reply_note.position.new_line).to eq(inline_note.new_pos) expect(reply_note.position.new_line).to eq(inline_note.new_pos)
end end
it 'falls back to comments if diff comments fail to validate' do
# https://gitlab.com/gitlab-org/gitlab-test/compare/c1acaa58bbcbc3eafe538cb8274ba387047b69f8...5937ac0a7beb003549fc5fd26fc247ad
inline_note = instance_double(
BitbucketServer::Representation::PullRequestComment,
file_type: 'REMOVED',
from_sha: sample.commits.first,
to_sha: sample.commits.last,
file_path: '.gitmodules',
old_pos: 8,
new_pos: 9,
note: 'This is a note with an invalid line position.',
author_email: project.owner.email,
comments: [],
created_at: now,
updated_at: now)
inline_comment = instance_double(
BitbucketServer::Representation::Activity,
comment?: true,
inline_comment?: true,
merge_event?: false,
comment: inline_note)
expect(subject.client).to receive(:activities).and_return([inline_comment])
expect { subject.execute }.to change { MergeRequest.count }.by(1)
merge_request = MergeRequest.first
expect(merge_request.notes.count).to eq(1)
note = merge_request.notes.first
expect(note.note).to start_with('Comment on file:')
end
it 'restores branches of inaccessible SHAs' do it 'restores branches of inaccessible SHAs' do
end end
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment