content: /*
   * Copyright The OpenTelemetry Authors
   * SPDX-License-Identifier: Apache-2.0
   */
  
  package io.opentelemetry.instrumentation.javahttpclient;
  
  import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
  import io.opentelemetry.instrumentation.testing.junit.http.AbstractHttpClientTest;
  import io.opentelemetry.instrumentation.testing.junit.http.HttpClientInstrumentationExtension;
  import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions;
  import java.net.http.HttpClient;
  import java.util.Collections;
  import org.junit.jupiter.api.Nested;
  import org.junit.jupiter.api.extension.RegisterExtension;
  
  class JavaHttpClientTest {
  
    abstract static class AbstractTest extends AbstractJavaHttpClientTest {
      @RegisterExtension
      static final InstrumentationExtension testing = HttpClientInstrumentationExtension.forLibrary();
  
      @Override
      protected HttpClient configureHttpClient(HttpClient httpClient) {
        return JavaHttpClientTelemetry.builder(testing.getOpenTelemetry())
            .setCapturedRequestHeaders(
                Collections.singletonList(AbstractHttpClientTest.TEST_REQUEST_HEADER))
            .setCapturedResponseHeaders(
                Collections.singletonList(AbstractHttpClientTest.TEST_RESPONSE_HEADER))
            .build()
            .newHttpClient(httpClient);
      }
    }
  
    @Nested
    class Http1ClientTest extends AbstractTest {
  
      @Override
      protected void configureHttpClientBuilder(HttpClient.Builder httpClientBuilder) {
        httpClientBuilder.version(HttpClient.Version.HTTP_1_1);
      }
    }
  
    @Nested
    class Http2ClientTest extends AbstractTest {
  
      @Override
      protected void configureHttpClientBuilder(HttpClient.Builder httpClientBuilder) {
        httpClientBuilder.version(HttpClient.Version.HTTP_2);
      }
  
      @Override
      protected void configure(HttpClientTestOptions.Builder optionsBuilder) {
        super.configure(optionsBuilder);
  
        optionsBuilder.setHttpProtocolVersion(
            uri -> {
              String uriString = uri.toString();
              if (uriString.equals("http://localhost:61/")
                  || uriString.equals("https://192.0.2.1/")) {
                return "1.1";
              }
              return "2";
            });
      }
    }
  }
  