assertTrue(Arrays.asList(channel1, channel2).contains(second));
    }

    @Test
    void testLeastLoadedStrategy() {
        loadBalancer.setStrategy(LoadBalancingStrategy.LEAST_LOADED);

        // Add pending operations to channel1 to make it more loaded
        channel1.addPendingOperation(mockMessage);

        when(mockChannelManager.getHealthyChannels()).thenReturn(Arrays.asList(channel1, channel2));

        
        // 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());

            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);

        } 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 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());

                channel.truncate(0);
                props.store(Channels.newOutputStream(channel), "Last modified on: " + new Date());

                lock.release();
                lock = null;

                channel.close();
                channel = null;
            } catch (IOException e) {
                getLogger()
                        .debug(

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