Commit 4eebd231 authored by James Lopez's avatar James Lopez

update code based on feedback

parent 79208294
......@@ -19,38 +19,27 @@ module Gitlab
def initialize(klass, attributes)
@klass = klass < Label ? Label : klass
@attributes = attributes
@group = @attributes[:group]
@project = @attributes[:project]
@group = @attributes['group']
@project = @attributes['project']
end
def find
find_or_action do
label? ? @klass.new(project_attributes) : @klass.create(project_attributes)
end
find_object || @klass.create(project_attributes)
end
private
def find_or_action(&block)
@klass.where(where_clause).first || yield
def find_object
@klass.where(where_clause).first
end
def where_clause
@attributes.slice('title').map do |key, value|
if @group
project_group_clause(key, value)
else
project_clause(key, value)
end
end.reduce(:or)
end
scope_clause = table[:project_id].eq(@project.id)
scope_clause = scope_clause.or(table[:group_id].eq(@group.id)) if @group
def project_group_clause(key, value)
table[key].eq(value).and(table[:project_id].eq(@project.id).or(table[:group_id].eq(@group.id)))
end
def project_clause(key, value)
table[key].eq(value).and(table[:project_id].eq(@project.id))
table[key].eq(value).and(scope_clause)
end.reduce(:or)
end
def table
......@@ -58,7 +47,7 @@ module Gitlab
end
def project_attributes
@attributes.except(:group).tap do |atts|
@attributes.except('group').tap do |atts|
if label?
atts['type'] = 'ProjectLabel' # Always create project labels
elsif milestone?
......@@ -80,27 +69,21 @@ module Gitlab
@klass == Milestone
end
# If an existing group milesone used the IID
# If an existing group milestone used the IID
# claim the IID back and set the group milestone to use one available
# This is neccessary to fix situations like the following:
# This is necessary to fix situations like the following:
# - Importing into a user namespace project with exported group milestones
# where the IID of the Group milestone could conflict with a project one.
def claim_iid
# The milestone has to be a group milestone, as it's the only case where
# we set the IID as the maximum. The rest of them are fixed.
group_milestone = @project.milestones.find_by(iid: @attributes['iid'])
group_milestone.update!(iid: max_milestone_iid(group_milestone)) if group_milestone
end
milestone = @project.milestones.find_by(iid: @attributes['iid'])
def max_milestone_iid(group_milestone)
init_iid = [@attributes['iid'], @project.milestones.maximum(:iid)].compact.max + 1
return unless milestone
InternalId::InternalIdGenerator.new(group_milestone,
{ project: @project },
:milestones,
init_iid
).generate
milestone.iid = nil
milestone.ensure_project_iid!
milestone.save!
end
end
end
......
......@@ -267,8 +267,8 @@ module Gitlab
# Can't use IDs as validation exists calling `group` or `project` attributes
finder_hash = parsed_relation_hash.tap do |hash|
hash[:group] = @project.group if relation_class.attribute_method?('group_id')
hash[:project] = @project
hash['group'] = @project.group if relation_class.attribute_method?('group_id')
hash['project'] = @project
hash.delete('project_id')
end
......
......@@ -15,18 +15,18 @@ describe Gitlab::ImportExport::GroupProjectObjectBuilder do
group_label = create(:group_label, 'name': 'group label', 'group': project.group)
expect(described_class.build(Label,
title: 'group label',
project: project,
group: project.group)).to eq(group_label)
'title' => 'group label',
'project' => project,
'group' => project.group)).to eq(group_label)
end
it 'initializes a new label' do
it 'creates a new label' do
label = described_class.build(Label,
title: 'group label',
project: project,
group: project.group)
'title' => 'group label',
'project' => project,
'group' => project.group)
expect(label.persisted?).to be false
expect(label.persisted?).to be true
end
end
......@@ -35,16 +35,16 @@ describe Gitlab::ImportExport::GroupProjectObjectBuilder do
milestone = create(:milestone, 'name' => 'group milestone', 'group' => project.group)
expect(described_class.build(Milestone,
title: 'group milestone',
project: project,
group: project.group)).to eq(milestone)
'title' => 'group milestone',
'project' => project,
'group' => project.group)).to eq(milestone)
end
it 'creates a new milestone' do
milestone = described_class.build(Milestone,
title: 'group milestone',
project: project,
group: project.group)
'title' => 'group milestone',
'project' => project,
'group' => project.group)
expect(milestone.persisted?).to be true
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