feat: linked task get_by_id

Author: jorwoods

tableauserverclient/server/endpoint/linked_tasks_endpoint.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Tuple
+from typing import List, Optional, Tuple, Union
 
 from tableauserverclient.helpers.logging import logger
 from tableauserverclient.models.linked_tasks_item import LinkedTaskItem
@@ -24,3 +24,12 @@ def get(self, req_options: Optional["RequestOptions"] = None) -> Tuple[List[Link
         pagination_item = PaginationItem.from_response(server_response.content, self.parent_srv.namespace)
         all_group_items = LinkedTaskItem.from_response(server_response.content, self.parent_srv.namespace)
         return all_group_items, pagination_item
+
+    @api(version="3.15")
+    def get_by_id(self, linked_task: Union[LinkedTaskItem, str]) -> LinkedTaskItem:
+        task_id = getattr(linked_task, "id", linked_task)
+        logger.info("Querying all linked tasks on site")
+        url = f"{self.baseurl}/{task_id}"
+        server_response = self.get_request(url)
+        all_group_items = LinkedTaskItem.from_response(server_response.content, self.parent_srv.namespace)
+        return all_group_items[0]

test/test_linked_tasks.py

@@ -73,3 +73,29 @@ def test_get_linked_tasks(self):
         assert task.num_steps == 1
         assert task.schedule is not None
         assert task.schedule.id == "be077332-d01d-481b-b2f3-917e463d4dca"
+
+    def test_get_by_id_str_linked_task(self):
+        id_ = "1b8211dc-51a8-45ce-a831-b5921708e03e"
+
+        with requests_mock.mock() as m:
+            m.get(f"{self.baseurl}/{id_}", text=GET_LINKED_TASKS.read_text())
+            task = self.server.linked_tasks.get_by_id(id_)
+
+        assert task.id == "1b8211dc-51a8-45ce-a831-b5921708e03e"
+        assert task.num_steps == 1
+        assert task.schedule is not None
+        assert task.schedule.id == "be077332-d01d-481b-b2f3-917e463d4dca"
+
+    def test_get_by_id_obj_linked_task(self):
+        id_ = "1b8211dc-51a8-45ce-a831-b5921708e03e"
+        in_task = LinkedTaskItem()
+        in_task.id = id_
+
+        with requests_mock.mock() as m:
+            m.get(f"{self.baseurl}/{id_}", text=GET_LINKED_TASKS.read_text())
+            task = self.server.linked_tasks.get_by_id(in_task)
+
+        assert task.id == "1b8211dc-51a8-45ce-a831-b5921708e03e"
+        assert task.num_steps == 1
+        assert task.schedule is not None
+        assert task.schedule.id == "be077332-d01d-481b-b2f3-917e463d4dca"