Commit 4af1712d authored by Jarka Košanová's avatar Jarka Košanová

Extract code from Issues::MoveService

Extract code to make it easier reusable
- introduce AttributesRewriter and ContentRewriter
  - support group entites when rewriting content
- make Uploader copy_to working for Namespaces
parent 2e690c82
# frozen_string_literal: true
module Issuable
module Clone
class AttributesRewriter < ::Issuable::Clone::BaseService
def initialize(current_user, original_entity, new_entity)
@current_user = current_user
@original_entity = original_entity
@new_entity = new_entity
end
def execute
new_entity.update(milestone: cloneable_milestone, labels: cloneable_labels)
copy_resource_label_events
end
private
def cloneable_milestone
title = original_entity.milestone&.title
return unless title
params = { title: title, project_ids: new_entity.project&.id, group_ids: group&.id }
milestones = MilestonesFinder.new(params).execute
milestones.first
end
def cloneable_labels
params = {
project_id: new_entity.project&.id,
group_id: group&.id,
title: original_entity.labels.select(:title),
include_ancestor_groups: true
}
params[:only_group_labels] = true if new_parent.is_a?(Group)
LabelsFinder.new(current_user, params).execute
end
def copy_resource_label_events
original_entity.resource_label_events.find_in_batches do |batch|
events = batch.map do |event|
entity_key = new_entity.is_a?(Issue) ? 'issue_id' : 'epic_id'
# rubocop: disable CodeReuse/ActiveRecord
event.attributes
.except('id', 'reference', 'reference_html')
.merge(entity_key => new_entity.id, 'action' => ResourceLabelEvent.actions[event.action])
# rubocop: enable CodeReuse/ActiveRecord
end
Gitlab::Database.bulk_insert(ResourceLabelEvent.table_name, events)
end
end
def entity_key
new_entity.class.name.parameterize('_').foreign_key
end
end
end
end
# frozen_string_literal: true
module Issuable
module Clone
class BaseService < IssuableBaseService
attr_reader :original_entity, :new_entity
alias_method :old_project, :project
def execute(original_entity, new_project = nil)
@original_entity = original_entity
# Using transaction because of a high resources footprint
# on rewriting notes (unfolding references)
#
ActiveRecord::Base.transaction do
@new_entity = create_new_entity
update_new_entity
update_old_entity
create_notes
end
end
private
def update_new_entity
rewriters = [ContentRewriter, AttributesRewriter]
rewriters.each do |rewriter|
rewriter.new(current_user, original_entity, new_entity).execute
end
end
def update_old_entity
close_issue
end
def create_notes
add_note_from
add_note_to
end
def close_issue
close_service = Issues::CloseService.new(old_project, current_user)
close_service.execute(original_entity, notifications: false, system_note: false)
end
def new_parent
new_entity.project ? new_entity.project : new_entity.group
end
def group
if new_entity.project&.group && current_user.can?(:read_group, new_entity.project.group)
new_entity.project.group
end
end
end
end
end
# frozen_string_literal: true
module Issuable
module Clone
class ContentRewriter < ::Issuable::Clone::BaseService
def initialize(current_user, original_entity, new_entity)
@current_user = current_user
@original_entity = original_entity
@new_entity = new_entity
@project = original_entity.project
end
def execute
rewrite_description
rewrite_award_emoji(original_entity, new_entity)
rewrite_notes
end
private
def rewrite_description
new_entity.update(description: rewrite_content(original_entity.description))
end
def rewrite_notes
original_entity.notes_with_associations.find_each do |note|
new_note = note.dup
new_params = {
project: new_entity.project, noteable: new_entity,
note: rewrite_content(new_note.note),
created_at: note.created_at,
updated_at: note.updated_at
}
if note.system_note_metadata
new_params[:system_note_metadata] = note.system_note_metadata.dup
end
new_note.update(new_params)
rewrite_award_emoji(note, new_note)
end
end
def rewrite_content(content)
return unless content
rewriters = [Gitlab::Gfm::ReferenceRewriter, Gitlab::Gfm::UploadsRewriter]
rewriters.inject(content) do |text, klass|
rewriter = klass.new(text, old_project, current_user)
rewriter.rewrite(new_parent)
end
end
def rewrite_award_emoji(old_awardable, new_awardable)
old_awardable.award_emoji.each do |award|
new_award = award.dup
new_award.awardable = new_awardable
new_award.save
end
end
end
end
end
# frozen_string_literal: true # frozen_string_literal: true
module Issues module Issues
class MoveService < Issues::BaseService class MoveService < Issuable::Clone::BaseService
MoveError = Class.new(StandardError) MoveError = Class.new(StandardError)
def execute(issue, new_project) def execute(issue, target_project)
@old_issue = issue @target_project = target_project
@old_project = @project
@new_project = new_project
unless issue.can_move?(current_user, new_project) unless issue.can_move?(current_user, @target_project)
raise MoveError, 'Cannot move issue due to insufficient permissions!' raise MoveError, 'Cannot move issue due to insufficient permissions!'
end end
if @project == new_project if @project == @target_project
raise MoveError, 'Cannot move issue to project it originates from!' raise MoveError, 'Cannot move issue to project it originates from!'
end end
# Using transaction because of a high resources footprint super
# on rewriting notes (unfolding references)
#
ActiveRecord::Base.transaction do
@new_issue = create_new_issue
update_new_issue
update_old_issue
end
notify_participants notify_participants
@new_issue new_entity
end end
private private
def update_new_issue def update_old_entity
rewrite_notes super
copy_resource_label_events
rewrite_issue_award_emoji
add_note_moved_from
end
def update_old_issue
add_note_moved_to
close_issue
mark_as_moved mark_as_moved
end end
def create_new_issue def create_new_entity
new_params = { id: nil, iid: nil, label_ids: cloneable_label_ids, new_params = {
milestone_id: cloneable_milestone_id, id: nil,
project: @new_project, author: @old_issue.author, iid: nil,
description: rewrite_content(@old_issue.description), project: @target_project,
assignee_ids: @old_issue.assignee_ids } author: original_entity.author,
assignee_ids: original_entity.assignee_ids
new_params = @old_issue.serializable_hash.symbolize_keys.merge(new_params)
CreateService.new(@new_project, @current_user, new_params).execute
end
# rubocop: disable CodeReuse/ActiveRecord
def cloneable_label_ids
params = {
project_id: @new_project.id,
title: @old_issue.labels.pluck(:title),
include_ancestor_groups: true
} }
LabelsFinder.new(current_user, params).execute.pluck(:id) new_params = original_entity.serializable_hash.symbolize_keys.merge(new_params)
end CreateService.new(@target_project, @current_user, new_params).execute
# rubocop: enable CodeReuse/ActiveRecord
def cloneable_milestone_id
title = @old_issue.milestone&.title
return unless title
if @new_project.group && can?(current_user, :read_group, @new_project.group)
group_id = @new_project.group.id
end
params =
{ title: title, project_ids: @new_project.id, group_ids: group_id }
milestones = MilestonesFinder.new(params).execute
milestones.first&.id
end
def rewrite_notes
@old_issue.notes_with_associations.find_each do |note|
new_note = note.dup
new_params = { project: @new_project, noteable: @new_issue,
note: rewrite_content(new_note.note),
created_at: note.created_at,
updated_at: note.updated_at }
new_note.update(new_params)
rewrite_award_emoji(note, new_note)
end
end
# rubocop: disable CodeReuse/ActiveRecord
def copy_resource_label_events
@old_issue.resource_label_events.find_in_batches do |batch|
events = batch.map do |event|
event.attributes
.except('id', 'reference', 'reference_html')
.merge('issue_id' => @new_issue.id, 'action' => ResourceLabelEvent.actions[event.action])
end end
Gitlab::Database.bulk_insert(ResourceLabelEvent.table_name, events) def mark_as_moved
end original_entity.update(moved_to: new_entity)
end
# rubocop: enable CodeReuse/ActiveRecord
def rewrite_issue_award_emoji
rewrite_award_emoji(@old_issue, @new_issue)
end
def rewrite_award_emoji(old_awardable, new_awardable)
old_awardable.award_emoji.each do |award|
new_award = award.dup
new_award.awardable = new_awardable
new_award.save
end
end
def rewrite_content(content)
return unless content
rewriters = [Gitlab::Gfm::ReferenceRewriter,
Gitlab::Gfm::UploadsRewriter]
rewriters.inject(content) do |text, klass|
rewriter = klass.new(text, @old_project, @current_user)
rewriter.rewrite(@new_project)
end
end end
def close_issue def notify_participants
close_service = CloseService.new(@old_project, @current_user) notification_service.async.issue_moved(original_entity, new_entity, @current_user)
close_service.execute(@old_issue, notifications: false, system_note: false)
end end
def add_note_moved_from def add_note_from
SystemNoteService.noteable_moved(@new_issue, @new_project, SystemNoteService.noteable_moved(new_entity, @target_project,
@old_issue, @current_user, original_entity, current_user,
direction: :from) direction: :from)
end end
def add_note_moved_to def add_note_to
SystemNoteService.noteable_moved(@old_issue, @old_project, SystemNoteService.noteable_moved(original_entity, old_project,
@new_issue, @current_user, new_entity, current_user,
direction: :to) direction: :to)
end end
def mark_as_moved
@old_issue.update(moved_to: @new_issue)
end
def notify_participants
notification_service.async.issue_moved(@old_issue, @new_issue, @current_user)
end
end end
end end
...@@ -149,9 +149,9 @@ class FileUploader < GitlabUploader ...@@ -149,9 +149,9 @@ class FileUploader < GitlabUploader
# return a new uploader with a file copy on another project # return a new uploader with a file copy on another project
def self.copy_to(uploader, to_project) def self.copy_to(uploader, to_project)
moved = uploader.dup.tap do |u| moved = self.new(to_project)
u.model = to_project moved.object_store = uploader.object_store
end moved.filename = uploader.filename
moved.copy_file(uploader.file) moved.copy_file(uploader.file)
moved moved
......
...@@ -31,19 +31,19 @@ module Gitlab ...@@ -31,19 +31,19 @@ module Gitlab
class ReferenceRewriter class ReferenceRewriter
RewriteError = Class.new(StandardError) RewriteError = Class.new(StandardError)
def initialize(text, source_project, current_user) def initialize(text, source_parent, current_user)
@text = text @text = text
@source_project = source_project @source_parent = source_parent
@current_user = current_user @current_user = current_user
@original_html = markdown(text) @original_html = markdown(text)
@pattern = Gitlab::ReferenceExtractor.references_pattern @pattern = Gitlab::ReferenceExtractor.references_pattern
end end
def rewrite(target_project) def rewrite(target_parent)
return @text unless needs_rewrite? return @text unless needs_rewrite?
@text.gsub(@pattern) do |reference| @text.gsub(@pattern) do |reference|
unfold_reference(reference, Regexp.last_match, target_project) unfold_reference(reference, Regexp.last_match, target_parent)
end end
end end
...@@ -53,14 +53,14 @@ module Gitlab ...@@ -53,14 +53,14 @@ module Gitlab
private private
def unfold_reference(reference, match, target_project) def unfold_reference(reference, match, target_parent)
before = @text[0...match.begin(0)] before = @text[0...match.begin(0)]
after = @text[match.end(0)..-1] after = @text[match.end(0)..-1]
referable = find_referable(reference) referable = find_referable(reference)
return reference unless referable return reference unless referable
cross_reference = build_cross_reference(referable, target_project) cross_reference = build_cross_reference(referable, target_parent)
return reference if reference == cross_reference return reference if reference == cross_reference
if cross_reference.nil? if cross_reference.nil?
...@@ -72,17 +72,17 @@ module Gitlab ...@@ -72,17 +72,17 @@ module Gitlab
end end
def find_referable(reference) def find_referable(reference)
extractor = Gitlab::ReferenceExtractor.new(@source_project, extractor = Gitlab::ReferenceExtractor.new(@source_parent,
@current_user) @current_user)
extractor.analyze(reference) extractor.analyze(reference)
extractor.all.first extractor.all.first
end end
def build_cross_reference(referable, target_project) def build_cross_reference(referable, target_parent)
if referable.respond_to?(:project) if referable.respond_to?(:project)
referable.to_reference(target_project) referable.to_reference(target_parent)
else else
referable.to_reference(@source_project, target_project: target_project) referable.to_reference(@source_parent, target_project: target_parent)
end end
end end
...@@ -91,7 +91,7 @@ module Gitlab ...@@ -91,7 +91,7 @@ module Gitlab
end end
def markdown(text) def markdown(text)
Banzai.render(text, project: @source_project, no_original_data: true) Banzai.render(text, project: @source_parent, no_original_data: true)
end end
end end
end end
......
...@@ -16,14 +16,15 @@ module Gitlab ...@@ -16,14 +16,15 @@ module Gitlab
@pattern = FileUploader::MARKDOWN_PATTERN @pattern = FileUploader::MARKDOWN_PATTERN
end end
def rewrite(target_project) def rewrite(target_parent)
return @text unless needs_rewrite? return @text unless needs_rewrite?
@text.gsub(@pattern) do |markdown| @text.gsub(@pattern) do |markdown|
file = find_file(@source_project, $~[:secret], $~[:file]) file = find_file(@source_project, $~[:secret], $~[:file])
break markdown unless file.try(:exists?) break markdown unless file.try(:exists?)
moved = FileUploader.copy_to(file, target_project) klass = target_parent.is_a?(Namespace) ? NamespaceFileUploader : FileUploader
moved = klass.copy_to(file, target_parent)
moved.markdown_link moved.markdown_link
end end
end end
......
...@@ -2,13 +2,14 @@ module Gitlab ...@@ -2,13 +2,14 @@ module Gitlab
module QuickActions module QuickActions
class CommandDefinition class CommandDefinition
attr_accessor :name, :aliases, :description, :explanation, :params, attr_accessor :name, :aliases, :description, :explanation, :params,
:condition_block, :parse_params_block, :action_block :condition_block, :parse_params_block, :action_block, :warning
def initialize(name, attributes = {}) def initialize(name, attributes = {})
@name = name @name = name
@aliases = attributes[:aliases] || [] @aliases = attributes[:aliases] || []
@description = attributes[:description] || '' @description = attributes[:description] || ''
@warning = attributes[:warning] || ''
@explanation = attributes[:explanation] || '' @explanation = attributes[:explanation] || ''
@params = attributes[:params] || [] @params = attributes[:params] || []
@condition_block = attributes[:condition_block] @condition_block = attributes[:condition_block]
...@@ -33,11 +34,13 @@ module Gitlab ...@@ -33,11 +34,13 @@ module Gitlab
def explain(context, arg) def explain(context, arg)
return unless available?(context) return unless available?(context)
if explanation.respond_to?(:call) message = if explanation.respond_to?(:call)
execute_block(explanation, context, arg) execute_block(explanation, context, arg)
else else
explanation explanation
end end
warning.empty? ? message : "#{message} (#{warning})"
end end
def execute(context, arg) def execute(context, arg)
...@@ -61,6 +64,7 @@ module Gitlab ...@@ -61,6 +64,7 @@ module Gitlab
name: name, name: name,
aliases: aliases, aliases: aliases,
description: desc, description: desc,
warning: warning,
params: prms params: prms
} }
end end
......
...@@ -31,6 +31,10 @@ module Gitlab ...@@ -31,6 +31,10 @@ module Gitlab
@description = block_given? ? block : text @description = block_given? ? block : text
end end
def warning(message = '')
@warning = message
end
# Allows to define params for the next quick action. # Allows to define params for the next quick action.
# These params are shown in the autocomplete menu. # These params are shown in the autocomplete menu.
# #
...@@ -133,6 +137,7 @@ module Gitlab ...@@ -133,6 +137,7 @@ module Gitlab
name, name,
aliases: aliases, aliases: aliases,
description: @description, description: @description,
warning: @warning,
explanation: @explanation, explanation: @explanation,
params: @params, params: @params,
condition_block: @condition_block, condition_block: @condition_block,
...@@ -150,6 +155,7 @@ module Gitlab ...@@ -150,6 +155,7 @@ module Gitlab
@explanation = nil @explanation = nil
@params = nil @params = nil
@condition_block = nil @condition_block = nil
@warning = nil
@parse_params_block = nil @parse_params_block = nil
end end
end end
......
...@@ -210,6 +210,19 @@ describe Gitlab::QuickActions::CommandDefinition do ...@@ -210,6 +210,19 @@ describe Gitlab::QuickActions::CommandDefinition do
end end
end end
context 'when warning is set' do
before do
subject.explanation = 'Explanation'
subject.warning = 'dangerous!'
end
it 'returns this static string' do
result = subject.explain({}, nil)
expect(result).to eq 'Explanation (dangerous!)'
end
end
context 'when the explanation is dynamic' do context 'when the explanation is dynamic' do
before do before do
subject.explanation = proc { |arg| "Dynamic #{arg}" } subject.explanation = proc { |arg| "Dynamic #{arg}" }
......
...@@ -12,6 +12,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -12,6 +12,7 @@ describe Gitlab::QuickActions::Dsl do
params 'The first argument' params 'The first argument'
explanation 'Static explanation' explanation 'Static explanation'
warning 'Possible problem!'
command :explanation_with_aliases, :once, :first do |arg| command :explanation_with_aliases, :once, :first do |arg|
arg arg
end end
...@@ -64,6 +65,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -64,6 +65,7 @@ describe Gitlab::QuickActions::Dsl do
expect(no_args_def.condition_block).to be_nil expect(no_args_def.condition_block).to be_nil
expect(no_args_def.action_block).to be_a_kind_of(Proc) expect(no_args_def.action_block).to be_a_kind_of(Proc)
expect(no_args_def.parse_params_block).to be_nil expect(no_args_def.parse_params_block).to be_nil
expect(no_args_def.warning).to eq('')
expect(explanation_with_aliases_def.name).to eq(:explanation_with_aliases) expect(explanation_with_aliases_def.name).to eq(:explanation_with_aliases)
expect(explanation_with_aliases_def.aliases).to eq([:once, :first]) expect(explanation_with_aliases_def.aliases).to eq([:once, :first])
...@@ -73,6 +75,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -73,6 +75,7 @@ describe Gitlab::QuickActions::Dsl do
expect(explanation_with_aliases_def.condition_block).to be_nil expect(explanation_with_aliases_def.condition_block).to be_nil
expect(explanation_with_aliases_def.action_block).to be_a_kind_of(Proc) expect(explanation_with_aliases_def.action_block).to be_a_kind_of(Proc)
expect(explanation_with_aliases_def.parse_params_block).to be_nil expect(explanation_with_aliases_def.parse_params_block).to be_nil
expect(explanation_with_aliases_def.warning).to eq('Possible problem!')
expect(dynamic_description_def.name).to eq(:dynamic_description) expect(dynamic_description_def.name).to eq(:dynamic_description)
expect(dynamic_description_def.aliases).to eq([]) expect(dynamic_description_def.aliases).to eq([])
...@@ -82,6 +85,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -82,6 +85,7 @@ describe Gitlab::QuickActions::Dsl do
expect(dynamic_description_def.condition_block).to be_nil expect(dynamic_description_def.condition_block).to be_nil
expect(dynamic_description_def.action_block).to be_a_kind_of(Proc) expect(dynamic_description_def.action_block).to be_a_kind_of(Proc)
expect(dynamic_description_def.parse_params_block).to be_nil expect(dynamic_description_def.parse_params_block).to be_nil
expect(dynamic_description_def.warning).to eq('')
expect(cc_def.name).to eq(:cc) expect(cc_def.name).to eq(:cc)
expect(cc_def.aliases).to eq([]) expect(cc_def.aliases).to eq([])
...@@ -91,6 +95,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -91,6 +95,7 @@ describe Gitlab::QuickActions::Dsl do
expect(cc_def.condition_block).to be_nil expect(cc_def.condition_block).to be_nil
expect(cc_def.action_block).to be_nil expect(cc_def.action_block).to be_nil
expect(cc_def.parse_params_block).to be_nil expect(cc_def.parse_params_block).to be_nil
expect(cc_def.warning).to eq('')
expect(cond_action_def.name).to eq(:cond_action) expect(cond_action_def.name).to eq(:cond_action)
expect(cond_action_def.aliases).to eq([]) expect(cond_action_def.aliases).to eq([])
...@@ -100,6 +105,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -100,6 +105,7 @@ describe Gitlab::QuickActions::Dsl do
expect(cond_action_def.condition_block).to be_a_kind_of(Proc) expect(cond_action_def.condition_block).to be_a_kind_of(Proc)
expect(cond_action_def.action_block).to be_a_kind_of(Proc) expect(cond_action_def.action_block).to be_a_kind_of(Proc)
expect(cond_action_def.parse_params_block).to be_nil expect(cond_action_def.parse_params_block).to be_nil
expect(cond_action_def.warning).to eq('')
expect(with_params_parsing_def.name).to eq(:with_params_parsing) expect(with_params_parsing_def.name).to eq(:with_params_parsing)
expect(with_params_parsing_def.aliases).to eq([]) expect(with_params_parsing_def.aliases).to eq([])
...@@ -109,6 +115,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -109,6 +115,7 @@ describe Gitlab::QuickActions::Dsl do
expect(with_params_parsing_def.condition_block).to be_nil expect(with_params_parsing_def.condition_block).to be_nil
expect(with_params_parsing_def.action_block).to be_a_kind_of(Proc) expect(with_params_parsing_def.action_block).to be_a_kind_of(Proc)
expect(with_params_parsing_def.parse_params_block).to be_a_kind_of(Proc) expect(with_params_parsing_def.parse_params_block).to be_a_kind_of(Proc)
expect(with_params_parsing_def.warning).to eq('')
expect(substitution_def.name).to eq(:something) expect(substitution_def.name).to eq(:something)
expect(substitution_def.aliases).to eq([]) expect(substitution_def.aliases).to eq([])
...@@ -118,6 +125,7 @@ describe Gitlab::QuickActions::Dsl do ...@@ -118,6 +125,7 @@ describe Gitlab::QuickActions::Dsl do
expect(substitution_def.condition_block).to be_nil expect(substitution_def.condition_block).to be_nil
expect(substitution_def.action_block.call('text')).to eq('text Some complicated thing you want in here') expect(substitution_def.action_block.call('text')).to eq('text Some complicated thing you want in here')
expect(substitution_def.parse_params_block).to be_nil expect(substitution_def.parse_params_block).to be_nil
expect(substitution_def.warning).to eq('')
end end
end end
end end
# frozen_string_literal: true
require 'spec_helper'
describe Issuable::Clone::AttributesRewriter do
let(:user) { create(:user) }
let(:group) { create(:group) }
let(:project1) { create(:project, :public, group: group) }
let(:project2) { create(:project, :public, group: group) }
let(:original_issue) { create(:issue, project: project1) }
let(:new_issue) { create(:issue, project: project2) }
subject { described_class.new(user, original_issue, new_issue) }
context 'setting labels' do
it 'sets labels present in the new project and group labels' do
project1_label_1 = create(:label, title: 'label1', project: project1)
project1_label_2 = create(:label, title: 'label2', project: project1)
project2_label_1 = create(:label, title: 'label1', project: project2)
group_label = create(:group_label, title: 'group_label', group: group)
create(:label, title: 'label3', project: project2)
original_issue.update(labels: [project1_label_1, project1_label_2, group_label])
subject.execute
expect(new_issue.reload.labels).to match_array([project2_label_1, group_label])
end
it 'does not set any labels when not used on the original issue' do
subject.execute
expect(new_issue.reload.labels).to be_empty
end
it 'copies the resource label events' do
resource_label_events = create_list(:resource_label_event, 2, issue: original_issue)
subject.execute
expected = resource_label_events.map(&:label_id)
expect(new_issue.resource_label_events.map(&:label_id)).to match_array(expected)
end
end
context 'setting milestones' do
it 'sets milestone to nil when old issue milestone is not in the new project' do
milestone = create(:milestone, title: 'milestone', project: project1)
original_issue.update(milestone: milestone)
subject.execute
expect(new_issue.reload.milestone).to be_nil
end
it 'copies the milestone when old issue milestone title is in the new project' do
milestone_project1 = create(:milestone, title: 'milestone', project: project1)
milestone_project2 = create(:milestone, title: 'milestone', project: project2)
original_issue.update(milestone: milestone_project1)
subject.execute
expect(new_issue.reload.milestone).to eq(milestone_project2)
end
it 'copies the milestone when old issue milestone is a group milestone' do
milestone = create(:milestone, title: 'milestone', group: group)
original_issue.update(milestone: milestone)
subject.execute
expect(new_issue.reload.milestone).to eq(milestone)
end
end
end
# frozen_string_literal: true
require 'spec_helper'
describe Issuable::Clone::ContentRewriter do
let(:user) { create(:user) }
let(:group) { create(:group) }
let(:project1) { create(:project, :public, group: group) }
let(:project2) { create(:project, :public, group: group) }
let(:other_issue) { create(:issue, project: project1) }
let(:merge_request) { create(:merge_request) }
subject { described_class.new(user, original_issue, new_issue)}
let(:description) { 'Simple text' }
let(:original_issue) { create(:issue, description: description, project: project1) }
let(:new_issue) { create(:issue, project: project2) }
context 'rewriting award emojis' do
it 'copies the award emojis' do
create(:award_emoji, awardable: original_issue, name: 'thumbsup')
create(:award_emoji, awardable: original_issue, name: 'thumbsdown')
expect { subject.execute }.to change { AwardEmoji.count }.by(2)
expect(new_issue.award_emoji.map(&:name)).to match_array(%w(thumbsup thumbsdown))
end
end
context 'rewriting description' do
before do
subject.execute
end
context 'when description is a simple text' do
it 'does not rewrite the description' do
expect(new_issue.reload.description).to eq(original_issue.description)
end
end
context 'when description contains a local reference' do
let(:description) { "See ##{other_issue.iid}" }
it 'rewrites the local reference correctly' do
expected_description = "See #{project1.path}##{other_issue.iid}"
expect(new_issue.reload.description).to eq(expected_description)
end
end
context 'when description contains a cross reference' do
let(:description) { "See #{merge_request.project.full_path}!#{merge_request.iid}" }
it 'rewrites the cross reference correctly' do
expected_description = "See #{merge_request.project.full_path}!#{merge_request.iid}"
expect(new_issue.reload.description).to eq(expected_description)
end
end
context 'when description contains a user reference' do
let(:description) { "FYU #{user.to_reference}" }
it 'works with a user reference' do
expect(new_issue.reload.description).to eq("FYU #{user.to_reference}")
end
end
context 'when description contains uploads' do
let(:uploader) { build(:file_uploader, project: project1) }
let(:description) { "Text and #{uploader.markdown_link}" }
it 'rewrites uploads in the description' do
upload = Upload.last
expect(new_issue.description).not_to eq(description)
expect(new_issue.description).to match(/Text and #{FileUploader::MARKDOWN_PATTERN}/)
expect(upload.secret).not_to eq(uploader.secret)
expect(new_issue.description).to include(upload.secret)
expect(new_issue.description).to include(upload.path)
end
end
end
context 'rewriting notes' do
context 'simple notes' do
let!(:notes) do
[
create(:note, noteable: original_issue, project: project1,
created_at: 2.weeks.ago, updated_at: 1.week.ago),
create(:note, noteable: original_issue, project: project1),
create(:note, system: true, noteable: original_issue, project: project1)
]
end
let!(:system_note_metadata) { create(:system_note_metadata, note: notes.last) }
let!(:award_emoji) { create(:award_emoji, awardable: notes.first, name: 'thumbsup')}
before do
subject.execute
end
it 'rewrites existing notes in valid order' do
expect(new_issue.notes.order('id ASC').pluck(:note).first(3)).to eq(notes.map(&:note))
end
it 'copies all the issue notes' do
expect(new_issue.notes.count).to eq(3)
end
it 'does not change the note attributes' do
subject.execute
new_note = new_issue.notes.first
expect(new_note.note).to eq(notes.first.note)
expect(new_note.author).to eq(notes.first.author)
end
it 'copies the award emojis' do
subject.execute
new_note = new_issue.notes.first
new_note.award_emoji.first.name = 'thumbsup'
end
it 'copies system_note_metadata for system note' do
new_note = new_issue.notes.last
expect(new_note.system_note_metadata.action).to eq(system_note_metadata.action)
expect(new_note.system_note_metadata.id).not_to eq(system_note_metadata.id)
end
end
context 'notes with reference' do
let(:text) do
"See ##{other_issue.iid} and #{merge_request.project.full_path}!#{merge_request.iid}"
end
let!(:note) { create(:note, noteable: original_issue, note: text, project: project1) }
it 'rewrites the references correctly' do
subject.execute
new_note = new_issue.notes.first
expected_text = "See #{other_issue.project.path}##{other_issue.iid} and #{merge_request.project.full_path}!#{merge_request.iid}"
expect(new_note.note).to eq(expected_text)
expect(new_note.author).to eq(note.author)
end
end
end
end
This diff is collapsed.
...@@ -81,19 +81,24 @@ describe FileUploader do ...@@ -81,19 +81,24 @@ describe FileUploader do
end end
describe 'copy_to' do describe 'copy_to' do
shared_examples 'returns a valid uploader' do
describe 'returned uploader' do
let(:new_project) { create(:project) } let(:new_project) { create(:project) }
let(:moved) { described_class.copy_to(subject, new_project) } let(:moved) { described_class.copy_to(subject, new_project) }
shared_examples 'returns a valid uploader' do
describe 'returned uploader' do
it 'generates a new secret' do it 'generates a new secret' do
expect(subject).to be expect(subject).to be
expect(described_class).to receive(:generate_secret).once.and_call_original expect(described_class).to receive(:generate_secret).once.and_call_original
expect(moved).to be expect(moved).to be
end end
it 'create new upload' do it 'creates new upload correctly' do
expect(moved.upload).not_to eq(subject.upload) upload = moved.upload
expect(upload).not_to eq(subject.upload)
expect(upload.model).to eq(new_project)
expect(upload.uploader).to eq('FileUploader')
expect(upload.secret).not_to eq(subject.upload.secret)
end end
it 'copies the file' do it 'copies the file' do
...@@ -111,6 +116,12 @@ describe FileUploader do ...@@ -111,6 +116,12 @@ describe FileUploader do
end end
include_examples 'returns a valid uploader' include_examples 'returns a valid uploader'
it 'copies the file to the correct location' do
expect(moved.upload.path).to eq("#{moved.upload.secret}/dk.png")
expect(moved.file.path).to end_with("public/uploads/#{new_project.disk_path}/#{moved.upload.secret}/dk.png")
expect(moved.filename).to eq('dk.png')
end
end end
context 'files are stored remotely' do context 'files are stored remotely' do
...@@ -121,6 +132,12 @@ describe FileUploader do ...@@ -121,6 +132,12 @@ describe FileUploader do
end end
include_examples 'returns a valid uploader' include_examples 'returns a valid uploader'
it 'copies the file to the correct location' do
expect(moved.upload.path).to eq("#{new_project.disk_path}/#{moved.upload.secret}/dk.png")
expect(moved.file.path).to eq("#{new_project.disk_path}/#{moved.upload.secret}/dk.png")
expect(moved.filename).to eq('dk.png')
end
end end
end end
......
...@@ -55,4 +55,62 @@ describe NamespaceFileUploader do ...@@ -55,4 +55,62 @@ describe NamespaceFileUploader do
it_behaves_like "migrates", to_store: described_class::Store::REMOTE it_behaves_like "migrates", to_store: described_class::Store::REMOTE
it_behaves_like "migrates", from_store: described_class::Store::REMOTE, to_store: described_class::Store::LOCAL it_behaves_like "migrates", from_store: described_class::Store::REMOTE, to_store: described_class::Store::LOCAL
end end
describe 'copy_to' do
let(:group) { create(:group) }
let(:moved) { described_class.copy_to(subject, group) }
shared_examples 'returns a valid uploader' do
it 'generates a new secret' do
expect(subject).to be
expect(described_class).to receive(:generate_secret).once.and_call_original
expect(moved).to be
end
it 'creates new upload correctly' do
upload = moved.upload
expect(upload).not_to eq(subject.upload)
expect(upload.model).to eq(group)
expect(upload.uploader).to eq('NamespaceFileUploader')
expect(upload.secret).not_to eq(subject.upload.secret)
end
it 'copies the file' do
expect(subject.file).to exist
expect(moved.file).to exist
expect(subject.file).not_to eq(moved.file)
expect(subject.object_store).to eq(moved.object_store)
end
end
context 'files are stored locally' do
before do
subject.store!(fixture_file_upload('spec/fixtures/dk.png'))
end
include_examples 'returns a valid uploader'
it 'copies the file to the correct location' do
expect(moved.upload.path).to eq("#{moved.upload.secret}/dk.png")
expect(moved.file.path).to end_with("system/namespace/#{group.id}/#{moved.upload.secret}/dk.png")
expect(moved.filename).to eq('dk.png')
end
end
context 'files are stored remotely' do
before do
stub_uploads_object_storage
subject.store!(fixture_file_upload('spec/fixtures/dk.png'))
subject.migrate!(ObjectStorage::Store::REMOTE)
end
include_examples 'returns a valid uploader'
it 'copies the file to the correct location' do
expect(moved.file.path).to eq("namespace/#{group.id}/#{moved.upload.secret}/dk.png")
expect(moved.filename).to eq('dk.png')
end
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