ChannelInfo selected = loadBalancer.selectChannel(mockMessage);
        assertEquals(channel1, selected);
    }

    @Test
    void testNoChannelsAvailable() {
        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.emptyList());

        assertThrows(ChannelLoadBalancer.NoAvailableChannelException.class, () -> loadBalancer.selectChannel(mockMessage));