fix: pass updated keyword args to Publisher/Subscriber client in goog…

…le/pubsub
testcontainers · May 22, 2022 · 7717d76 · 7717d76
1 parent df0c28a
commit 7717d76
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/testcontainers/google/pubsub.py b/testcontainers/google/pubsub.py
@@ -55,10 +55,14 @@ def _get_channel(self, channel=None):
 
     def get_publisher_client(self, **kwargs):
         from google.cloud import pubsub
-        kwargs['channel'] = self._get_channel(kwargs.get('channel'))
+        from google.auth import credentials
+        kwargs['client_options'] = dict(api_endpoint=self.get_pubsub_emulator_host())
+        kwargs['credentials'] = credentials.AnonymousCredentials()
         return pubsub.PublisherClient(**kwargs)
 
     def get_subscriber_client(self, **kwargs):
         from google.cloud import pubsub
-        kwargs['channel'] = self._get_channel(kwargs.get('channel'))
+        from google.auth import credentials
+        kwargs['client_options'] = dict(api_endpoint=self.get_pubsub_emulator_host())
+        kwargs['credentials'] = credentials.AnonymousCredentials()
         return pubsub.SubscriberClient(**kwargs)