updates pagination

src/orca/services/nextflowtower/client.py

@@ -113,7 +113,9 @@ def request_paged(self, method: str, path: str, **kwargs) -> dict[str, Any]:
         while num_items < total_size:
             kwargs["params"]["offset"] = num_items
             json = self.request_json(method, path, **kwargs)
-            total_size = json.pop("totalSize")
+            total_size = (
+                json.pop("totalSize") if json.get("totalSize") else json.pop("total")
+            )
             key_name, items = json.popitem()
             num_items += len(items)
             all_items.extend(items)
@@ -134,7 +136,7 @@ def get(self, path: str, **kwargs) -> dict[str, Any]:
             A dictionary from deserializing the JSON response.
         """
         json = self.request_json("GET", path, **kwargs)
-        if "totalSize" in json:
+        if "totalSize" in json or "total" in json:
             json = self.request_paged("GET", path, **kwargs)
         return json
 

tests/services/nextflowtower/test_client.py

@@ -155,7 +155,6 @@ def test_that_get_workflow_tasks_works(client, mocker, get_response):
     mock.return_value = get_response("get_workflow_tasks")
     result = client.get_workflow_tasks(workspace_id=98765, workflow_id="123456789")
     mock.assert_called()
-    print(result)
     assert len(result) == 4