refactor(tracing) Rename grpc's test proto content

Author: hossein-raeisi

tests/integrations/grpc/grpc_test_service.proto

@@ -0,0 +1,11 @@
+syntax = "proto3";
+
+package grpc_test_server;
+
+service gRPCTestService{
+  rpc TestServe(gRPCTestMessage) returns (gRPCTestMessage);
+}
+
+message gRPCTestMessage {
+  string text = 1;
+}

tests/integrations/grpc/grpc_test_service_pb2.py

@@ -15,7 +15,7 @@ else:
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
 @typing_extensions.final
-class TestMessage(google.protobuf.message.Message):
+class gRPCTestMessage(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     TEXT_FIELD_NUMBER: builtins.int
@@ -25,8 +25,6 @@ class TestMessage(google.protobuf.message.Message):
         *,
         text: builtins.str = ...,
     ) -> None: ...
-    def ClearField(
-        self, field_name: typing_extensions.Literal["text", b"text"]
-    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["text", b"text"]) -> None: ...
 
-global___TestMessage = TestMessage
+global___gRPCTestMessage = gRPCTestMessage

tests/integrations/grpc/test_service_pb2.pyi renamed to tests/integrations/grpc/grpc_test_service_pb2.pyi

@@ -0,0 +1,66 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+import tests.integrations.grpc.grpc_test_service_pb2 as grpc__test__service__pb2
+
+
+class gRPCTestServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.TestServe = channel.unary_unary(
+                '/grpc_test_server.gRPCTestService/TestServe',
+                request_serializer=grpc__test__service__pb2.gRPCTestMessage.SerializeToString,
+                response_deserializer=grpc__test__service__pb2.gRPCTestMessage.FromString,
+                )
+
+
+class gRPCTestServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def TestServe(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_gRPCTestServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'TestServe': grpc.unary_unary_rpc_method_handler(
+                    servicer.TestServe,
+                    request_deserializer=grpc__test__service__pb2.gRPCTestMessage.FromString,
+                    response_serializer=grpc__test__service__pb2.gRPCTestMessage.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'grpc_test_server.gRPCTestService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class gRPCTestService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def TestServe(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/grpc_test_server.gRPCTestService/TestServe',
+            grpc__test__service__pb2.gRPCTestMessage.SerializeToString,
+            grpc__test__service__pb2.gRPCTestMessage.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

tests/integrations/grpc/grpc_test_service_pb2_grpc.py

@@ -11,11 +11,11 @@
 from sentry_sdk.consts import OP
 from sentry_sdk.integrations.grpc.client import ClientInterceptor
 from sentry_sdk.integrations.grpc.server import ServerInterceptor
-from tests.integrations.grpc.test_service_pb2 import TestMessage
-from tests.integrations.grpc.test_service_pb2_grpc import (
-    TestServiceServicer,
-    add_TestServiceServicer_to_server,
-    TestServiceStub,
+from tests.integrations.grpc.grpc_test_service_pb2 import gRPCTestMessage
+from tests.integrations.grpc.grpc_test_service_pb2_grpc import (
+    gRPCTestServiceServicer,
+    add_gRPCTestServiceServicer_to_server,
+    gRPCTestServiceStub,
 )
 
 PORT = 50051
@@ -30,8 +30,8 @@ def test_grpc_server_starts_transaction(sentry_init, capture_events_forksafe):
     server = _set_up()
 
     with grpc.insecure_channel(f"localhost:{PORT}") as channel:
-        stub = TestServiceStub(channel)
-        stub.TestServe(TestMessage(text="test"))
+        stub = gRPCTestServiceStub(channel)
+        stub.TestServe(gRPCTestMessage(text="test"))
 
     _tear_down(server=server)
 
@@ -55,7 +55,7 @@ def test_grpc_server_continues_transaction(sentry_init, capture_events_forksafe)
     server = _set_up()
 
     with grpc.insecure_channel(f"localhost:{PORT}") as channel:
-        stub = TestServiceStub(channel)
+        stub = gRPCTestServiceStub(channel)
 
         with start_transaction() as transaction:
             metadata = (
@@ -75,7 +75,7 @@ def test_grpc_server_continues_transaction(sentry_init, capture_events_forksafe)
                     ),
                 ),
             )
-            stub.TestServe(TestMessage(text="test"), metadata=metadata)
+            stub.TestServe(gRPCTestMessage(text="test"), metadata=metadata)
 
     _tear_down(server=server)
 
@@ -102,10 +102,10 @@ def test_grpc_client_starts_span(sentry_init, capture_events_forksafe):
 
     with grpc.insecure_channel(f"localhost:{PORT}") as channel:
         channel = grpc.intercept_channel(channel, *interceptors)
-        stub = TestServiceStub(channel)
+        stub = gRPCTestServiceStub(channel)
 
         with start_transaction():
-            stub.TestServe(TestMessage(text="test"))
+            stub.TestServe(gRPCTestMessage(text="test"))
 
     _tear_down(server=server)
 
@@ -139,10 +139,10 @@ def test_grpc_client_and_servers_interceptors_integration(
 
     with grpc.insecure_channel(f"localhost:{PORT}") as channel:
         channel = grpc.intercept_channel(channel, *interceptors)
-        stub = TestServiceStub(channel)
+        stub = gRPCTestServiceStub(channel)
 
         with start_transaction():
-            stub.TestServe(TestMessage(text="test"))
+            stub.TestServe(gRPCTestMessage(text="test"))
 
     _tear_down(server=server)
 
@@ -162,7 +162,7 @@ def _set_up():
         interceptors=[ServerInterceptor(find_name=_find_name)],
     )
 
-    add_TestServiceServicer_to_server(TestService, server)
+    add_gRPCTestServiceServicer_to_server(TestService, server)
     server.add_insecure_port(f"[::]:{PORT}")
     server.start()
 
@@ -177,7 +177,7 @@ def _find_name(request):
     return request.__class__
 
 
-class TestService(TestServiceServicer):
+class TestService(gRPCTestServiceServicer):
     events = []
 
     @staticmethod
@@ -186,4 +186,4 @@ def TestServe(request, context):  # noqa: N802
         with hub.start_span(op="test", description="test"):
             pass
 
-        return TestMessage(text=request.text)
+        return gRPCTestMessage(text=request.text)