DATA-4449: Update CLI/SDK for RenameDataPipeline (#964)

Author: Emerald-Z

src/viam/app/data_client.py

@@ -72,6 +72,7 @@
     ListDataPipelineRunsResponse,
     ListDataPipelinesRequest,
     ListDataPipelinesResponse,
+    RenameDataPipelineRequest,
 )
 from viam.proto.app.datapipelines import (
     DataPipeline as ProtoDataPipeline,
@@ -1969,6 +1970,21 @@ async def create_data_pipeline(
         response: CreateDataPipelineResponse = await self._data_pipelines_client.CreateDataPipeline(request, metadata=self._metadata)
         return response.id
 
+    async def rename_data_pipeline(self, id: str, name: str) -> None:
+        """Rename a data pipeline by its ID.
+        ::
+
+            await data_client.rename_data_pipeline(id="<YOUR-DATA-PIPELINE-ID>", name="<YOUR-NEW-NAME>")
+
+        Args:
+            id (str): The ID of the data pipeline to rename.
+            name (str): The new name of the data pipeline.
+        """
+        if not id or not name:
+            raise ValueError("id and name are required")
+        request = RenameDataPipelineRequest(id=id, name=name)
+        await self._data_pipelines_client.RenameDataPipeline(request, metadata=self._metadata)
+
     async def delete_data_pipeline(self, id: str) -> None:
         """Delete a data pipeline by its ID.
 

tests/mocks/services.py

@@ -259,6 +259,7 @@
     ListDataPipelineRunsResponse,
     ListDataPipelinesRequest,
     ListDataPipelinesResponse,
+    RenameDataPipelineResponse,
     UnimplementedDataPipelinesServiceBase,
 )
 from viam.proto.app.dataset import (
@@ -1188,6 +1189,13 @@ async def ListDataPipelineRuns(self, stream: Stream[ListDataPipelineRunsRequest,
         self.id = request.id
         await stream.send_message(ListDataPipelineRunsResponse(pipeline_id=self.id, runs=self.runs_response))
 
+    async def RenameDataPipeline(self, stream) -> None:
+        request = await stream.recv_message()
+        assert request is not None
+        self.id = request.id
+        self.name = request.name
+        await stream.send_message(RenameDataPipelineResponse())
+
 
 class MockMLTraining(UnimplementedMLTrainingServiceBase):
     def __init__(self, job_id: str, training_metadata: TrainingJobMetadata):

tests/test_data_pipelines.py

@@ -116,3 +116,9 @@ async def test_list_data_pipeline_runs(self, service: MockDataPipelines):
             runs = await client.list_data_pipeline_runs(ID)
             runs._client = None
             assert runs == DATA_PIPELINE_RUNS_PAGE
+
+    async def test_rename_data_pipeline(self, service: MockDataPipelines):
+        async with ChannelFor([service]) as channel:
+            client = DataClient(channel, DATA_SERVICE_METADATA)
+            await client.rename_data_pipeline(ID, NAME)
+            assert service.name == NAME