test: restore test coverage for find_task_id_batched and status_task_batched

Author: jcristau

test/test_util_taskcluster.py

@@ -190,6 +190,27 @@ def mock_client(service):
     mock_index.findTask.assert_called_with(index)
 
 
+def test_find_task_id_batched(monkeypatch, responses, root_url):
+    monkeypatch.setattr(os, "environ", {"TASKCLUSTER_ROOT_URL": root_url})
+    responses.post(
+        f"{root_url}/api/index/v1/tasks/indexes",
+        json={
+            "tasks": [{"taskId": "abc", "namespace": "index.abc"}],
+            "continuationToken": "abc",
+        },
+    )
+    responses.post(
+        f"{root_url}/api/index/v1/tasks/indexes",
+        json={
+            "tasks": [{"taskId": "def", "namespace": "index.def"}],
+            "continuationToken": None,
+        },
+    )
+
+    result = tc.find_task_id_batched(["index.abc", "index.def"])
+    assert result == {"index.abc": "abc", "index.def": "def"}
+
+
 def test_get_artifact_from_index(monkeypatch):
     index = "foo"
     path = "file.txt"
@@ -298,6 +319,40 @@ def mock_client(service):
     mock_queue.status.assert_called_with(tid)
 
 
+def test_status_task_batched(monkeypatch, responses, root_url):
+    monkeypatch.setattr(os, "environ", {"TASKCLUSTER_ROOT_URL": root_url})
+    responses.post(
+        f"{root_url}/api/queue/v1/tasks/status",
+        json={
+            "statuses": [
+                {
+                    "taskId": "abc",
+                    "status": {"taskId": "abc", "state": "completed", "runs": []},
+                }
+            ],
+            "continuationToken": "abc",
+        },
+    )
+    responses.post(
+        f"{root_url}/api/queue/v1/tasks/status",
+        json={
+            "statuses": [
+                {
+                    "taskId": "def",
+                    "status": {"taskId": "def", "state": "exception", "runs": []},
+                }
+            ],
+            "continuationToken": None,
+        },
+    )
+
+    result = tc.status_task_batched(["abc", "def"])
+    assert result == {
+        "abc": {"taskId": "abc", "state": "completed", "runs": []},
+        "def": {"taskId": "def", "state": "exception", "runs": []},
+    }
+
+
 def test_state_task(monkeypatch):
     tid = "123"