FileEntry mock = mock(FileEntry.class);
        when(mock.getName()).thenReturn("test.txt");
        when(mock.getType()).thenReturn(1);
        when(mock.getAttributes()).thenReturn(128);
        when(mock.createTime()).thenReturn(1_500L);
        when(mock.lastModified()).thenReturn(1_800L);
        when(mock.length()).thenReturn(100L);

        assertEquals("test.txt", mock.getName());

        when(mockConfig.getNetbiosLocalAddress()).thenReturn(InetAddress.getByName("127.0.0.1"));
        when(mockConfig.getBroadcastAddress()).thenReturn(InetAddress.getByName("255.255.255.255"));
        when(mockConfig.getNetbiosSndBufSize()).thenReturn(576);
        when(mockConfig.getNetbiosRcvBufSize()).thenReturn(576);
        when(mockConfig.getResolveOrder()).thenReturn(Arrays.asList(ResolverType.RESOLVER_DNS)); // Only DNS, no broadcast

        // Setup mock notify information
        when(mockNotifyInfo1.getAction()).thenReturn(FileNotifyInformation.FILE_ACTION_ADDED);
        when(mockNotifyInfo1.getFileName()).thenReturn("test1.txt");

        when(mockNotifyInfo2.getAction()).thenReturn(FileNotifyInformation.FILE_ACTION_MODIFIED);
        when(mockNotifyInfo2.getFileName()).thenReturn("test2.txt");
    }

    @Test

            mockedCrypto.when(Crypto::getMD5).thenReturn(mockMd5);
            byte[] innerDigest = "inner_digest".getBytes();
            byte[] finalDigest = "final_digest".getBytes();

            when(mockMd5.digest()).thenReturn(innerDigest);
            when(mockMd5.digest(any(byte[].class))).thenReturn(finalDigest);

            HMACT64 hmac = new HMACT64(TEST_KEY);

     */
    @Test
    void testIsConnected() {
        when(smbTreeHandle.isConnected()).thenReturn(true);
        assertTrue(smbTreeHandle.isConnected(), "isConnected() should return true");
        when(smbTreeHandle.isConnected()).thenReturn(false);
        assertFalse(smbTreeHandle.isConnected(), "isConnected() should return false after status change");
    }

    /**

        SmbTreeHandleImpl dh = mock(SmbTreeHandleImpl.class);
        when(sh.isSMB2()).thenReturn(true);
        when(dh.isSMB2()).thenReturn(true);
        when(sh.isSameTree(dh)).thenReturn(true);

        // Source open returns a handle that reports size 0
        SmbFileHandleImpl sfd = mock(SmbFileHandleImpl.class);
        when(sfd.getInitialSize()).thenReturn(0L);

        doNothing().when(messageBlock).reset();

        when(messageBlock.getMid()).thenReturn(mid);
        when(messageBlock.getCommand()).thenReturn(command);
        when(messageBlock.getDigest()).thenReturn(digest);
        when(messageBlock.getResponse()).thenReturn(response);

        // When - simulate a typical usage sequence
        messageBlock.setMid(mid);

            // Mock the NdrBuffer responses for decoding
            when(mockNdrBuffer.dec_ndr_long()).thenReturn(0x12345678);
            when(mockNdrBuffer.dec_ndr_short()).thenReturn((int) (short) 0x9ABC, (int) (short) 0xDEF0);
            when(mockNdrBuffer.dec_ndr_small()).thenReturn((int) (byte) 0x11, (int) (byte) 0x22, // clock seq bytes

        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.singletonList(channel1));

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

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

            when(mockBuffer.dec_ndr_small()).thenReturn(5)
                    .thenReturn(0)
                    .thenReturn(DcerpcConstants.RPC_PT_RESPONSE)
                    .thenReturn(DcerpcConstants.RPC_C_PF_BROADCAST);
            when(mockBuffer.dec_ndr_long()).thenReturn(0x00000010);
            when(mockBuffer.dec_ndr_short()).thenReturn(100).thenReturn(1); // Non-zero auth_value length