diff --git a/app/models/katello/content_view_version.rb b/app/models/katello/content_view_version.rb index ffb222787fc..8bf99ef8143 100644 --- a/app/models/katello/content_view_version.rb +++ b/app/models/katello/content_view_version.rb @@ -167,6 +167,10 @@ def promote_puppet_environment? (!content_counts.blank? && content_counts.dig(PuppetModule::CONTENT_TYPE) > 0) || self.content_view.force_puppet_environment? end + def puppet_module_count + content_counts.dig(PuppetModule::CONTENT_TYPE) if content_counts? + end + def archived_repos self.default? ? self.repositories : self.repos(nil) end diff --git a/test/models/content_view_version_test.rb b/test/models/content_view_version_test.rb index c93ddcbbae2..6f8af3ba3b0 100644 --- a/test/models/content_view_version_test.rb +++ b/test/models/content_view_version_test.rb @@ -196,7 +196,6 @@ def test_with_puppet_module katello_content_view_puppet_environments(:library_dev_staging_view_library_puppet_env).puppet_modules << puppet_module puppet_cv_env = katello_content_view_puppet_environments(:dev_view_puppet_environment) puppet_cv_env.puppet_modules << puppet_module - assert_include ContentViewVersion.with_puppet_module(puppet_module), puppet_cv_env.content_view_version end @@ -208,6 +207,7 @@ def test_promote_puppet_environment? @cvv.content_view.force_puppet_environment = false @cvv.content_counts = { "puppet_module" => 2 } + assert_equal 2, @cvv.puppet_module_count assert @cvv.promote_puppet_environment? end