channel2 = new ChannelInfo("channel2", mockTransport2, local, remote2);
        channel2.setState(ChannelState.ESTABLISHED);
    }

    @Test
    void testSingleChannelSelection() {
        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.singletonList(channel1));

        ChannelInfo selected = loadBalancer.selectChannel(mockMessage);