} 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();
            }

                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

            channels.add(channel);
            if (primaryChannel == null) {
                primaryChannel = channel;
            }
        }

        public void removeChannel(ChannelInfo channel) {
            channels.remove(channel);
            if (primaryChannel == channel && !channels.isEmpty()) {
                primaryChannel = channels.get(0);
            }
        }

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