python: apply bazel_namespace_package_hack in _run_time_type_check_main.py and tests_aio

Author: AgraVator

bazel/_run_time_type_check_main.py

@@ -23,6 +23,13 @@
 
 from typeguard import install_import_hook
 
+try:
+    from tests import bazel_namespace_package_hack
+
+    bazel_namespace_package_hack.sys_path_to_site_dir_hack()
+except Exception:
+    pass
+
 
 # AIO
 install_import_hook('grpc.aio')

src/python/grpcio_tests/tests/BUILD.bazel

@@ -20,5 +20,6 @@ py_library(
         "//examples/python/errors:__subpackages__",
         "//src/python/grpcio_tests/tests/interop:__subpackages__",
         "//src/python/grpcio_tests/tests/status:__subpackages__",
+        "//src/python/grpcio_tests/tests_aio:__subpackages__",
     ],
 )

src/python/grpcio_tests/tests_aio/unit/BUILD.bazel

@@ -45,6 +45,7 @@ py_library(
         "//src/proto/grpc/testing:py_messages_proto",
         "//src/proto/grpc/testing:test_py_pb2_grpc",
         "//src/python/grpcio/grpc:grpcio",
+        "//src/python/grpcio_tests/tests:bazel_namespace_package_hack",
         "//src/python/grpcio_tests/tests/unit:resources",
     ],
 )

src/python/grpcio_tests/tests_aio/unit/_test_server.py

@@ -20,6 +20,10 @@
 import grpc
 from grpc.experimental import aio
 
+from tests import bazel_namespace_package_hack
+
+bazel_namespace_package_hack.sys_path_to_site_dir_hack()
+
 from src.proto.grpc.testing import empty_pb2
 from src.proto.grpc.testing import messages_pb2
 from src.proto.grpc.testing import test_pb2_grpc