fix test

Author: crossoverJie

instrumentation/armeria/armeria-grpc-1.14/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/armeria/grpc/v1_14/ArmeriaGrpcTest.java

@@ -22,6 +22,7 @@
 import example.Helloworld;
 import io.grpc.Status;
 import io.grpc.stub.StreamObserver;
+import io.opentelemetry.api.common.AttributeKey;
 import io.opentelemetry.api.trace.SpanKind;
 import io.opentelemetry.instrumentation.testing.junit.AgentInstrumentationExtension;
 import io.opentelemetry.semconv.incubating.MessageIncubatingAttributes;
@@ -33,6 +34,15 @@ class ArmeriaGrpcTest {
   @RegisterExtension
   static final AgentInstrumentationExtension testing = AgentInstrumentationExtension.create();
 
+  private static final AttributeKey<Long> RPC_CLIENT_REQUEST_BODY_SIZE =
+      AttributeKey.longKey("rpc.client.request.body.size");
+  private static final AttributeKey<Long> RPC_CLIENT_RESPONSE_BODY_SIZE =
+      AttributeKey.longKey("rpc.client.response.body.size");
+  private static final AttributeKey<Long> RPC_SERVER_REQUEST_BODY_SIZE =
+      AttributeKey.longKey("rpc.server.request.body.size");
+  private static final AttributeKey<Long> RPC_SERVER_RESPONSE_BODY_SIZE =
+      AttributeKey.longKey("rpc.server.response.body.size");
+
   @RegisterExtension
   static final ServerExtension server =
       new ServerExtension() {
@@ -63,12 +73,12 @@ void grpcInstrumentation() {
     GreeterGrpc.GreeterBlockingStub client =
         GrpcClients.builder(server.httpUri()).build(GreeterGrpc.GreeterBlockingStub.class);
 
-    Helloworld.Response response =
-        testing.runWithSpan(
-            "parent",
-            () -> client.sayHello(Helloworld.Request.newBuilder().setName("test").build()));
+    Helloworld.Request request = Helloworld.Request.newBuilder().setName("test").build();
+    Helloworld.Response response = testing.runWithSpan("parent", () -> client.sayHello(request));
 
     assertThat(response.getMessage()).isEqualTo("Hello test");
+    int requestSerializedSize = request.getSerializedSize();
+    int responseSerializedSize = response.getSerializedSize();
 
     testing.waitAndAssertTraces(
         trace ->
@@ -85,6 +95,8 @@ void grpcInstrumentation() {
                             equalTo(RPC_GRPC_STATUS_CODE, (long) Status.Code.OK.value()),
                             equalTo(SERVER_ADDRESS, "127.0.0.1"),
                             equalTo(SERVER_PORT, (long) server.httpPort()))
+                            equalTo(RPC_CLIENT_RESPONSE_BODY_SIZE, responseSerializedSize),
+                            equalTo(RPC_CLIENT_REQUEST_BODY_SIZE, requestSerializedSize))
                         .hasEventsSatisfyingExactly(
                             event ->
                                 event
@@ -110,6 +122,8 @@ void grpcInstrumentation() {
                             equalTo(RPC_GRPC_STATUS_CODE, (long) Status.Code.OK.value()),
                             equalTo(SERVER_ADDRESS, "127.0.0.1"),
                             equalTo(SERVER_PORT, server.httpPort()))
+                            equalTo(RPC_SERVER_REQUEST_BODY_SIZE, responseSerializedSize),
+                            equalTo(RPC_SERVER_RESPONSE_BODY_SIZE, requestSerializedSize))
                         .hasEventsSatisfyingExactly(
                             event ->
                                 event