GOV-667: Added back is retry method

Author: ankitpatnaik-atlan

pyatlan/client/transport.py

@@ -82,8 +82,11 @@ def handle_request(self, request: httpx.Request) -> httpx.Response:
         """
         logger.debug("handle_request started request=%s", request)
 
-        send_method = partial(self._transport.handle_request)
-        response = self._retry_operation(request, send_method)
+        if self.retry.is_retryable_method(request.method):
+            send_method = partial(self._transport.handle_request)
+            response = self._retry_operation(request, send_method)
+        else:
+            response = self._transport.handle_request(request)
 
         logger.debug(
             "handle_request finished request=%s response=%s", request, response
@@ -196,8 +199,11 @@ async def handle_async_request(self, request: httpx.Request) -> httpx.Response:
         """
         logger.debug("handle_async_request started request=%s", request)
 
-        send_method = partial(self._transport.handle_async_request)
-        response = await self._retry_operation_async(request, send_method)
+        if self.retry.is_retryable_method(request.method):
+            send_method = partial(self._transport.handle_async_request)
+            response = await self._retry_operation_async(request, send_method)
+        else:
+            response = await self._transport.handle_async_request(request)
 
         logger.debug(
             "handle_async_request finished request=%s response=%s", request, response

tests/integration/transport_test.py

@@ -79,7 +79,7 @@ def test_duplicate_prevention_on_timeout(client: AtlanClient):  # noqa: F811
     policy_name = f"{MODULE_NAME}_DupCheck"
 
     transport = PyatlanSyncTransport(
-        retry=Retry(total=3, backoff_factor=0),
+        retry=Retry(total=3, backoff_factor=0, allowed_methods=["POST"]),
         client=client,
         trust_env=True,
     )
@@ -151,7 +151,7 @@ def test_duplicate_prevention_short_circuits_when_policy_exists(
     }
 
     transport = PyatlanSyncTransport(
-        retry=Retry(total=3, backoff_factor=0),
+        retry=Retry(total=3, backoff_factor=0, allowed_methods=["POST"]),
         client=client,
         trust_env=True,
     )

tests/unit/test_transport.py

@@ -293,7 +293,7 @@ async def test_propagates_search_error(self):
 class TestPyatlanSyncTransportRetry:
     def _make_transport(self, client=None):
         transport = PyatlanSyncTransport(
-            retry=Retry(total=3, backoff_factor=0),
+            retry=Retry(total=3, backoff_factor=0, allowed_methods=["POST"]),
             client=client,
             trust_env=False,
         )
@@ -391,7 +391,7 @@ def test_no_duplicate_check_on_first_attempt(self):
 class TestPyatlanAsyncTransportRetry:
     def _make_transport(self, client=None):
         transport = PyatlanAsyncTransport(
-            retry=Retry(total=3, backoff_factor=0),
+            retry=Retry(total=3, backoff_factor=0, allowed_methods=["POST"]),
             client=client,
             trust_env=False,
         )