Bazel rules for Python grpcio_reflection

5 files changed, 104 insertions, 0 deletions

diff --git a/src/proto/grpc/reflection/v1alpha/BUILD b/src/proto/grpc/reflection/v1alpha/BUILD
index 4605418447..4d919d029e 100644
--- a/src/proto/grpc/reflection/v1alpha/BUILD
+++ b/src/proto/grpc/reflection/v1alpha/BUILD
@@ -22,3 +22,11 @@ grpc_proto_library(
     name = "reflection_proto",
     srcs = ["reflection.proto"],
 )
+
+filegroup(
+    name = "reflection_proto_file",
+    srcs = [
+        "reflection.proto",
+    ],
+)
+
diff --git a/src/proto/grpc/testing/BUILD b/src/proto/grpc/testing/BUILD
index 16721ff2ed..7d68f87d3b 100644
--- a/src/proto/grpc/testing/BUILD
+++ b/src/proto/grpc/testing/BUILD
@@ -15,6 +15,8 @@
 licenses(["notice"])  # Apache v2
 
 load("//bazel:grpc_build_system.bzl", "grpc_proto_library", "grpc_package")
+load("@grpc_python_dependencies//:requirements.bzl", "requirement")
+load("@org_pubref_rules_protobuf//python:rules.bzl", "py_proto_library")
 
 grpc_package(name = "testing", visibility = "public")
 
@@ -58,6 +60,15 @@ grpc_proto_library(
     has_services = False,
 )
 
+py_proto_library(
+    name = "py_empty_proto",
+    protos = ["empty.proto",],
+    with_grpc = True,
+    deps = [
+        requirement('protobuf'),
+    ],
+)
+
 grpc_proto_library(
     name = "messages_proto",
     srcs = ["messages.proto"],
diff --git a/src/proto/grpc/testing/proto2/BUILD.bazel b/src/proto/grpc/testing/proto2/BUILD.bazel
new file mode 100644
index 0000000000..c4c4f004ef
--- /dev/null
+++ b/src/proto/grpc/testing/proto2/BUILD.bazel
@@ -0,0 +1,30 @@
+load("@grpc_python_dependencies//:requirements.bzl", "requirement")
+load("@org_pubref_rules_protobuf//python:rules.bzl", "py_proto_library")
+
+package(default_visibility = ["//visibility:public"])
+
+py_proto_library(
+    name = "empty2_proto",
+    protos = [
+        "empty2.proto",
+    ],
+    with_grpc = True,
+    deps = [
+        requirement('protobuf'),
+    ],
+)
+
+py_proto_library(
+    name = "empty2_extensions_proto",
+    protos = [
+        "empty2_extensions.proto",
+    ],
+    proto_deps = [
+        ":empty2_proto",
+    ],
+    with_grpc = True,
+    deps = [
+        requirement('protobuf'),
+    ],
+)
+
diff --git a/src/python/grpcio_reflection/grpc_reflection/v1alpha/BUILD.bazel b/src/python/grpcio_reflection/grpc_reflection/v1alpha/BUILD.bazel
new file mode 100644
index 0000000000..3a2ba26371
--- /dev/null
+++ b/src/python/grpcio_reflection/grpc_reflection/v1alpha/BUILD.bazel
@@ -0,0 +1,34 @@
+load("@grpc_python_dependencies//:requirements.bzl", "requirement")
+load("@org_pubref_rules_protobuf//python:rules.bzl", "py_proto_library")
+
+package(default_visibility = ["//visibility:public"])
+
+genrule(
+    name = "mv_reflection_proto",
+    srcs = [
+        "//src/proto/grpc/reflection/v1alpha:reflection_proto_file",
+    ],
+    outs = ["reflection.proto",],
+    cmd = "cp $< $@",
+)
+
+py_proto_library(
+    name = "py_reflection_proto",
+    protos = [":mv_reflection_proto",],
+    with_grpc = True,
+    deps = [
+        requirement('protobuf'),
+    ],
+)
+
+py_library(
+    name = "grpc_reflection",
+    srcs = ["reflection.py",],
+    deps = [
+        ":py_reflection_proto",
+        "//src/python/grpcio/grpc:grpcio",
+        requirement('protobuf'),
+    ],
+    imports=["../../",],
+)
+
diff --git a/src/python/grpcio_tests/tests/reflection/BUILD.bazel b/src/python/grpcio_tests/tests/reflection/BUILD.bazel
new file mode 100644
index 0000000000..c0efb0b7ce
--- /dev/null
+++ b/src/python/grpcio_tests/tests/reflection/BUILD.bazel
@@ -0,0 +1,21 @@
+load("@grpc_python_dependencies//:requirements.bzl", "requirement")
+
+package(default_visibility = ["//visibility:public"])
+
+py_test(
+    name="_reflection_servicer_test",
+    size="small",
+    timeout="moderate",
+    srcs=["_reflection_servicer_test.py",],
+    main="_reflection_servicer_test.py",
+    deps=[
+        "//src/python/grpcio/grpc:grpcio",
+        "//src/python/grpcio_reflection/grpc_reflection/v1alpha:grpc_reflection",
+        "//src/python/grpcio_tests/tests/unit:test_common",
+        "//src/proto/grpc/testing:py_empty_proto",
+        "//src/proto/grpc/testing/proto2:empty2_extensions_proto",
+        requirement('protobuf'),
+    ],
+    imports=["../../",],
+)
+