update polling interval (Azure#19119)

AzureSDKAutomation · Jun 4, 2021 · 0331778 · 0331778
1 parent d1337f2
commit 0331778
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 6 deletions.
diff --git a/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_client.py b/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_client.py
@@ -16,7 +16,13 @@
 )
 from ._user_agent import USER_AGENT
 from ._polling import TranslationPolling, DocumentTranslationLROPollingMethod
-from ._helpers import get_http_logging_policy, convert_datetime, get_authentication_policy, get_translation_input
+from ._helpers import (
+    get_http_logging_policy,
+    convert_datetime,
+    get_authentication_policy,
+    get_translation_input,
+    POLLING_INTERVAL
+)
 if TYPE_CHECKING:
     from azure.core.paging import ItemPaged
     from azure.core.credentials import TokenCredential, AzureKeyCredential
@@ -64,14 +70,15 @@ def __init__(self, endpoint, credential, **kwargs):
         self._api_version = kwargs.pop('api_version', None)
 
         authentication_policy = get_authentication_policy(credential)
-
+        polling_interval = kwargs.pop("polling_interval", POLLING_INTERVAL)
         self._client = _BatchDocumentTranslationClient(
             endpoint=endpoint,
             credential=credential,  # type: ignore
             api_version=self._api_version,
             sdk_moniker=USER_AGENT,
             authentication_policy=authentication_policy,
             http_logging_policy=get_http_logging_policy(),
+            polling_interval=polling_interval,
             **kwargs
         )
 

diff --git a/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_helpers.py b/sdk/translation/azure-ai-translation-document/azure/ai/translation/document/_helpers.py
@@ -17,6 +17,7 @@
 )
 from ._models import DocumentTranslationInput
 COGNITIVE_KEY_HEADER = "Ocp-Apim-Subscription-Key"
+POLLING_INTERVAL = 1
 
 
 def get_translation_input(args, kwargs, continuation_token):

diff --git a/...nslation/azure-ai-translation-document/azure/ai/translation/document/aio/_client_async.py b/...nslation/azure-ai-translation-document/azure/ai/translation/document/aio/_client_async.py
@@ -17,7 +17,13 @@
     FileFormat,
     DocumentStatusResult
 )
-from .._helpers import get_http_logging_policy, convert_datetime, get_authentication_policy, get_translation_input
+from .._helpers import (
+    get_http_logging_policy,
+    convert_datetime,
+    get_authentication_policy,
+    get_translation_input,
+    POLLING_INTERVAL
+)
 from ._async_polling import AsyncDocumentTranslationLROPollingMethod, AsyncDocumentTranslationLROPoller
 from .._polling import TranslationPolling
 if TYPE_CHECKING:
@@ -67,13 +73,15 @@ def __init__(
         self._api_version = kwargs.pop('api_version', None)
 
         authentication_policy = get_authentication_policy(credential)
+        polling_interval = kwargs.pop("polling_interval", POLLING_INTERVAL)
         self._client = _BatchDocumentTranslationClient(
             endpoint=endpoint,
             credential=credential,  # type: ignore
             api_version=self._api_version,
             sdk_moniker=USER_AGENT,
             authentication_policy=authentication_policy,
             http_logging_policy=get_http_logging_policy(),
+            polling_interval=polling_interval,
             **kwargs
         )
 

diff --git a/sdk/translation/azure-ai-translation-document/tests/preparer.py b/sdk/translation/azure-ai-translation-document/tests/preparer.py
@@ -39,9 +39,6 @@ def create_resource(self, name, **kwargs):
         # set polling interval to 0 for recorded tests
         if not self.is_live:
             self.client_kwargs["polling_interval"] = 0
-        else:
-            # default is 30s, but our tests translate very small docs so this helps speed up live testing
-            self.client_kwargs["polling_interval"] = 5
 
         client = self.client_cls(
             doctranslation_test_endpoint,