From f76fb756be675ddaf6ab4f24f2cca8a8a4f9508c Mon Sep 17 00:00:00 2001
From: Mateusz Rzeszutek <mrzeszutek@splunk.com>
Date: Tue, 22 Sep 2020 09:37:36 +0200
Subject: [PATCH] Add smoke test for Logback MDC instrumentation (#1227)

---
 smoke-tests/smoke-tests.gradle                |  1 +
 .../opentelemetry/smoketest/SmokeTest.groovy  | 19 +++++++++++++++++++
 .../smoketest/SpringBootSmokeTest.groovy      | 14 +++++++++++---
 3 files changed, 31 insertions(+), 3 deletions(-)

diff --git a/smoke-tests/smoke-tests.gradle b/smoke-tests/smoke-tests.gradle
index 8b82a0e7e0..b5fcb57995 100644
--- a/smoke-tests/smoke-tests.gradle
+++ b/smoke-tests/smoke-tests.gradle
@@ -8,6 +8,7 @@ dependencies {
 
   implementation deps.slf4j
   implementation deps.opentelemetryProto
+  implementation deps.opentelemetryApi
   implementation "org.testcontainers:testcontainers:1.14.3"
   implementation 'com.fasterxml.jackson.core:jackson-databind:2.11.2'
   implementation 'com.google.protobuf:protobuf-java-util:3.12.4'
diff --git a/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SmokeTest.groovy b/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SmokeTest.groovy
index 3c20786137..597b6bf043 100644
--- a/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SmokeTest.groovy
+++ b/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SmokeTest.groovy
@@ -16,6 +16,8 @@
 
 package io.opentelemetry.smoketest
 
+import static java.util.stream.Collectors.toSet
+
 import com.fasterxml.jackson.databind.ObjectMapper
 import com.google.protobuf.util.JsonFormat
 import io.opentelemetry.auto.test.utils.OkHttpUtils
@@ -23,6 +25,7 @@ import io.opentelemetry.proto.collector.trace.v1.ExportTraceServiceRequest
 import io.opentelemetry.proto.common.v1.AnyValue
 import io.opentelemetry.proto.trace.v1.Span
 import java.util.concurrent.TimeUnit
+import java.util.regex.Pattern
 import java.util.stream.Stream
 import okhttp3.OkHttpClient
 import okhttp3.Request
@@ -31,6 +34,7 @@ import org.slf4j.LoggerFactory
 import org.testcontainers.containers.GenericContainer
 import org.testcontainers.containers.Network
 import org.testcontainers.containers.output.Slf4jLogConsumer
+import org.testcontainers.containers.output.ToStringConsumer
 import org.testcontainers.containers.wait.strategy.Wait
 import org.testcontainers.utility.MountableFile
 import spock.lang.Shared
@@ -38,6 +42,7 @@ import spock.lang.Specification
 
 abstract class SmokeTest extends Specification {
   private static final Logger logger = LoggerFactory.getLogger(SmokeTest)
+  private static final Pattern TRACE_ID_PATTERN = Pattern.compile(".*traceId=(?<traceId>[a-zA-Z0-9]+).*")
 
   private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper()
 
@@ -86,9 +91,11 @@ abstract class SmokeTest extends Specification {
   }
 
   def startTarget(int jdk) {
+    def output = new ToStringConsumer()
     target = new GenericContainer<>(getTargetImage(jdk))
       .withExposedPorts(8080)
       .withNetwork(network)
+      .withLogConsumer(output)
       .withLogConsumer(new Slf4jLogConsumer(logger))
       .withCopyFileToContainer(MountableFile.forHostPath(agentPath), "/opentelemetry-javaagent-all.jar")
       .withEnv("JAVA_TOOL_OPTIONS", "-javaagent:/opentelemetry-javaagent-all.jar")
@@ -97,6 +104,7 @@ abstract class SmokeTest extends Specification {
       .withEnv("OTEL_OTLP_ENDPOINT", "collector:55680")
       .withEnv(extraEnv)
     target.start()
+    output
   }
 
   def cleanup() {
@@ -172,4 +180,15 @@ abstract class SmokeTest extends Specification {
 
     return content
   }
+
+  protected static Set<String> getLoggedTraceIds(ToStringConsumer output) {
+    output.toUtf8String().lines()
+      .flatMap(SmokeTest.&findTraceId)
+      .collect(toSet())
+  }
+
+  private static Stream<String> findTraceId(String log) {
+    def m = TRACE_ID_PATTERN.matcher(log)
+    m.matches() ? Stream.of(m.group("traceId")) : Stream.empty() as Stream<String>
+  }
 }
diff --git a/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SpringBootSmokeTest.groovy b/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SpringBootSmokeTest.groovy
index 4af6362dd7..b4ed2d11af 100644
--- a/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SpringBootSmokeTest.groovy
+++ b/smoke-tests/src/test/groovy/io/opentelemetry/smoketest/SpringBootSmokeTest.groovy
@@ -16,12 +16,14 @@
 
 package io.opentelemetry.smoketest
 
+import static java.util.stream.Collectors.toSet
 
 import io.opentelemetry.proto.collector.trace.v1.ExportTraceServiceRequest
+import io.opentelemetry.trace.TraceId
 import java.util.jar.Attributes
 import java.util.jar.JarFile
-import java.util.stream.Collectors
 import okhttp3.Request
+import spock.lang.Unroll
 
 class SpringBootSmokeTest extends SmokeTest {
 
@@ -29,9 +31,10 @@ class SpringBootSmokeTest extends SmokeTest {
     "open-telemetry-docker-dev.bintray.io/java/smoke-springboot-jdk$jdk:latest"
   }
 
+  @Unroll
   def "spring boot smoke test on JDK #jdk"(int jdk) {
     setup:
-    startTarget(jdk)
+    def output = startTarget(jdk)
     String url = "http://localhost:${target.getMappedPort(8080)}/greeting"
     def request = new Request.Builder().url(url).get().build()
 
@@ -49,7 +52,12 @@ class SpringBootSmokeTest extends SmokeTest {
 
     [currentAgentVersion] as Set == findResourceAttribute(traces, "telemetry.auto.version")
       .map { it.stringValue }
-      .collect(Collectors.toSet())
+      .collect(toSet())
+
+    then: "correct traceIds are logged via MDC instrumentation"
+    getLoggedTraceIds(output) == getSpanStream(traces)
+      .map({ TraceId.bytesToHex(it.getTraceId().toByteArray()) })
+      .collect(toSet())
 
     cleanup:
     stopTarget()