wip

Author: evanchooly

dd-java-agent/instrumentation/kafka-clients-3.8/src/test/groovy/KafkaClientTestBase.groovy

@@ -152,7 +152,7 @@ abstract class KafkaClientTestBase extends VersionedNamingTestBase {
   }
 
   @Flaky
-  def "test kafka produce and consume"() {
+  public def "test kafka produce and consume"() {
     setup:
     // Create and start a Kafka container using Testcontainers
     KafkaContainer kafkaContainer = new KafkaContainer(DockerImageName.parse("confluentinc/cp-kafka:latest")).withEmbeddedZookeeper().withEnv("KAFKA_CREATE_TOPICS", SHARED_TOPIC)

dd-java-agent/instrumentation/kafka-clients-3.8/src/test/groovy/KafkaCodeOriginForkedTest.groovy

@@ -0,0 +1,44 @@
+class KafkaCodeOriginForkedTest extends KafkaClientForkedTest {
+  @Override
+  void configurePreAgent() {
+    super.configurePreAgent()
+
+    println("**********************")
+    codeOriginSetup()
+  }
+
+  def "test flaming dumpster"() {
+    assert false
+    super."test kafka produce and consume"
+  }
+
+  @Override
+  int version() {
+    1
+  }
+
+  @Override
+  String service() {
+    return "KafkaClientTest"
+  }
+
+  @Override
+  String operationForProducer() {
+    "kafka.send"
+  }
+
+  @Override
+  String operationForConsumer() {
+    return "kafka.process"
+  }
+
+  @Override
+  String serviceForTimeInQueue() {
+    "kafka-queue"
+  }
+
+  @Override
+  boolean hasQueueSpan() {
+    false
+  }
+}

dd-java-agent/testing/build.gradle

@@ -66,6 +66,9 @@ dependencies {
   implementation 'org.junit.platform:junit-platform-runner:1.9.0'
   implementation project(':dd-java-agent:appsec')
 
+  implementation project(':dd-java-agent:agent-debugger')
+  implementation libs.bundles.mockito
+
   testImplementation project(':utils:test-utils')
   testImplementation project(':dd-java-agent:instrumentation:trace-annotation')
 

dd-java-agent/testing/src/main/groovy/datadog/trace/agent/test/AgentTestRunner.groovy

@@ -3,6 +3,17 @@ package datadog.trace.agent.test
 import ch.qos.logback.classic.Level
 import ch.qos.logback.classic.Logger
 import ch.qos.logback.classic.util.ContextInitializer
+import com.datadog.debugger.agent.ClassesToRetransformFinder
+import com.datadog.debugger.agent.Configuration
+import com.datadog.debugger.agent.ConfigurationUpdater
+import com.datadog.debugger.agent.DebuggerTransformer
+import com.datadog.debugger.agent.DenyListHelper
+import com.datadog.debugger.agent.JsonSnapshotSerializer
+import com.datadog.debugger.codeorigin.DefaultCodeOriginRecorder
+import com.datadog.debugger.instrumentation.InstrumentationResult
+import com.datadog.debugger.probe.ProbeDefinition
+import com.datadog.debugger.sink.DebuggerSink
+import com.datadog.debugger.sink.ProbeStatusSink
 import com.google.common.collect.Sets
 import datadog.communication.ddagent.DDAgentFeaturesDiscovery
 import datadog.communication.monitor.Monitoring
@@ -28,6 +39,7 @@ import datadog.trace.api.time.SystemTimeSource
 import datadog.trace.bootstrap.ActiveSubsystems
 import datadog.trace.bootstrap.CallDepthThreadLocalMap
 import datadog.trace.bootstrap.InstrumentationErrors
+import datadog.trace.bootstrap.debugger.DebuggerContext
 import datadog.trace.bootstrap.instrumentation.api.AgentDataStreamsMonitoring
 import datadog.trace.bootstrap.instrumentation.api.AgentSpan
 import datadog.trace.bootstrap.instrumentation.api.AgentTracer.TracerAPI
@@ -71,7 +83,10 @@ import static datadog.communication.http.OkHttpUtils.buildHttpClient
 import static datadog.trace.api.ConfigDefaults.DEFAULT_AGENT_HOST
 import static datadog.trace.api.ConfigDefaults.DEFAULT_AGENT_TIMEOUT
 import static datadog.trace.api.ConfigDefaults.DEFAULT_TRACE_AGENT_PORT
+import static datadog.trace.api.config.TraceInstrumentationConfig.CODE_ORIGIN_FOR_SPANS_ENABLED
 import static datadog.trace.bootstrap.instrumentation.api.AgentTracer.closePrevious
+import static org.mockito.Mockito.mock
+import static org.mockito.Mockito.when
 
 /**
  * A spock test runner which automatically applies instrumentation and exposes a global trace
@@ -288,6 +303,39 @@ abstract class AgentTestRunner extends DDSpecification implements AgentBuilder.L
     ((Logger) LoggerFactory.getLogger("org.testcontainers")).setLevel(Level.DEBUG)
   }
 
+  void codeOriginSetup() {
+    injectSysConfig(CODE_ORIGIN_FOR_SPANS_ENABLED, "true", true)
+
+    def configuration = Configuration.builder()
+    .setService("grpc code origin test")
+    .build()
+
+    def config = mock(Config.class)
+    when(config.isDebuggerEnabled()).thenReturn(true)
+    when(config.isDebuggerClassFileDumpEnabled()).thenReturn(true)
+    when(config.isDebuggerVerifyByteCode()).thenReturn(false)
+    when(config.getFinalDebuggerSnapshotUrl())
+    .thenReturn("http://localhost:8126/debugger/v1/input")
+    when(config.getFinalDebuggerSymDBUrl()).thenReturn("http://localhost:8126/symdb/v1/input")
+    when(config.getDebuggerCodeOriginMaxUserFrames()).thenReturn(8)
+
+    def probeStatusSink = mock(ProbeStatusSink.class)
+
+    def sink = new DebuggerSink(config, probeStatusSink)
+    def configurationUpdater = new ConfigurationUpdater(INSTRUMENTATION, DebuggerTransformer::new, config, sink, new ClassesToRetransformFinder())
+
+    def currentTransformer = new DebuggerTransformer(config, configuration, {
+      ProbeDefinition definition, InstrumentationResult result ->
+    }, sink)
+    INSTRUMENTATION.addTransformer(currentTransformer)
+
+    DebuggerContext.initProbeResolver(configurationUpdater)
+    DebuggerContext.initClassFilter(new DenyListHelper(null))
+    DebuggerContext.initValueSerializer(new JsonSnapshotSerializer())
+
+    DebuggerContext.initCodeOrigin(new DefaultCodeOriginRecorder(config, configurationUpdater))
+  }
+
   @SuppressForbidden
   void setupSpec() {