From 09adbd02324cf03c2a382d118f39e64f0bed6f8d Mon Sep 17 00:00:00 2001
From: Stuart McCulloch <stuart.mcculloch@datadoghq.com>
Date: Mon, 29 Jun 2026 14:07:28 +0100
Subject: [PATCH] Extend OTLP HTTP coverage to GRPC

---
 dd-trace-core/build.gradle                    |  7 +-
 .../otlp/common/OtlpGrpcRequestBodyTest.java  | 91 +++++++++++++++++++
 2 files changed, 97 insertions(+), 1 deletion(-)
 create mode 100644 dd-trace-core/src/test/java/datadog/trace/core/otlp/common/OtlpGrpcRequestBodyTest.java

diff --git a/dd-trace-core/build.gradle b/dd-trace-core/build.gradle
index ee8ef78ee16..0da1ae17470 100644
--- a/dd-trace-core/build.gradle
+++ b/dd-trace-core/build.gradle
@@ -54,7 +54,12 @@ excludedClassesCoverage += [
   // TODO CorePropagation will be removed during context refactoring
   'datadog.trace.core.propagation.CorePropagation',
   // TODO DSM propagator will be tested once fully migrated
-  'datadog.trace.core.datastreams.DataStreamPropagator'
+  'datadog.trace.core.datastreams.DataStreamPropagator',
+  // send() requires live HTTP/2 server
+  'datadog.trace.core.otlp.common.OtlpGrpcSender',
+  // covered by OTLP system-tests
+  'datadog.trace.core.otlp.logs.OtlpLogsService',
+  'datadog.trace.core.otlp.metrics.OtlpMetricsService'
 ]
 
 addTestSuite('traceAgentTest')
diff --git a/dd-trace-core/src/test/java/datadog/trace/core/otlp/common/OtlpGrpcRequestBodyTest.java b/dd-trace-core/src/test/java/datadog/trace/core/otlp/common/OtlpGrpcRequestBodyTest.java
new file mode 100644
index 00000000000..702be21873c
--- /dev/null
+++ b/dd-trace-core/src/test/java/datadog/trace/core/otlp/common/OtlpGrpcRequestBodyTest.java
@@ -0,0 +1,91 @@
+package datadog.trace.core.otlp.common;
+
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.ByteBuffer;
+import java.util.zip.GZIPInputStream;
+import okhttp3.MediaType;
+import okio.Buffer;
+import org.junit.jupiter.api.Test;
+
+class OtlpGrpcRequestBodyTest {
+
+  @Test
+  void contentTypeIsApplicationGrpc() {
+    OtlpGrpcRequestBody body =
+        new OtlpGrpcRequestBody(
+            new OtlpPayload(ByteBuffer.wrap(new byte[] {1, 2, 3}), "application/grpc+proto"),
+            false);
+
+    assertEquals(MediaType.get("application/grpc"), body.contentType());
+  }
+
+  @Test
+  void contentLengthIsHeaderPlusPayloadWhenUncompressed() {
+    OtlpGrpcRequestBody body =
+        new OtlpGrpcRequestBody(
+            new OtlpPayload(ByteBuffer.wrap(new byte[] {1, 2, 3, 4}), "application/grpc+proto"),
+            false);
+
+    // 5-byte header (1 flag + 4 length) + 4-byte payload = 9
+    assertEquals(9, body.contentLength());
+  }
+
+  @Test
+  void contentLengthIsNegativeOneWhenGzipped() {
+    OtlpGrpcRequestBody body =
+        new OtlpGrpcRequestBody(
+            new OtlpPayload(ByteBuffer.wrap(new byte[] {1, 2, 3, 4}), "application/grpc+proto"),
+            true);
+
+    assertEquals(-1, body.contentLength());
+  }
+
+  @Test
+  void writeToProducesGrpcFrameWithUncompressedFlagWhenNotGzipped() throws IOException {
+    byte[] data = {10, 20, 30, 40, 50};
+    OtlpGrpcRequestBody body =
+        new OtlpGrpcRequestBody(
+            new OtlpPayload(ByteBuffer.wrap(data), "application/grpc+proto"), false);
+    Buffer sink = new Buffer();
+
+    body.writeTo(sink);
+
+    assertEquals(0, sink.readByte()); // uncompressed flag
+    assertEquals(data.length, sink.readInt()); // 4-byte big-endian length
+    assertArrayEquals(data, sink.readByteArray()); // payload
+  }
+
+  @Test
+  void writeToProducesGrpcFrameWithCompressedFlagAndGzipDataWhenGzipped() throws IOException {
+    byte[] data = "the quick brown fox jumps over the lazy dog".getBytes();
+    OtlpGrpcRequestBody body =
+        new OtlpGrpcRequestBody(
+            new OtlpPayload(ByteBuffer.wrap(data), "application/grpc+proto"), true);
+    Buffer sink = new Buffer();
+
+    body.writeTo(sink);
+
+    assertEquals(1, sink.readByte()); // compressed flag
+    int gzipLength = sink.readInt(); // 4-byte big-endian gzip content length
+    byte[] gzipped = sink.readByteArray(gzipLength);
+    assertArrayEquals(data, gunzip(gzipped));
+  }
+
+  private static byte[] gunzip(byte[] gzipped) throws IOException {
+    ByteArrayOutputStream out = new ByteArrayOutputStream();
+    try (InputStream gz = new GZIPInputStream(new ByteArrayInputStream(gzipped))) {
+      byte[] buf = new byte[256];
+      int n;
+      while ((n = gz.read(buf)) > 0) {
+        out.write(buf, 0, n);
+      }
+    }
+    return out.toByteArray();
+  }
+}