io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions Maven / Gradle / Ivy

Go to download
/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

package io.opentelemetry.instrumentation.testing.junit.http;

import com.google.errorprone.annotations.CanIgnoreReturnValue;
import io.opentelemetry.api.common.AttributeKey;
import io.opentelemetry.semconv.trace.attributes.SemanticAttributes;
import java.net.URI;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
import java.util.function.BiFunction;
import java.util.function.Function;

public final class HttpClientTestOptions {

  public static final Set> DEFAULT_HTTP_ATTRIBUTES =
      Collections.unmodifiableSet(
          new HashSet<>(
              Arrays.asList(
                  SemanticAttributes.NET_PEER_NAME,
                  SemanticAttributes.NET_PEER_PORT,
                  SemanticAttributes.HTTP_URL,
                  SemanticAttributes.HTTP_METHOD,
                  SemanticAttributes.HTTP_FLAVOR,
                  SemanticAttributes.HTTP_USER_AGENT)));

  public static final BiFunction DEFAULT_EXPECTED_CLIENT_SPAN_NAME_MAPPER =
      (uri, method) -> method != null ? "HTTP " + method : "HTTP request";

  Function>> httpAttributes = unused -> DEFAULT_HTTP_ATTRIBUTES;

  BiFunction expectedClientSpanNameMapper =
      DEFAULT_EXPECTED_CLIENT_SPAN_NAME_MAPPER;

  Integer responseCodeOnRedirectError = null;
  String userAgent = null;

  BiFunction clientSpanErrorMapper = (uri, exception) -> exception;

  BiFunction singleConnectionFactory = (host, port) -> null;

  boolean testWithClientParent = true;
  boolean testRedirects = true;
  boolean testCircularRedirects = true;
  int maxRedirects = 2;
  boolean testReusedRequest = true;
  boolean testConnectionFailure = true;
  boolean testReadTimeout = false;
  boolean testRemoteConnection = true;
  boolean testHttps = true;
  boolean testCausality = true;
  boolean testCausalityWithCallback = true;
  boolean testCallback = true;
  boolean testCallbackWithParent = true;
  boolean testCallbackWithImplicitParent = false;
  boolean testErrorWithCallback = true;

  HttpClientTestOptions() {}

  @CanIgnoreReturnValue
  public HttpClientTestOptions setHttpAttributes(
      Function>> httpAttributes) {
    this.httpAttributes = httpAttributes;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setExpectedClientSpanNameMapper(
      BiFunction expectedClientSpanNameMapper) {
    this.expectedClientSpanNameMapper = expectedClientSpanNameMapper;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setResponseCodeOnRedirectError(int responseCodeOnRedirectError) {
    this.responseCodeOnRedirectError = responseCodeOnRedirectError;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setUserAgent(String userAgent) {
    this.userAgent = userAgent;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setClientSpanErrorMapper(
      BiFunction clientSpanErrorMapper) {
    this.clientSpanErrorMapper = clientSpanErrorMapper;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setSingleConnectionFactory(
      BiFunction singleConnectionFactory) {
    this.singleConnectionFactory = singleConnectionFactory;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions setMaxRedirects(int maxRedirects) {
    this.maxRedirects = maxRedirects;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestWithClientParent() {
    testWithClientParent = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestRedirects() {
    testRedirects = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestCircularRedirects() {
    testCircularRedirects = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestReusedRequest() {
    testReusedRequest = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestConnectionFailure() {
    testConnectionFailure = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions enableTestReadTimeout() {
    testReadTimeout = true;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestRemoteConnection() {
    testRemoteConnection = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestHttps() {
    testHttps = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestCausality() {
    testCausality = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestCausalityWithCallback() {
    testCausalityWithCallback = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestCallback() {
    testCallback = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestCallbackWithParent() {
    testCallbackWithParent = false;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions enableTestCallbackWithImplicitParent() {
    testCallbackWithImplicitParent = true;
    return this;
  }

  @CanIgnoreReturnValue
  public HttpClientTestOptions disableTestErrorWithCallback() {
    testErrorWithCallback = false;
    return this;
  }
}