}
    }

    /**
     * Removes all channels for a session.
     *
     * @param sessionId the session identifier
     */
    public void removeChannels(String sessionId) {
        ChannelGroup channelGroup = sessionChannels.remove(sessionId);
        if (channelGroup != null) {
            for (ChannelInfo channel : channelGroup.getChannels()) {
                try {

            } else {
                // Remove channel after max retries
                removeChannel(failedChannel);
            }
        } catch (Exception e) {
            // Recovery not possible, remove synchronously
            removeChannel(failedChannel);
            return;
        }
    }

    /**
     * Shutdown the failover handler
     */
    public void shutdown() {

        failover.handleFailure(failedChannel, error);

        assertEquals(ChannelState.FAILED, failedChannel.getState());
        verify(mockChannelManager).removeChannel(failedChannel);
    }

    @Test
    void testFailoverStateCreation() {
        ChannelFailover.FailoverState state = new ChannelFailover.FailoverState("test-channel");

    }

    /**
     * Remove a channel
     *
     * @param channel channel to remove
     */
    public void removeChannel(ChannelInfo channel) {
        channels.remove(channel.getChannelId());
        try {
            if (channel.getTransport() != null) {
                channel.getTransport().close();
            }

        // Remove should not throw exception even if channel doesn't exist
        assertDoesNotThrow(() -> channelManager.removeChannel(channel));
    }

    @Test
    void testShutdown() {
        assertDoesNotThrow(() -> channelManager.shutdown());

        // Multiple shutdowns should be safe

                handleFailure(channel, e);
            }
        });
    }
    
    private void removeChannel(ChannelInfo channel) {
        log.info("Removing failed channel {} after max retries", channel.getChannelId());
        
        manager.removeChannel(channel);
        failoverStates.remove(channel.getChannelId());
        
        // Try to establish a replacement channel