RSDK-11648- add version param for get_fragment (#972)

Author: stuqdog

src/viam/app/app_client.py

@@ -1766,7 +1766,7 @@ async def list_fragments(
         response: ListFragmentsResponse = await self._app_client.ListFragments(request, metadata=self._metadata)
         return [Fragment.from_proto(fragment=fragment) for fragment in response.fragments]
 
-    async def get_fragment(self, fragment_id: str) -> Fragment:
+    async def get_fragment(self, fragment_id: str, version: Optional[str] = None) -> Fragment:
         """Get a fragment.
 
         ::
@@ -1778,6 +1778,7 @@ async def get_fragment(self, fragment_id: str) -> Fragment:
 
         Args:
             fragment_id (str): ID of the fragment to get.
+            version (str): Optional specification of the fragment version to get (revision or tag).
 
         Raises:
             GRPCError: If an invalid fragment ID is passed.
@@ -1787,7 +1788,7 @@ async def get_fragment(self, fragment_id: str) -> Fragment:
 
         For more information, see `Fleet Management API <https://docs.viam.com/dev/reference/apis/fleet/#getfragment>`_.
         """
-        request = GetFragmentRequest(id=fragment_id)
+        request = GetFragmentRequest(id=fragment_id, version=version)
         response: GetFragmentResponse = await self._app_client.GetFragment(request, metadata=self._metadata)
         return Fragment.from_proto(fragment=response.fragment)
 

tests/mocks/services.py

@@ -1627,6 +1627,7 @@ async def GetFragment(self, stream: Stream[GetFragmentRequest, GetFragmentRespon
         request = await stream.recv_message()
         assert request is not None
         self.fragment_id = request.id
+        self.version = request.version
         await stream.send_message(GetFragmentResponse(fragment=self.fragment))
 
     async def GetFragmentHistory(self, stream: Stream[GetFragmentHistoryRequest, GetFragmentHistoryResponse]) -> None:

tests/test_app_client.py

@@ -574,8 +574,9 @@ async def test_list_fragments(self, service: MockApp):
     async def test_get_fragment(self, service: MockApp):
         async with ChannelFor([service]) as channel:
             client = AppClient(channel, METADATA, ID)
-            fragment = await client.get_fragment(fragment_id=ID)
+            fragment = await client.get_fragment(fragment_id=ID, version=VERSION)
             assert service.fragment_id == ID
+            assert service.version == VERSION
             assert fragment.proto == FRAGMENT
 
     async def test_create_fragment(self, service: MockApp):