response: Response?,
        ) {
          t.printStackTrace(System.out)
          latch.countDown()
        }
      },
    )

    check(latch.await(30, TimeUnit.SECONDS)) { "Timed out waiting for test $number to finish." }
    val endNanos = System.nanoTime()
    val tookMs = TimeUnit.NANOSECONDS.toMillis(endNanos - startNanos.get())
    println("Took ${tookMs}ms")
  }

     *
     * @param error the error to check
     * @return true if error might be recoverable, false otherwise
     */
    private boolean isRecoverableError(Exception error) {
        // Timeout errors are often recoverable
        if (error instanceof SocketTimeoutException) {
            return true;
        }

        // Some socket errors might be recoverable

        verify(mockDeferredNdrBuffer, times(9)).enc_ndr_long(anyInt()); // 3 enc_ndr_long for each of 3 unicode strings
        verify(mockDeferredNdrBuffer, times(3)).enc_ndr_long(0);
        verify(mockDeferredNdrBuffer, times(3)).advance(anyInt());
        verify(mockDeferredNdrBuffer, times(4)).derive(anyInt()); // 3 for strings + 1 for guid
        verify(mockDeferredNdrBuffer, times(18)).enc_ndr_short(0); // 5+6+7 shorts with value 0

  }

  private fun logWithTime(message: String) {
    val startNs = startNs
    val timeMs =
      if (startNs == null) {
        // Event occurred before start, for an example an early cancel.
        0L
      } else {
        TimeUnit.NANOSECONDS.toMillis(System.nanoTime() - startNs)
      }

    logger.invoke("[$timeMs ms] $message")
  }

        assertTrue(transport.isSMB2());
        assertFalse(transport.isSMB2());
        verify(transport, times(2)).isSMB2();

        doThrow(new SmbException("query failed")).when(transport).isSMB2();
        assertThrows(SmbException.class, () -> transport.isSMB2());
        verify(transport, times(3)).isSMB2();
    }

    // Inflight requests count including edge values
    @ParameterizedTest

        assertEquals(400L, mockAllocInfo.getFree(), "free space should match stubbed value");
        verify(mockAllocInfo, times(1)).getCapacity();
        verify(mockAllocInfo, times(1)).getFree();
    }

    /**
     * Parameterised test of capacity values, including edge cases such as
     * zero and negative capacities.
     */
    @ParameterizedTest

        // SubCommand should be TRANS_WAIT_NAMED_PIPE (0x53)
        assertEquals(SmbComTransaction.TRANS_WAIT_NAMED_PIPE, pipe.subCommand);

        // Timeout is set to 0xFFFFFFFF (-1 when cast to int)
        assertEquals(0xFFFFFFFF, pipe.timeout);

        // Max parameter and data counts should be 0
        assertEquals(0, pipe.maxParameterCount);
        assertEquals(0, pipe.maxDataCount);

import okhttp3.internal.http2.ErrorCode
import okhttp3.testing.Flaky
import org.junit.jupiter.api.AfterEach
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.Timeout
import org.junit.jupiter.api.extension.RegisterExtension
import org.junitpioneer.jupiter.RetryingTest
import org.opentest4j.TestAbortedException

/**

        }
    }

    @Override
    public ByteBuffer receive(int timeout) throws IOException {
        if (state != RdmaConnectionState.ESTABLISHED && state != RdmaConnectionState.CONNECTED) {
            throw new IOException("Connection not established");
        }

        try {
            socket.setSoTimeout(timeout);

            // First, read the header to determine message size

    assert response.json() == {"message": "Not timed"}
    assert "X-Response-Time" not in response.headers


def test_get_timed(client: TestClient):
    response = client.get("/timed")
    assert response.json() == {"message": "It's the time of my life"}
    assert "X-Response-Time" in response.headers