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):