// 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