@Mock
    private SmbTransport mockTransport2;

    @Mock
    private CommonServerMessageBlock mockMessage;

    private ChannelLoadBalancer loadBalancer;
    private ChannelInfo channel1;
    private ChannelInfo channel2;

    @BeforeEach
    void setUp() throws UnknownHostException {
        loadBalancer = new ChannelLoadBalancer(mockChannelManager);

        channelInfo.addBytesSent(1000);
        channelInfo.addBytesReceived(2000);
        channelInfo.incrementRequestsSent();
        channelInfo.incrementErrors();

        assertEquals(1000, channelInfo.getBytesSent());
        assertEquals(2000, channelInfo.getBytesReceived());
        assertEquals(1, channelInfo.getRequestsSent());
        assertEquals(1, channelInfo.getErrors());

        }
    }

    private ChannelInfo selectRoundRobin(Collection<ChannelInfo> channels) {
        List<ChannelInfo> list = new ArrayList<>(channels);
        int index = Math.abs(roundRobinCounter.getAndIncrement() % list.size());
        return list.get(index);
    }

    private ChannelInfo selectLeastLoaded(Collection<ChannelInfo> channels) {
        ChannelInfo leastLoadedChannel = null;

        NetworkInterfaceInfo localInterface = new NetworkInterfaceInfo(addr, 1000);
        NetworkInterfaceInfo remoteInterface = new NetworkInterfaceInfo(addr, 1000);
        ChannelInfo healthyChannel = new ChannelInfo("test-channel", mockTransport, localInterface, remoteInterface);
        healthyChannel.setState(ChannelState.ESTABLISHED);

        // Use reflection or package-private method to add channel for testing

        NetworkInterfaceInfo remoteNic = new NetworkInterfaceInfo(remoteAddr, 445);

        // Mock transport doesn't have isConnected, but we can work around it

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

        // Verify channel properties
        assertEquals("test-channel", channel.getChannelId());

    }

    @Test
    void testPendingOperationRedistribution() throws Exception {
        // Setup pending operations
        failedChannel.addPendingOperation(mockOperation);

        ChannelInfo alternativeChannel = mock(ChannelInfo.class);
        when(mockLoadBalancer.selectChannel(mockOperation)).thenReturn(alternativeChannel);

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