fix: updated artifact fetch for taskgraph 19 compatibility

Author: abhishekmadan30

test/actions/test_release_promotion.py

@@ -49,25 +49,22 @@ def inner(previous_graphs=None):
             )
 
         # Only the parameters from the first previous graph is downloaded.
-        responses.add(
-            method="GET",
-            url=f"{tc_url}/api/queue/v1/task/{list(previous_graphs.keys())[0]}/artifacts/public%2Fparameters.yml",
-            json=parameters,
-        )
+        # get_artifact does a two-step fetch: getLatestArtifact returns
+        # {"url": url}, then the content is fetched from that url.
+        url = f"{tc_url}/api/queue/v1/task/{list(previous_graphs.keys())[0]}/artifacts/public%2Fparameters.yml"
+        responses.add(method="GET", url=url, json={"url": url})
+        responses.add(method="GET", url=url, json=parameters)
 
         tid = count(0)
         for decision_id, full_task_graph in previous_graphs.items():
-            responses.add(
-                method="GET",
-                url=f"{tc_url}/api/queue/v1/task/{decision_id}/artifacts/public%2Ffull-task-graph.json",
-                json=full_task_graph.to_json(),
-            )
+            url = f"{tc_url}/api/queue/v1/task/{decision_id}/artifacts/public%2Ffull-task-graph.json"
+            responses.add(method="GET", url=url, json={"url": url})
+            responses.add(method="GET", url=url, json=full_task_graph.to_json())
+
             label_to_taskid = {label: int(next(tid)) for label in full_task_graph.tasks}
-            responses.add(
-                method="GET",
-                url=f"{tc_url}/api/queue/v1/task/{decision_id}/artifacts/public%2Flabel-to-taskid.json",
-                json=label_to_taskid,
-            )
+            url = f"{tc_url}/api/queue/v1/task/{decision_id}/artifacts/public%2Flabel-to-taskid.json"
+            responses.add(method="GET", url=url, json={"url": url})
+            responses.add(method="GET", url=url, json=label_to_taskid)
 
     return inner
 

test/transforms/test_replicate.py

@@ -214,10 +214,11 @@ def test_decision_task(responses, run_replicate):
     expected = get_expected(prefix, *task_defs)
 
     counter = count()
-    responses.get(
-        f"{TC_ROOT_URL}/api/queue/v1/task/{task_id}/artifacts/public%2Ftask-graph.json",
-        json={next(counter): task_def for task_def in task_defs},
+    url = (
+        f"{TC_ROOT_URL}/api/queue/v1/task/{task_id}/artifacts/public%2Ftask-graph.json"
     )
+    responses.get(url, json={"url": url})
+    responses.get(url, json={next(counter): task_def for task_def in task_defs})
     result = run_replicate(task)
     assert result == expected
 
@@ -269,9 +270,10 @@ def test_filtered_out(responses, run_replicate, target_def):
     task_defs = get_target_defs(target_def)
 
     counter = count()
-    responses.get(
-        f"{TC_ROOT_URL}/api/queue/v1/task/{task_id}/artifacts/public%2Ftask-graph.json",
-        json={next(counter): task_def for task_def in task_defs},
+    url = (
+        f"{TC_ROOT_URL}/api/queue/v1/task/{task_id}/artifacts/public%2Ftask-graph.json"
     )
+    responses.get(url, json={"url": url})
+    responses.get(url, json={next(counter): task_def for task_def in task_defs})
     result = run_replicate(task)
     assert len(result) == 0