Fix wrong use of ActiveRecord in PoolRepository

This commit is contained in:
Jacob Vosmaer 2019-04-20 13:27:53 +00:00 committed by Stan Hu
parent 9c2b37d7c4
commit a02e22438d
6 changed files with 16 additions and 8 deletions

View file

@ -7,7 +7,7 @@ class PoolRepository < ApplicationRecord
include Shardable include Shardable
include AfterCommitQueue include AfterCommitQueue
has_one :source_project, class_name: 'Project' belongs_to :source_project, class_name: 'Project'
validates :source_project, presence: true validates :source_project, presence: true
has_many :member_projects, class_name: 'Project' has_many :member_projects, class_name: 'Project'
@ -99,7 +99,8 @@ class PoolRepository < ApplicationRecord
end end
def inspect def inspect
"#<#{self.class.name} id:#{id} state:#{state} disk_path:#{disk_path} source_project: #{source_project.full_path}>" source = source_project ? source_project.full_path : 'nil'
"#<#{self.class.name} id:#{id} state:#{state} disk_path:#{disk_path} source_project: #{source}>"
end end
private private

View file

@ -2131,13 +2131,11 @@ class Project < ApplicationRecord
end end
def create_new_pool_repository def create_new_pool_repository
pool = begin pool = PoolRepository.safe_find_or_create_by!(shard: Shard.by_name(repository_storage), source_project: self)
create_pool_repository!(shard: Shard.by_name(repository_storage), source_project: self) update!(pool_repository: pool)
rescue ActiveRecord::RecordNotUnique
pool_repository(true)
end
pool.schedule unless pool.scheduled? pool.schedule unless pool.scheduled?
pool pool
end end

View file

@ -0,0 +1,6 @@
---
title: Fix wrong use of ActiveRecord in PoolRepository
merge_request: 27464
author:
type: fixed

View file

@ -5,6 +5,7 @@ FactoryBot.define do
before(:create) do |pool| before(:create) do |pool|
pool.source_project = create(:project, :repository) pool.source_project = create(:project, :repository)
pool.source_project.update!(pool_repository: pool)
end end
trait :scheduled do trait :scheduled do

View file

@ -5,7 +5,7 @@ require 'spec_helper'
describe PoolRepository do describe PoolRepository do
describe 'associations' do describe 'associations' do
it { is_expected.to belong_to(:shard) } it { is_expected.to belong_to(:shard) }
it { is_expected.to have_one(:source_project) } it { is_expected.to belong_to(:source_project) }
it { is_expected.to have_many(:member_projects) } it { is_expected.to have_many(:member_projects) }
end end

View file

@ -4702,6 +4702,8 @@ describe Project do
it 'returns that pool repository' do it 'returns that pool repository' do
expect(subject).not_to be_empty expect(subject).not_to be_empty
expect(subject[:pool_repository]).to be_persisted expect(subject[:pool_repository]).to be_persisted
expect(project.reload.pool_repository).to eq(subject[:pool_repository])
end end
end end
end end