add test brigde for exporters

Author: zeitlinger

instrumentation/methods/javaagent/src/declarativeConfigTest/resources/declarative-config.yaml

@@ -1,4 +1,28 @@
 file_format: "0.4"
+tracer_provider:
+  processors:
+    - simple:
+        exporter:
+          test_bridge: { }
+    - simple:
+        exporter:
+          console: { }
+
+logger_provider:
+  processors:
+    - simple:
+        exporter:
+          test_bridge: { }
+
+meter_provider:
+  readers:
+    - periodic:
+        # Set really long interval. We'll call forceFlush when we need the metrics
+        # instead of collecting them periodically.
+        interval: 1000000
+        exporter:
+          test_bridge: { }
+
 instrumentation/development:
   java:
     methods:

testing-common/build.gradle.kts

@@ -69,6 +69,7 @@ dependencies {
   implementation("org.slf4j:jul-to-slf4j")
   implementation("io.opentelemetry:opentelemetry-exporter-logging")
   implementation("io.opentelemetry.contrib:opentelemetry-baggage-processor")
+  implementation("io.opentelemetry:opentelemetry-sdk-extension-autoconfigure-spi")
   api(project(":instrumentation-api-incubator"))
 
   annotationProcessor("com.google.auto.service:auto-service")

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/LibraryTestRunner.java

@@ -10,6 +10,7 @@
 import io.opentelemetry.api.GlobalOpenTelemetry;
 import io.opentelemetry.api.OpenTelemetry;
 import io.opentelemetry.api.baggage.propagation.W3CBaggagePropagator;
+import io.opentelemetry.api.incubator.config.GlobalConfigProvider;
 import io.opentelemetry.api.logs.LoggerProvider;
 import io.opentelemetry.api.metrics.MeterProvider;
 import io.opentelemetry.api.trace.TracerBuilder;
@@ -20,6 +21,9 @@
 import io.opentelemetry.context.propagation.TextMapPropagator;
 import io.opentelemetry.contrib.baggage.processor.BaggageSpanProcessor;
 import io.opentelemetry.exporter.logging.LoggingSpanExporter;
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeLogRecordExporterComponentProvider;
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeMetricExporterComponentProvider;
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeSpanExporterComponentProvider;
 import io.opentelemetry.sdk.OpenTelemetrySdk;
 import io.opentelemetry.sdk.common.CompletableResultCode;
 import io.opentelemetry.sdk.logs.SdkLoggerProvider;
@@ -59,10 +63,14 @@ public final class LibraryTestRunner extends InstrumentationTestRunner {
 
   static {
     GlobalOpenTelemetry.resetForTest();
+    GlobalConfigProvider.resetForTest();
 
     testSpanExporter = InMemorySpanExporter.create();
     testMetricExporter = InMemoryMetricExporter.create(AggregationTemporality.DELTA);
     testLogRecordExporter = InMemoryLogRecordExporter.create();
+    TestBridgeSpanExporterComponentProvider.setSpanExporter(testSpanExporter);
+    TestBridgeMetricExporterComponentProvider.setMetricExporter(testMetricExporter);
+    TestBridgeLogRecordExporterComponentProvider.setLogRecordExporter(testLogRecordExporter);
 
     metricReader =
         PeriodicMetricReader.builder(testMetricExporter)
@@ -112,6 +120,7 @@ private LibraryTestRunner() {
   public void beforeTestClass() {
     // just in case: if there was any test that modified the global instance, reset it
     GlobalOpenTelemetry.resetForTest();
+    GlobalConfigProvider.resetForTest();
     GlobalOpenTelemetry.set(openTelemetrySdk);
   }
 

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/junit/InstrumentationExtension.java

@@ -95,6 +95,7 @@ public final void waitAndAssertMetrics(
   }
 
   @SafeVarargs
+  @SuppressWarnings("varargs")
   public final void waitAndAssertMetrics(
       String instrumentationName, Consumer<MetricAssert>... assertions) {
     testRunner.waitAndAssertMetrics(instrumentationName, assertions);

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/provider/TestBridgeLogRecordExporterComponentProvider.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.testing.provider;
+
+import io.opentelemetry.api.incubator.config.DeclarativeConfigProperties;
+import io.opentelemetry.sdk.autoconfigure.spi.internal.ComponentProvider;
+import io.opentelemetry.sdk.logs.export.LogRecordExporter;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class TestBridgeLogRecordExporterComponentProvider
+    implements ComponentProvider<LogRecordExporter> {
+
+  private static final Logger logger =
+      LoggerFactory.getLogger(TestBridgeLogRecordExporterComponentProvider.class);
+
+  private static LogRecordExporter logRecordExporter;
+
+  @Override
+  public Class<LogRecordExporter> getType() {
+    return LogRecordExporter.class;
+  }
+
+  @Override
+  public String getName() {
+    return "test_bridge";
+  }
+
+  @Override
+  public LogRecordExporter create(DeclarativeConfigProperties config) {
+    return logRecordExporter;
+  }
+
+  public static void setLogRecordExporter(LogRecordExporter logRecordExporter) {
+    logger.info(
+        "Setting TestLogRecordExporterComponentProvider logRecord exporter to {}",
+        logRecordExporter.getClass().getName());
+    TestBridgeLogRecordExporterComponentProvider.logRecordExporter = logRecordExporter;
+  }
+}

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/provider/TestBridgeMetricExporterComponentProvider.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.testing.provider;
+
+import io.opentelemetry.api.incubator.config.DeclarativeConfigProperties;
+import io.opentelemetry.sdk.autoconfigure.spi.internal.ComponentProvider;
+import io.opentelemetry.sdk.metrics.export.MetricExporter;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class TestBridgeMetricExporterComponentProvider
+    implements ComponentProvider<MetricExporter> {
+
+  private static final Logger logger =
+      LoggerFactory.getLogger(TestBridgeMetricExporterComponentProvider.class);
+
+  private static MetricExporter metricExporter;
+
+  @Override
+  public Class<MetricExporter> getType() {
+    return MetricExporter.class;
+  }
+
+  @Override
+  public String getName() {
+    return "test_bridge";
+  }
+
+  @Override
+  public MetricExporter create(DeclarativeConfigProperties config) {
+    return metricExporter;
+  }
+
+  public static void setMetricExporter(MetricExporter metricExporter) {
+    logger.info(
+        "Setting TestMetricExporterComponentProvider metric exporter to {}",
+        metricExporter.getClass().getName());
+    TestBridgeMetricExporterComponentProvider.metricExporter = metricExporter;
+  }
+}

testing-common/src/main/java/io/opentelemetry/instrumentation/testing/provider/TestBridgeSpanExporterComponentProvider.java

@@ -0,0 +1,42 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.testing.provider;
+
+import io.opentelemetry.api.incubator.config.DeclarativeConfigProperties;
+import io.opentelemetry.sdk.autoconfigure.spi.internal.ComponentProvider;
+import io.opentelemetry.sdk.trace.export.SpanExporter;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class TestBridgeSpanExporterComponentProvider implements ComponentProvider<SpanExporter> {
+
+  private static final Logger logger =
+      LoggerFactory.getLogger(TestBridgeSpanExporterComponentProvider.class);
+
+  private static SpanExporter spanExporter;
+
+  @Override
+  public Class<SpanExporter> getType() {
+    return SpanExporter.class;
+  }
+
+  @Override
+  public String getName() {
+    return "test_bridge";
+  }
+
+  @Override
+  public SpanExporter create(DeclarativeConfigProperties config) {
+    return spanExporter;
+  }
+
+  public static void setSpanExporter(SpanExporter spanExporter) {
+    logger.info(
+        "Setting TestSpanExporterComponentProvider span exporter to {}",
+        spanExporter.getClass().getName());
+    TestBridgeSpanExporterComponentProvider.spanExporter = spanExporter;
+  }
+}

testing-common/src/main/resources/META-INF/services/io.opentelemetry.sdk.autoconfigure.spi.internal.ComponentProvider

@@ -0,0 +1,4 @@
+io.opentelemetry.instrumentation.testing.provider.TestBridgeSpanExporterComponentProvider
+io.opentelemetry.instrumentation.testing.provider.TestBridgeMetricExporterComponentProvider
+io.opentelemetry.instrumentation.testing.provider.TestBridgeLogRecordExporterComponentProvider
+

testing/agent-exporter/build.gradle.kts

@@ -15,6 +15,7 @@ dependencies {
   compileOnly(project(":javaagent-extension-api"))
   compileOnly(project(":javaagent-bootstrap"))
   compileOnly(project(":javaagent-tooling"))
+  implementation(project(":testing-common"))
 
   implementation("io.opentelemetry:opentelemetry-exporter-otlp-common")
   compileOnly("io.opentelemetry:opentelemetry-sdk-extension-autoconfigure")

testing/agent-exporter/src/main/java/io/opentelemetry/javaagent/testing/exporter/AgentTestingExporterFactory.java

@@ -5,6 +5,9 @@
 
 package io.opentelemetry.javaagent.testing.exporter;
 
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeLogRecordExporterComponentProvider;
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeMetricExporterComponentProvider;
+import io.opentelemetry.instrumentation.testing.provider.TestBridgeSpanExporterComponentProvider;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 
@@ -14,6 +17,12 @@ public final class AgentTestingExporterFactory {
   static final OtlpInMemoryMetricExporter metricExporter = new OtlpInMemoryMetricExporter();
   static final OtlpInMemoryLogRecordExporter logExporter = new OtlpInMemoryLogRecordExporter();
 
+  static {
+    TestBridgeSpanExporterComponentProvider.setSpanExporter(spanExporter);
+    TestBridgeMetricExporterComponentProvider.setMetricExporter(metricExporter);
+    TestBridgeLogRecordExporterComponentProvider.setLogRecordExporter(logExporter);
+  }
+
   public static List<byte[]> getSpanExportRequests() {
     return spanExporter.getCollectedExportRequests();
   }