Fix LRO poller to treat HTTP 404 and 406 as transient during transaction-status polling

Agent-Logs-Url: https://github.com/Azure/azure-sdk-for-python/sessions/6789a40b-e0da-4d5e-9cce-6cd42b9b4bd1

Co-authored-by: PallabPaul <20746357+PallabPaul@users.noreply.github.com>

Author: Copilot

sdk/confidentialledger/azure-confidentialledger/azure/confidentialledger/_operations/_patch.py

@@ -11,7 +11,7 @@
 import time
 from typing import Any, Callable, IO, List, Optional, Union, cast
 
-from azure.core.exceptions import ResourceNotFoundError
+from azure.core.exceptions import HttpResponseError, ResourceNotFoundError
 from azure.core.polling import PollingMethod, LROPoller, NoPolling
 
 from azure.confidentialledger._operations._operations import (
@@ -55,7 +55,6 @@ def __init__(
         self._latest_response: JSON = {}
 
         self._retry_not_found = retry_not_found
-        self._not_found_count = 0
 
     def initialize(self, client, initial_response, deserialization_callback):  # pylint: disable=unused-argument
         self._evaluate_response(initial_response)
@@ -102,15 +101,18 @@ def run(self) -> None:
                 try:
                     response = self._operation()
                     self._evaluate_response(response)
-                except ResourceNotFoundError as not_found_exception:
-                    # We'll allow some instances of resource not found to account for replication
-                    # delay if session stickiness is lost.
-
-                    self._not_found_count += 1
-
-                    not_retryable = not self._retry_not_found or self._give_up_not_found_error(not_found_exception)
-
-                    if not_retryable or self._not_found_count >= 3:
+                except HttpResponseError as http_exception:
+                    # 404: node doesn't know about the transaction yet (replication lag)
+                    # 406: node knows the transaction but it hasn't committed yet
+                    # Both are transient during transaction-status polling — treat as
+                    # pending so the poller retries.
+                    if http_exception.status_code in (404, 406) and self._retry_not_found:
+                        if (
+                            isinstance(http_exception, ResourceNotFoundError)
+                            and self._give_up_not_found_error(http_exception)
+                        ):
+                            raise
+                    else:
                         raise
                 if not self.finished():
                     time.sleep(self._polling_interval_s)

sdk/confidentialledger/azure-confidentialledger/azure/confidentialledger/aio/_operations/_patch.py

@@ -12,7 +12,7 @@
 import asyncio  # pylint: disable=do-not-import-asyncio
 from typing import Any, Callable, IO, Coroutine, List, Optional, Union, cast
 
-from azure.core.exceptions import ResourceNotFoundError
+from azure.core.exceptions import HttpResponseError, ResourceNotFoundError
 from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
 
 from azure.confidentialledger.aio._operations._operations import (
@@ -56,15 +56,18 @@ async def run(self) -> None:
                 try:
                     response = await self._operation()
                     self._evaluate_response(response)
-                except ResourceNotFoundError as not_found_exception:
-                    # We'll allow some instances of resource not found to account for replication
-                    # delay if session stickiness is lost.
-
-                    self._not_found_count += 1
-
-                    not_retryable = not self._retry_not_found or self._give_up_not_found_error(not_found_exception)
-
-                    if not_retryable or self._not_found_count >= 3:
+                except HttpResponseError as http_exception:
+                    # 404: node doesn't know about the transaction yet (replication lag)
+                    # 406: node knows the transaction but it hasn't committed yet
+                    # Both are transient during transaction-status polling — treat as
+                    # pending so the poller retries.
+                    if http_exception.status_code in (404, 406) and self._retry_not_found:
+                        if (
+                            isinstance(http_exception, ResourceNotFoundError)
+                            and self._give_up_not_found_error(http_exception)
+                        ):
+                            raise
+                    else:
                         raise
                 if not self.finished():
                     await asyncio.sleep(self._polling_interval_s)