Commit a02e2243 authored by Jacob Vosmaer's avatar Jacob Vosmaer Committed by Stan Hu

Fix wrong use of ActiveRecord in PoolRepository

parent 9c2b37d7
...@@ -7,7 +7,7 @@ class PoolRepository < ApplicationRecord ...@@ -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 ...@@ -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
......
...@@ -2131,13 +2131,11 @@ class Project < ApplicationRecord ...@@ -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
......
---
title: Fix wrong use of ActiveRecord in PoolRepository
merge_request: 27464
author:
type: fixed
...@@ -5,6 +5,7 @@ FactoryBot.define do ...@@ -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
......
...@@ -5,7 +5,7 @@ require 'spec_helper' ...@@ -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
......
...@@ -4702,6 +4702,8 @@ describe Project do ...@@ -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
......
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