// Should allow retries up to limit
        assertTrue(state.shouldRetry());

        state.incrementRetry(); // Retry 1
        assertTrue(state.shouldRetry());

        state.incrementRetry(); // Retry 2
        assertTrue(state.shouldRetry());

        state.incrementRetry(); // Retry 3
        assertFalse(state.shouldRetry()); // Max retries reached
    }

    @Test

     * Check if retry should be attempted
     *
     * @param attemptNumber the current attempt number
     * @return true if retry should be attempted
     */
    public boolean shouldRetry(int attemptNumber) {
        return retryPolicy.shouldRetry(this, attemptNumber);
    }

    /**
     * Get the delay before next retry
     *
     * @param attemptNumber the current attempt number
     * @return delay in milliseconds

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

        // Verify exponential backoff
        assertEquals(1000, exception.getRetryDelayMs(1));

                removeChannel(failedChannel);
                return;
            }
            // Recovery might be viable, attempt it with pre-created transport
            if (state.shouldRetry()) {
                scheduleRecoveryWithTransport(failedChannel, state, testTransport);
            } else {
                // Remove channel after max retries
                removeChannel(failedChannel);
            }

    }

    /**
     * Checks if retry should be attempted
     *
     * @return true if retry is recommended
     */
    public boolean shouldRetry() {
        return severity == Severity.RECOVERABLE || severity == Severity.TRANSIENT;
    }

    /**
     * Gets recommended retry delay in milliseconds
     *
     * @return retry delay or 0 if no retry

            this.failureTime = System.currentTimeMillis();
            this.retryCount = 0;
            this.nextRetryTime = failureTime + 1000;  // Initial 1 second delay
        }
        
        public boolean shouldRetry() {
            return retryCount < 3 && System.currentTimeMillis() >= nextRetryTime;
        }
        
        public void incrementRetry() {
            retryCount++;