// Then
        assertEquals(300L, dfsTtl, "Should delegate DFS TTL");
        assertEquals(30000, responseTimeout, "Should delegate response timeout");
        assertEquals(35000, socketTimeout, "Should delegate socket timeout");
        assertEquals(35000, connTimeout, "Should delegate connection timeout");
        assertEquals(10, maxMpx, "Should delegate max MPX count");

        verify(mockDelegate).getDfsTtl();

    private var editCount = 0

    override fun poll(): T = delegate.poll()

    override fun poll(
      timeout: Long,
      unit: TimeUnit,
    ): T? {
      return taskRunner.withLock {
        val waitUntil = nanoTime + unit.toNanos(timeout)
        while (true) {
          val result = poll()
          if (result != null) return@withLock result

            assertTrue(cbWithMetrics.getAverageResponseTime() > 0, "Average response time should be tracked");
            assertTrue(cbWithMetrics.getMinResponseTime() > 0, "Min response time should be tracked");
            assertTrue(cbWithMetrics.getMaxResponseTime() > 0, "Max response time should be tracked");
            assertTrue(cbWithMetrics.getP95ResponseTime() > 0, "P95 response time should be tracked");

    }

    /**
     * Gets the connection timeout in milliseconds.
     *
     * @return The connection timeout
     */
    public int getConnectTimeout() {
        return connectTimeout;
    }

    /**
     * Sets the connection timeout in milliseconds.
     *
     * @param connectTimeout The connection timeout
     */
    public void setConnectTimeout(final int connectTimeout) {

                // Simulate quick completion before timeout
                ResponseData responseData = new ResponseData();
                responseData.setHttpStatusCode(200);
                return responseData;
            }
        };

        // Test with timeout set
        client.setAccessTimeout(10);
        try {

Joe Tsai <******@****.***> 1503947278 -0700

Joe Tsai <******@****.***> 1503947278 -0700

Joe Tsai <******@****.***> 1503947278 -0700

Joe Tsai <******@****.***> 1503947278 -0700

Joe Tsai <******@****.***> 1503947278 -0700