Add additional test for authenticator

exporters/otlp/all/src/test/java/io/opentelemetry/exporter/otlp/http/metrics/OtlpHttpMetricExporterTest.java

@@ -9,6 +9,7 @@
 import static org.assertj.core.api.Assertions.assertThatCode;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
 
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.marshal.Marshaler;
 import io.opentelemetry.exporter.internal.otlp.metrics.ResourceMetricsMarshaler;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
@@ -114,6 +115,12 @@ public TelemetryExporterBuilder<MetricData> addHeader(String key, String value)
         return this;
       }
 
+      @Override
+      public TelemetryExporterBuilder<MetricData> setAuthenticator(Authenticator authenticator) {
+        Authenticator.setAuthenticatorOnDelegate(builder, authenticator);
+        return this;
+      }
+
       @Override
       public TelemetryExporterBuilder<MetricData> setTrustedCertificates(byte[] certificates) {
         builder.setTrustedCertificates(certificates);

exporters/otlp/all/src/test/java/io/opentelemetry/exporter/otlp/http/trace/OtlpHttpSpanExporterTest.java

@@ -5,6 +5,7 @@
 
 package io.opentelemetry.exporter.otlp.http.trace;
 
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.marshal.Marshaler;
 import io.opentelemetry.exporter.internal.otlp.traces.ResourceSpansMarshaler;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
@@ -59,6 +60,12 @@ public TelemetryExporterBuilder<SpanData> addHeader(String key, String value) {
         return this;
       }
 
+      @Override
+      public TelemetryExporterBuilder<SpanData> setAuthenticator(Authenticator authenticator) {
+        Authenticator.setAuthenticatorOnDelegate(builder, authenticator);
+        return this;
+      }
+
       @Override
       public TelemetryExporterBuilder<SpanData> setTrustedCertificates(byte[] certificates) {
         builder.setTrustedCertificates(certificates);

exporters/otlp/logs/src/test/java/io/opentelemetry/exporter/otlp/http/logs/OtlpHttpLogRecordExporterTest.java

@@ -5,6 +5,7 @@
 
 package io.opentelemetry.exporter.otlp.http.logs;
 
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.marshal.Marshaler;
 import io.opentelemetry.exporter.internal.otlp.logs.ResourceLogsMarshaler;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
@@ -60,6 +61,12 @@ public TelemetryExporterBuilder<LogRecordData> addHeader(String key, String valu
         return this;
       }
 
+      @Override
+      public TelemetryExporterBuilder<LogRecordData> setAuthenticator(Authenticator authenticator) {
+        Authenticator.setAuthenticatorOnDelegate(builder, authenticator);
+        return this;
+      }
+
       @Override
       public TelemetryExporterBuilder<LogRecordData> setTrustedCertificates(byte[] certificates) {
         builder.setTrustedCertificates(certificates);

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/AbstractHttpTelemetryExporterTest.java

@@ -333,6 +333,34 @@ void withHeaders() {
     }
   }
 
+  @Test
+  void withAuthenticator() {
+    TelemetryExporter<T> exporter =
+        exporterBuilder()
+            .setEndpoint(server.httpUri() + path)
+            .setAuthenticator(() -> Collections.singletonMap("key", "value"))
+            .build();
+
+    addHttpError(401);
+
+    try {
+      assertThat(
+              exporter
+                  .export(Collections.singletonList(generateFakeTelemetry()))
+                  .join(10, TimeUnit.SECONDS)
+                  .isSuccess())
+          .isTrue();
+      assertThat(httpRequests)
+          .element(0)
+          .satisfies(req -> assertThat(req.headers().get("key")).isNull());
+      assertThat(httpRequests)
+          .element(1)
+          .satisfies(req -> assertThat(req.headers().get("key")).isEqualTo("value"));
+    } finally {
+      exporter.shutdown();
+    }
+  }
+
   @Test
   void tls() throws Exception {
     TelemetryExporter<T> exporter =

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/GrpcLogRecordExporterBuilderWrapper.java

@@ -6,6 +6,7 @@
 package io.opentelemetry.exporter.otlp.testing.internal;
 
 import io.grpc.ManagedChannel;
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
 import io.opentelemetry.exporter.internal.retry.RetryUtil;
 import io.opentelemetry.exporter.otlp.logs.OtlpGrpcLogRecordExporterBuilder;
@@ -50,6 +51,11 @@ public TelemetryExporterBuilder<LogRecordData> addHeader(String key, String valu
     return this;
   }
 
+  @Override
+  public TelemetryExporterBuilder<LogRecordData> setAuthenticator(Authenticator authenticator) {
+    return this;
+  }
+
   @Override
   public TelemetryExporterBuilder<LogRecordData> setTrustedCertificates(byte[] certificates) {
     builder.setTrustedCertificates(certificates);

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/GrpcMetricExporterBuilderWrapper.java

@@ -6,6 +6,7 @@
 package io.opentelemetry.exporter.otlp.testing.internal;
 
 import io.grpc.ManagedChannel;
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
 import io.opentelemetry.exporter.internal.retry.RetryUtil;
 import io.opentelemetry.exporter.otlp.metrics.OtlpGrpcMetricExporterBuilder;
@@ -50,6 +51,11 @@ public TelemetryExporterBuilder<MetricData> addHeader(String key, String value)
     return this;
   }
 
+  @Override
+  public TelemetryExporterBuilder<MetricData> setAuthenticator(Authenticator authenticator) {
+    return this;
+  }
+
   @Override
   public TelemetryExporterBuilder<MetricData> setTrustedCertificates(byte[] certificates) {
     builder.setTrustedCertificates(certificates);

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/GrpcSpanExporterBuilderWrapper.java

@@ -6,6 +6,7 @@
 package io.opentelemetry.exporter.otlp.testing.internal;
 
 import io.grpc.ManagedChannel;
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
 import io.opentelemetry.exporter.internal.retry.RetryUtil;
 import io.opentelemetry.exporter.otlp.trace.OtlpGrpcSpanExporterBuilder;
@@ -51,6 +52,11 @@ public TelemetryExporterBuilder<SpanData> addHeader(String key, String value) {
     return this;
   }
 
+  @Override
+  public TelemetryExporterBuilder<SpanData> setAuthenticator(Authenticator authenticator) {
+    return this;
+  }
+
   @Override
   public TelemetryExporterBuilder<SpanData> setTrustedCertificates(byte[] certificates) {
     builder.setTrustedCertificates(certificates);

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/ManagedChannelTelemetryExporterBuilder.java

@@ -13,6 +13,7 @@
 import io.grpc.netty.NettyChannelBuilder;
 import io.netty.handler.ssl.SslContext;
 import io.opentelemetry.exporter.internal.TlsConfigHelper;
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.grpc.ManagedChannelUtil;
 import io.opentelemetry.exporter.internal.otlp.OtlpUserAgent;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
@@ -87,6 +88,11 @@ public TelemetryExporterBuilder<T> addHeader(String key, String value) {
     return this;
   }
 
+  @Override
+  public TelemetryExporterBuilder<T> setAuthenticator(Authenticator authenticator) {
+    return this;
+  }
+
   // When a user provides a Channel, we are not in control of TLS or retry config and reimplement it
   // here for use in tests. Technically we don't have to test them since they are out of the SDK's
   // control, but it's probably worth verifying the baseline functionality anyways.

exporters/otlp/testing-internal/src/main/java/io/opentelemetry/exporter/otlp/testing/internal/TelemetryExporterBuilder.java

@@ -6,6 +6,7 @@
 package io.opentelemetry.exporter.otlp.testing.internal;
 
 import io.grpc.ManagedChannel;
+import io.opentelemetry.exporter.internal.auth.Authenticator;
 import io.opentelemetry.exporter.internal.retry.RetryPolicy;
 import io.opentelemetry.exporter.otlp.logs.OtlpGrpcLogRecordExporterBuilder;
 import io.opentelemetry.exporter.otlp.metrics.OtlpGrpcMetricExporterBuilder;
@@ -40,6 +41,8 @@ static TelemetryExporterBuilder<LogRecordData> wrap(OtlpGrpcLogRecordExporterBui
 
   TelemetryExporterBuilder<T> addHeader(String key, String value);
 
+  TelemetryExporterBuilder<T> setAuthenticator(Authenticator authenticator);
+
   TelemetryExporterBuilder<T> setTrustedCertificates(byte[] certificates);
 
   TelemetryExporterBuilder<T> setClientTls(byte[] privateKeyPem, byte[] certificatePem);