Added method to poll tasks if needed

Signed-off-by: Mike Raineri <[email protected]>

Author: mraineri

redfish_protocol_validator/service_details.py

@@ -217,6 +217,7 @@ def test_ssdp_can_be_disabled(sut: SystemUnderTest):
         headers = {'If-Match': etag} if etag else {}
         r = sut.session.patch(sut.rhost + sut.mgr_net_proto_uri,
                               json=payload, headers=headers)
+        r = poll_task(sut, r)
         if r.ok:
             services = utils.discover_ssdp(search_target=SSDP_REDFISH)
             uuids = services.keys()

redfish_protocol_validator/utils.py

@@ -103,6 +103,24 @@ def is_text_in_extended_error(text: str, body: dict):
     return False
 
 
+def poll_task(sut, response):
+    # If the response doesn't show 202 Accepted, there's nothing to poll
+    if response.status_code != requests.codes.ACCEPTED:
+        return response
+
+    # Get the task monitor URI and poll it
+    task_monitor = response.headers.get('Location')
+    if task_monitor:
+        # Try for up to 1 minute at 5 second intervals
+        for _ in range(12):
+            time.sleep(5)
+            response = sut.session.get(sut.rhost + task_monitor)
+            # Once the task is done, break out
+            if response.status_code != requests.codes.ACCEPTED:
+                break
+    return response
+
+
 def get_sse_stream(sut):
     response = None
     event_dest_uri = None