NetworkInterfaceInfo remote2 = new NetworkInterfaceInfo(addr2, 10000); // Faster interface

        channel1 = new ChannelInfo("channel1", mockTransport1, local, remote1);
        channel1.setState(ChannelState.ESTABLISHED);

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

    @Test
    void testSingleChannelSelection() {

        
        // Establish connection
        channel.setState(ChannelState.CONNECTING);
        transport.connect();
        
        // Perform channel binding
        performChannelBinding(channel);
        
        // Add to active channels
        channels.put(channelId, channel);
        channel.setState(ChannelState.ESTABLISHED);
        

     */
    public ChannelInfo getChannelForTransport(SmbTransport transport) {
        return channels.values().stream().filter(c -> c.getTransport() == transport).findFirst().orElse(null);
    }

    /**
     * Remove a channel
     *
     * @param channel channel to remove
     */
    public void removeChannel(ChannelInfo channel) {
        channels.remove(channel.getChannelId());

        } catch (Exception e) {
            log.warn("Failed to recover channel {}: {}", channel.getChannelId(), e.getMessage());

            // Schedule next retry or remove
            handleFailure(channel, e);
        }
    }

    private void removeChannel(ChannelInfo channel) {
        log.info("Removing failed channel {} after max retries", channel.getChannelId());

            return new ArrayList<>(channels);
        }

        public void addChannel(ChannelInfo channel) {
            channels.add(channel);
            if (primaryChannel == null) {
                primaryChannel = channel;
            }
        }

        public void removeChannel(ChannelInfo channel) {
            channels.remove(channel);

            return false;
        ChannelInfo that = (ChannelInfo) obj;
        return channelId != null ? channelId.equals(that.channelId) : that.channelId == null;
    }

    @Override
    public int hashCode() {
        return channelId != null ? channelId.hashCode() : 0;
    }

            long pending = channel.getRequestsPending();
            if (pending < minPending) {
                minPending = pending;
                leastLoadedChannel = channel;
            }
        }
        return leastLoadedChannel != null ? leastLoadedChannel : channels.iterator().next();
    }

    private ChannelInfo selectWeightedRandom(Collection<ChannelInfo> channels) {

        ChannelInfo channel = new ChannelInfo("test-channel", mockTransport, localNic, remoteNic);
        channel.setState(ChannelState.ESTABLISHED);

        // Verify channel properties
        assertEquals("test-channel", channel.getChannelId());
        assertEquals(mockTransport, channel.getTransport());
        assertEquals(localNic, channel.getLocalInterface());

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

        when(mockLoadBalancer.selectChannel(mockOperation)).thenThrow(new ChannelLoadBalancer.NoAvailableChannelException("No channels"));

        IOException error = new IOException("Connection failed");

        // Should not throw exception even if no alternative channels
        assertDoesNotThrow(() -> failover.handleFailure(failedChannel, error));

        assertEquals(0, failedChannel.getRequestsPending());