Set timeout to infinite for publishing with ordering keys enabled

google/cloud/pubsub_v1/publisher/client.py

@@ -399,8 +399,10 @@ def on_publish_done(future):
                     transport = self._transport
                     base_retry = transport._wrapped_methods[transport.publish]._retry
                     retry = base_retry.with_deadline(2.0**32)
+                    timeout = 2.0**32
                 elif retry is not None:
                     retry = retry.with_deadline(2.0**32)
+                    timeout = 2.0**32
 
             # Delegate the publishing to the sequencer.
             sequencer = self._get_or_create_sequencer(topic, ordering_key)

tests/unit/pubsub_v1/publisher/test_publisher_client.py

@@ -28,12 +28,14 @@
 
 import pytest
 import time
+from typing import cast
 
 from google.api_core import gapic_v1
 from google.api_core import retry as retries
 from google.api_core.gapic_v1.client_info import METRICS_METADATA_KEY
 from google.cloud.pubsub_v1 import publisher
 from google.cloud.pubsub_v1 import types
+from google.cloud.pubsub_v1 import PublisherTransport
 
 from google.cloud.pubsub_v1.publisher import exceptions
 from google.cloud.pubsub_v1.publisher._sequencer import ordered_sequencer
@@ -313,7 +315,7 @@ def test_publish_with_ordering_key_uses_extended_retry_deadline(creds):
     _, kwargs = batch_class.call_args
 
     batch_commit_retry = kwargs["commit_retry"]
-    expected_retry = custom_retry.with_deadline(2.0**32)
+    expected_retry = custom_retry.with_deadline(2.0**32).with_timeout(2.0**32)
     _assert_retries_equal(batch_commit_retry, expected_retry)
 
 
@@ -342,7 +344,6 @@ def test_publish_with_ordering_key_with_no_retry(creds):
     # Check the retry settings used for the batch.
     batch_class.assert_called_once()
 
-
 def test_publish_attrs_bytestring(creds):
     client = publisher.Client(credentials=creds)