import okhttp3.Headers
import okhttp3.Headers.Companion.headersOf
import okio.Buffer
import okio.ByteString
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.Timeout

@Timeout(30)
class RecordedRequestTest {
  private val headers: Headers = Headers.EMPTY

  @Test fun testIPv4() {
    val socket =
      MockWebServerSocket(
        FakeSocket(

    artifactRules = rules
    publishArtifacts = PublishMode.ALWAYS
}

fun BuildType.applyDefaultSettings(
    os: Os = Os.LINUX,
    arch: Arch = Arch.AMD64,
    buildJvm: Jvm = BuildToolBuildJvm,
    timeout: Int = 30,
    artifactRuleOverride: String? = null,
) {
    val defaultArtifactRules =
        """
        *.psoutput => $HIDDEN_ARTIFACT_DESTINATION
        *.threaddump => $HIDDEN_ARTIFACT_DESTINATION

     */
    public void setMaxRetryCount(final int maxRetryCount) {
        this.maxRetryCount = maxRetryCount;
    }

    /**
     * Sets the connection timeout in milliseconds.
     *
     * @param connTimeout The connection timeout.
     */
    public void setConnTimeout(final long connTimeout) {
        this.connTimeout = connTimeout;
    }

    /**

  // TODO(lukes): Consider throwing InterruptedException when appropriate.
  @Override
  @ParametricNullness
  public V get() {
    return value;
  }

  @Override
  @ParametricNullness
  public V get(long timeout, TimeUnit unit) throws ExecutionException {
    checkNotNull(unit);
    return get();
  }

  @Override
  public boolean isCancelled() {
    return false;
  }

  @Override
  public boolean isDone() {

	// By default, empty data usage cache
	*d = dataUsageCache{}

	load := func(name string, timeout time.Duration) (bool, error) {
		// Abandon if more than time.Minute, so we don't hold up scanner.
		// drive timeout by default is 2 minutes, we do not need to wait longer.
		ctx, cancel := context.WithTimeout(ctx, timeout)
		defer cancel()

        this.index = index;
    }

    /**
     * Gets the scroll timeout in milliseconds.
     *
     * @return The scroll timeout.
     */
    public int getScrollTimeout() {
        return scrollTimeout;
    }

    /**
     * Sets the scroll timeout.
     * @param scrollTimeout The scroll timeout.
     */
    public void setScrollTimeout(final int scrollTimeout) {

        name = "Flaky Test Quarantine - ${testCoverage.asName()}"
        description = "Run all flaky tests skipped multiple times"

        applyDefaultSettings(os = os, arch = arch, buildJvm = BuildToolBuildJvm, timeout = 60)

        if (os == Os.LINUX) {
            steps {
                script {
                    // Because we exclude tests in `distributions-integ-tests` below, `@Flaky` won't work in that subproject.

    resultFuture = buildChainingFuture(inputFuture);
    listener = new MockFutureListener(resultFuture);
  }

  public void testFutureGetBeforeCallback() throws Exception {
    // Verify that get throws a timeout exception before the callback is called.
    assertThrows(TimeoutException.class, () -> resultFuture.get(1L, MILLISECONDS));
  }

  public void testFutureGetThrowsWrappedException() throws Exception {

	ctxt.SFTP = ctx.StringSlice("sftp")
	ctxt.Interface = ctx.String("interface")
	ctxt.UserTimeout = ctx.Duration("conn-user-timeout")
	ctxt.SendBufSize = ctx.Int("send-buf-size")
	ctxt.RecvBufSize = ctx.Int("recv-buf-size")
	ctxt.IdleTimeout = ctx.Duration("idle-timeout")
	ctxt.UserTimeout = ctx.Duration("conn-user-timeout")

	if conf := ctx.String("config"); len(conf) > 0 {
		err = mergeServerCtxtFromConfigFile(conf, ctxt)
	} else {

        // Given
        SmbOperationException.RetryPolicy policy = new SmbOperationException.RetryPolicy(3, 1000, 10000, 2.0, true);

        exception = new SmbOperationException(SmbOperationException.ErrorCode.CONNECTION_TIMEOUT, "Timeout occurred", null, policy, null);

        // When/Then - Should retry for attempts 1 and 2
        assertTrue(exception.shouldRetry(1));
        assertTrue(exception.shouldRetry(2));