From 62591c8ead85c33fa5a5c4cc7c2a26779cbd1075 Mon Sep 17 00:00:00 2001
From: "gcf-owl-bot[bot]" <78513119+gcf-owl-bot[bot]@users.noreply.github.com>
Date: Thu, 30 Sep 2021 19:08:13 +0000
Subject: [PATCH] feat: add RPC retry information for MacSign, MacVerify, and
 GenerateRandomBytes (#186)

- [ ] Regenerate this pull request now.

PiperOrigin-RevId: 399925901

Source-Link: https://github.com/googleapis/googleapis/commit/ae7e399135b2bc16e1f1f47edb270db2cdf98bd8

Source-Link: https://github.com/googleapis/googleapis-gen/commit/3a8128528ceddba0c1775401eef57ecb582db350
Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiM2E4MTI4NTI4Y2VkZGJhMGMxNzc1NDAxZWVmNTdlY2I1ODJkYjM1MCJ9
---
 .../key_management_service/async_client.py    | 36 +++++++++++++++--
 .../key_management_service/transports/base.py | 40 +++++++++++++++++--
 2 files changed, 70 insertions(+), 6 deletions(-)

diff --git a/google/cloud/kms_v1/services/key_management_service/async_client.py b/google/cloud/kms_v1/services/key_management_service/async_client.py
index d539c874..5d3c5557 100644
--- a/google/cloud/kms_v1/services/key_management_service/async_client.py
+++ b/google/cloud/kms_v1/services/key_management_service/async_client.py
@@ -2628,7 +2628,17 @@ async def mac_sign(
         # and friendly error handling.
         rpc = gapic_v1.method_async.wrap_method(
             self._client._transport.mac_sign,
-            default_timeout=None,
+            default_retry=retries.Retry(
+                initial=0.1,
+                maximum=60.0,
+                multiplier=1.3,
+                predicate=retries.if_exception_type(
+                    core_exceptions.DeadlineExceeded,
+                    core_exceptions.ServiceUnavailable,
+                ),
+                deadline=60.0,
+            ),
+            default_timeout=60.0,
             client_info=DEFAULT_CLIENT_INFO,
         )
 
@@ -2724,7 +2734,17 @@ async def mac_verify(
         # and friendly error handling.
         rpc = gapic_v1.method_async.wrap_method(
             self._client._transport.mac_verify,
-            default_timeout=None,
+            default_retry=retries.Retry(
+                initial=0.1,
+                maximum=60.0,
+                multiplier=1.3,
+                predicate=retries.if_exception_type(
+                    core_exceptions.DeadlineExceeded,
+                    core_exceptions.ServiceUnavailable,
+                ),
+                deadline=60.0,
+            ),
+            default_timeout=60.0,
             client_info=DEFAULT_CLIENT_INFO,
         )
 
@@ -2822,7 +2842,17 @@ async def generate_random_bytes(
         # and friendly error handling.
         rpc = gapic_v1.method_async.wrap_method(
             self._client._transport.generate_random_bytes,
-            default_timeout=None,
+            default_retry=retries.Retry(
+                initial=0.1,
+                maximum=60.0,
+                multiplier=1.3,
+                predicate=retries.if_exception_type(
+                    core_exceptions.DeadlineExceeded,
+                    core_exceptions.ServiceUnavailable,
+                ),
+                deadline=60.0,
+            ),
+            default_timeout=60.0,
             client_info=DEFAULT_CLIENT_INFO,
         )
 
diff --git a/google/cloud/kms_v1/services/key_management_service/transports/base.py b/google/cloud/kms_v1/services/key_management_service/transports/base.py
index 076d5495..24e04695 100644
--- a/google/cloud/kms_v1/services/key_management_service/transports/base.py
+++ b/google/cloud/kms_v1/services/key_management_service/transports/base.py
@@ -484,14 +484,48 @@ def _prep_wrapped_messages(self, client_info):
                 client_info=client_info,
             ),
             self.mac_sign: gapic_v1.method.wrap_method(
-                self.mac_sign, default_timeout=None, client_info=client_info,
+                self.mac_sign,
+                default_retry=retries.Retry(
+                    initial=0.1,
+                    maximum=60.0,
+                    multiplier=1.3,
+                    predicate=retries.if_exception_type(
+                        core_exceptions.DeadlineExceeded,
+                        core_exceptions.ServiceUnavailable,
+                    ),
+                    deadline=60.0,
+                ),
+                default_timeout=60.0,
+                client_info=client_info,
             ),
             self.mac_verify: gapic_v1.method.wrap_method(
-                self.mac_verify, default_timeout=None, client_info=client_info,
+                self.mac_verify,
+                default_retry=retries.Retry(
+                    initial=0.1,
+                    maximum=60.0,
+                    multiplier=1.3,
+                    predicate=retries.if_exception_type(
+                        core_exceptions.DeadlineExceeded,
+                        core_exceptions.ServiceUnavailable,
+                    ),
+                    deadline=60.0,
+                ),
+                default_timeout=60.0,
+                client_info=client_info,
             ),
             self.generate_random_bytes: gapic_v1.method.wrap_method(
                 self.generate_random_bytes,
-                default_timeout=None,
+                default_retry=retries.Retry(
+                    initial=0.1,
+                    maximum=60.0,
+                    multiplier=1.3,
+                    predicate=retries.if_exception_type(
+                        core_exceptions.DeadlineExceeded,
+                        core_exceptions.ServiceUnavailable,
+                    ),
+                    deadline=60.0,
+                ),
+                default_timeout=60.0,
                 client_info=client_info,
             ),
         }