when(readResponse.getFlags2()).thenReturn(SmbConstants.FLAGS2_SECURITY_SIGNATURES);
        when(readResponse.getSignSeq()).thenReturn(10);
        when(readResponse.getCommand()).thenReturn((int) ServerMessageBlock.SMB_COM_READ_ANDX);
        when(readResponse.getLength()).thenReturn(100);
        when(readResponse.getDataLength()).thenReturn(50);
        when(readResponse.getData()).thenReturn(readData);

        // Set up mock behaviors
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockContext.getCredentials()).thenReturn(mockCredentials);
        when(mockContext.getTransportPool()).thenReturn(mockTransportPool);
        when(mockConfig.getDfsTtl()).thenReturn(300L);

        // Mock transport to throw IOException - simulating connection failure

            samr.SamrSamEntry entry = new samr.SamrSamEntry();
            when(mockNdrBuffer.dec_ndr_long()).thenReturn(1, 100); // idx, _name_bufferp
            when(mockNdrBuffer.dec_ndr_short()).thenReturn(4, 6); // length, maximum_length
            when(mockDeferredBuffer.dec_ndr_long()).thenReturn(-1); // Invalid _name_buffers

            // When/Then: Should throw exception

 */
class SmbComTreeDisconnectTest {

    /**
     * Test the constructor of SmbComTreeDisconnect.
     */
    @Test
    void testConstructor() {
        // When
        SmbComTreeDisconnect smbComTreeDisconnect = new SmbComTreeDisconnect();

        // Then
        assertEquals(ServerMessageBlock.SMB_COM_TREE_DISCONNECT, smbComTreeDisconnect.command, "Command should be SMB_COM_TREE_DISCONNECT");
    }

    /**

        }

        @Test
        @DisplayName("Should return tree ID")
        void testGetTid() {
            // Given
            when(mockResponse.getTid()).thenReturn(12345);

            // When
            int tid = mockResponse.getTid();

            // Then
            assertEquals(12345, tid, "Should return the configured tree ID");
            verify(mockResponse).getTid();
        }

        @Test

     */
    @Test
    void testGetKey_KeyExists() throws LoginException {
        when(key1.getKeyType()).thenReturn(KEY_TYPE_1);
        when(key2.getKeyType()).thenReturn(KEY_TYPE_2);
        Set<Object> privateCredentials = new HashSet<>();
        privateCredentials.add(key1);
        privateCredentials.add(key2);
        when(subject.getPrivateCredentials()).thenReturn(privateCredentials);

    @BeforeEach
    void setUp() throws Exception {
        when(ctx.getConfig()).thenReturn(cfg);
        when(cfg.isSigningEnforced()).thenReturn(false);
        when(cfg.getSessionTimeout()).thenReturn(30_000);
        when(cfg.getResponseTimeout()).thenReturn(5_000);
        when(address.getHostAddress()).thenReturn("127.0.0.1");
        when(address.getHostName()).thenReturn("localhost");

        // When & Then - should handle gracefully
        assertThrows(Exception.class, () -> {
            response.readBytesWireFormat(buffer, offset);
        });
    }

    @Test
    @DisplayName("Should verify NotifyResponse interface implementation")
    void testNotifyResponseInterface() {
        // Then
        assertTrue(response instanceof NotifyResponse);

        when(kerberosKey.getKeyType()).thenReturn(ENCRYPTION_TYPE);

        try (MockedStatic<KerberosEncData> mockedEncData = Mockito.mockStatic(KerberosEncData.class)) {
            mockedEncData.when(() -> KerberosEncData.decrypt(ENCRYPTED_DATA, kerberosKey, ENCRYPTION_TYPE)).thenReturn(decryptedData);

            // When

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.net.InetAddress;
import java.net.UnknownHostException;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;