when(mockChannelManager.getLoadBalancer()).thenReturn(mockLoadBalancer);
    }

    @Test
    void testHandleFailure() {
        IOException error = new IOException("Connection failed");

        failover.handleFailure(failedChannel, error);

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

    @Test
    void testFailoverStateCreation() {

    }

    /**
     * Handle channel failure
     *
     * @param failedChannel failed channel
     * @param error error that caused failure
     */
    public void handleFailure(ChannelInfo failedChannel, Exception error) {
        log.warn("Channel {} failed: {}", failedChannel.getChannelId(), error.getMessage());

        // Mark channel as failed
        failedChannel.setState(ChannelState.FAILED);

    public ChannelInfo selectChannel(SMBMessage message) {
        return loadBalancer.selectChannel(message);
    }
    
    public void handleChannelFailure(ChannelInfo channel, Exception error) {
        failover.handleFailure(channel, error);
    }
    
    private void performHealthCheck() {
        for (ChannelInfo channel : channels.values()) {
            if (channel.getIdleTime() > 60000) {  // 1 minute idle

        channel.setState(ChannelState.ESTABLISHED);

        // Simulate failure
        IOException failure = new IOException("Connection lost");
        failover.handleFailure(channel, failure);

        // Verify channel state changed
        assertFalse(channel.isHealthy());
        assertEquals(ChannelState.FAILED, channel.getState());
    }

    @Test

     *
     * @param channel failed channel
     * @param error error that caused failure
     */
    public void handleChannelFailure(ChannelInfo channel, Exception error) {
        failover.handleFailure(channel, error);
    }

    /**
     * Get channel for specific transport
     *
     * @param transport transport instance
     * @return corresponding channel or null
     */