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() {

        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