}

    /**
     * Get the delay before next retry
     *
     * @param attemptNumber the current attempt number
     * @return delay in milliseconds
     */
    public long getRetryDelayMs(int attemptNumber) {
        return retryPolicy.getDelayMs(attemptNumber);
    }

    /**
     * Add context information
     *
     * @param key the context key
     * @param value the context value

        assertFalse(exception.shouldRetry(3)); // Max attempts reached

        // Verify exponential backoff
        assertEquals(1000, exception.getRetryDelayMs(1));
        assertEquals(2000, exception.getRetryDelayMs(2));
        assertEquals(4000, exception.getRetryDelayMs(3));
    }

    @Test
    @DisplayName("Should handle no retry policy")
    void testNoRetryPolicy() {
        // When