Fix capsule download policy update

robottelo/hosts.py

@@ -1656,6 +1656,12 @@ def capsule_setup(self, sat_host=None, capsule_cert_opts=None, **installer_kwarg
                 f'A core service is not running at capsule host\n{result.stdout}'
             )
 
+    def update_download_policy(self, policy):
+        """Updates capsule's download policy to desired value"""
+        proxy = self.nailgun_smart_proxy.read()
+        proxy.download_policy = policy
+        proxy.update(['download_policy'])
+
     def set_rex_script_mode_provider(self, mode='ssh'):
         """Set provider for remote execution script mode. One of: ssh(default),
         pull-mqtt, ssh-async"""

tests/foreman/api/test_capsulecontent.py

@@ -43,16 +43,6 @@ class TestCapsuleContentManagement:
     interactions and use capsule.
     """
 
-    def update_capsule_download_policy(
-        self, module_capsule_configured, download_policy, module_target_sat
-    ):
-        """Updates capsule's download policy to desired value"""
-        proxy = module_target_sat.api.SmartProxy(
-            id=module_capsule_configured.nailgun_capsule.id
-        ).read()
-        proxy.download_policy = download_policy
-        proxy.update(['download_policy'])
-
     @pytest.mark.tier3
     @pytest.mark.skip_if_not_set('capsule', 'clients', 'fake_manifest')
     def test_positive_insights_puppet_package_availability(self, module_capsule_configured):
@@ -595,7 +585,7 @@ def test_positive_on_demand_sync(
         assert function_lce.id in [capsule_lce['id'] for capsule_lce in result['results']]
 
         # Update capsule's download policy to on_demand
-        self.update_capsule_download_policy(module_capsule_configured, 'on_demand')
+        module_capsule_configured.update_download_policy('on_demand')
 
         # Create a content view with the repository
         cv = target_sat.api.ContentView(organization=function_org, repository=[repo]).create()
@@ -669,7 +659,7 @@ def test_positive_update_with_immediate_sync(
             url=repo_url,
         ).create()
         # Update capsule's download policy to on_demand to match repository's policy
-        self.update_capsule_download_policy(module_capsule_configured, 'on_demand')
+        module_capsule_configured.update_download_policy('on_demand')
         # Associate the lifecycle environment with the capsule
         module_capsule_configured.nailgun_capsule.content_add_lifecycle_environment(
             data={'environment_id': function_lce.id}
@@ -706,7 +696,7 @@ def test_positive_update_with_immediate_sync(
         assert repo.download_policy == 'immediate'
 
         # Update capsule's download policy as well
-        self.update_capsule_download_policy(module_capsule_configured, 'immediate')
+        module_capsule_configured.update_download_policy('immediate')
 
         # Sync repository once again
         repo.sync()
@@ -808,7 +798,7 @@ def test_positive_sync_kickstart_repo(
         assert lce.id in [capsule_lce['id'] for capsule_lce in result['results']]
 
         # Update capsule's download policy to on_demand
-        self.update_capsule_download_policy(module_capsule_configured, 'on_demand')
+        module_capsule_configured.update_download_policy('on_demand')
 
         # Create a content view with the repository
         cv = target_sat.api.ContentView(