From 00e4e79addb71d15539850670f21683412d4bb90 Mon Sep 17 00:00:00 2001
From: Peter Ondrejka <pondrejk@redhat.com>
Date: Thu, 23 May 2024 11:11:00 +0200
Subject: [PATCH] fix in pull mode longrun test

---
 tests/foreman/api/test_remoteexecution.py | 34 ++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/tests/foreman/api/test_remoteexecution.py b/tests/foreman/api/test_remoteexecution.py
index bdaa303233f..5ac23cd0917 100644
--- a/tests/foreman/api/test_remoteexecution.py
+++ b/tests/foreman/api/test_remoteexecution.py
@@ -206,6 +206,12 @@ def test_negative_time_to_pickup(
 @pytest.mark.tier3
 @pytest.mark.no_containers
 @pytest.mark.rhel_ver_list('8')
+@pytest.mark.parametrize(
+    'setting_update',
+    ['remote_execution_global_proxy=False'],
+    ids=["no_global_proxy"],
+    indirect=True,
+)
 def test_positive_check_longrunning_job(
     module_org,
     module_target_sat,
@@ -213,6 +219,7 @@ def test_positive_check_longrunning_job(
     module_ak_with_cv,
     module_capsule_configured_mqtt,
     rhel_contenthost,
+    setting_update,
 ):
     """Time to pickup setting doesn't disrupt longrunning jobs
 
@@ -223,10 +230,35 @@ def test_positive_check_longrunning_job(
 
     :CaseImportance: Medium
 
-    :bz: 2118651
+    :bz: 2118651, 2158738
 
     :parametrized: yes
     """
+
+    client_repo = ohsnap.dogfood_repository(
+        settings.ohsnap,
+        product='client',
+        repo='client',
+        release='client',
+        os_release=rhel_contenthost.os_version.major,
+    )
+    # Update module_capsule_configured_mqtt to include module_org/smart_proxy_location
+    module_target_sat.cli.Capsule.update(
+        {
+            'name': module_capsule_configured_mqtt.hostname,
+            'organization-ids': module_org.id,
+            'location-ids': smart_proxy_location.id,
+        }
+    )
+    # register host with pull provider rex
+    result = rhel_contenthost.register(
+        module_org,
+        smart_proxy_location,
+        module_ak_with_cv.name,
+        module_capsule_configured_mqtt,
+        setup_remote_execution_pull=True,
+        repo=client_repo.baseurl,
+    )
     template_id = (
         module_target_sat.api.JobTemplate()
         .search(query={'search': 'name="Run Command - Script Default"'})[0]