fix: Add emulator support to schema service

google/pubsub_v1/services/schema_service/client.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 #
 from collections import OrderedDict
+import functools
 import os
 import re
 from typing import Dict, Mapping, Optional, Sequence, Tuple, Type, Union
@@ -39,6 +40,8 @@
 from google.pubsub_v1.services.schema_service import pagers
 from google.pubsub_v1.types import schema
 from google.pubsub_v1.types import schema as gp_schema
+
+import grpc
 from .transports.base import SchemaServiceTransport, DEFAULT_CLIENT_INFO
 from .transports.grpc import SchemaServiceGrpcTransport
 from .transports.grpc_asyncio import SchemaServiceGrpcAsyncIOTransport
@@ -411,6 +414,15 @@ def __init__(
  )
 
  Transport = type(self).get_transport_class(transport)
+
+ emulator_host = os.environ.get("PUBSUB_EMULATOR_HOST")
+ if emulator_host:
+ if issubclass(Transport, type(self)._transport_registry["grpc"]):
+ channel = grpc.insecure_channel(target=emulator_host)
+ else:
+ channel = grpc.aio.insecure_channel(target=emulator_host)
+ Transport = functools.partial(Transport, channel=channel)
+
  self._transport = Transport(
  credentials=credentials,
  credentials_file=client_options.credentials_file,

owlbot.py

@@ -85,6 +85,7 @@
  clients_to_patch = [
  library / f"google/pubsub_{library.name}/services/publisher/client.py",
  library / f"google/pubsub_{library.name}/services/subscriber/client.py",
+ library / f"google/pubsub_{library.name}/services/schema_service/client.py",
  ]
  err_msg = (
  "Expected replacements for gRPC channel to use with the emulator not made."
@@ -97,8 +98,8 @@
 
  count = s.replace(
  clients_to_patch,
- f"from google\.pubsub_{library.name}\.types import pubsub",
- "\g<0>\n\nimport grpc",
+ f"from \.transports\.base",
+ "\nimport grpc\n\g<0>",
  )
 
  if count < len(clients_to_patch):