Adding convenience method to project model

2 files changed, 19 insertions, 0 deletions

diff --git a/app/models/project.rb b/app/models/project.rb
index 58df4019450..c72d3a3b725 100644
--- a/app/models/project.rb
+++ b/app/models/project.rb
@@ -2073,6 +2073,10 @@ class Project < ActiveRecord::Base
     pool_repository&.link_repository(repository)
   end
 
+  def has_pool_repository?
+    pool_repository.present?
+  end
+
   private
 
   def merge_requests_allowing_collaboration(source_branch = nil)
diff --git a/spec/models/project_spec.rb b/spec/models/project_spec.rb
index 7b364395faf..1f9088c2e6b 100644
--- a/spec/models/project_spec.rb
+++ b/spec/models/project_spec.rb
@@ -4602,6 +4602,21 @@ describe Project do
     end
   end
 
+  describe '#has_pool_repsitory?' do
+    it 'returns false when it does not have a pool repository' do
+      subject = create(:project, :repository)
+
+      expect(subject.has_pool_repository?).to be false
+    end
+
+    it 'returns true when it has a pool repository' do
+      pool    = create(:pool_repository, :ready)
+      subject = create(:project, :repository, pool_repository: pool)
+
+      expect(subject.has_pool_repository?).to be true
+    end
+  end
+
   def rugged_config
     rugged_repo(project.repository).config
   end